Vision Language Image Captioning With Unified Encoder Decoder Architecture

via “vision-language image captioning with unified encoder-decoder architecture”

image-to-text model by undefined. 21,87,494 downloads.

Unique: Uses a lightweight ViT-B/16 image encoder paired with a 6-layer GPT-2 text decoder (139M total parameters), enabling efficient deployment on edge devices while maintaining competitive caption quality through contrastive vision-language pre-training on 14M image-text pairs. The unified architecture supports both image-text matching and caption generation without separate model heads.

vs others: Significantly smaller and faster than CLIP-based captioning pipelines (which require separate caption generation models) while maintaining comparable quality to larger models like ViLBERT or LXMERT due to superior pre-training data curation and contrastive learning approach.