FLAT trains one encoder for both text-to-image and image-to-text tasks

Researchers present FLAT (Flexible-Length Aligned Transmodal representations), a pre-training framework built to fix a specific bottleneck in multimodal AI systems. Today's typical setup trains a visual encoder first, using contrastive or self-supervised learning, and only afterward attaches a separate downstream generative model. Because the encoder's embeddings are frozen by the time the generative model is trained, this two-stage split caps how good the generation can get. FLAT instead optimizes a shared multimodal encoder together with its text-to-image (T2I) and image-to-text (I2T) decoders in a single pass, combining contrastive alignment (which makes embeddings good at distinguishing concepts) with bidirectional cross-modal generative objectives (which makes the same embeddings usable as direct conditions for generation). Architecturally, FLAT maps both images and text into one continuous 1D sequence space, and applies nested dropout over the first K tokens of that sequence so the model can produce outputs at variable lengths rather than one fixed size. This single pre-training stage already supports both cross-modal retrieval and generation across a range of prefix lengths K, hitting a T2I GenEval score of 71.1. Fine-tuning the model for specific tasks pushes results further, matching state-of-the-art baselines: 83.1 GenEval on text-to-image generation, 40.5 BLEU-4 and 138.6 CIDEr on MS-COCO image captioning, and Recall@5 scores of 86.8 (image-to-text) and 75.8 (text-to-image) on MS-COCO, plus 98.3 (image-to-text) and 93.6 (text-to-image) on Flickr30K. Qualitative tests show the resulting representations natively support linear interpolation between embeddings, arithmetic operations in the latent space, and zero-shot composed retrieval (finding results for a query that combines an image with a text edit, without task-specific training).
Key facts
- FLAT jointly trains one multimodal encoder with text-to-image and image-to-text decoders in a single pre-training stage, instead of training the encoder and generative model separately.
- The single-stage pre-training alone reaches a T2I GenEval score of 71.1 across variable prefix lengths K.
- After task-specific fine-tuning, FLAT scores 83.1 GenEval on text-to-image generation and 40.5 BLEU-4 / 138.6 CIDEr on MS-COCO image captioning.
- Retrieval Recall@5 reaches 86.8 (image-to-text) and 75.8 (text-to-image) on MS-COCO, and 98.3 / 93.6 on Flickr30K.
- Visual and text inputs are mapped into one continuous 1D sequence space, with nested dropout over prefix-K tokens enabling flexible-length embeddings for both retrieval and generation.
Why it matters
Most multimodal systems still train a visual encoder first and bolt a generative model onto it afterward, which freezes the encoder's embeddings before the generative side ever sees them and caps how well generation can perform. FLAT trains both parts together from the start, so the same embeddings work as discriminative descriptors for retrieval and as direct conditions for generation, removing that bottleneck by construction.
Who it affects
The work targets researchers and engineers building multimodal systems that need both retrieval (finding matching images or text) and generation (producing images from text or captions from images) from a shared representation, rather than maintaining separate pipelines for each direction.
How to use it
FLAT's core mechanism is nested dropout applied over the first K tokens of a unified 1D sequence representation, which lets a single pre-trained model output embeddings of different lengths on demand. That means one model, pre-trained once, already covers cross-modal retrieval and generation at multiple output-length settings without retraining for each one; task-specific fine-tuning is used on top to push individual benchmark scores further.
How solid is it
The single pre-training stage alone reaches a T2I GenEval score of 71.1. After task-specific fine-tuning, the paper reports 83.1 GenEval on text-to-image generation, 40.5 BLEU-4 and 138.6 CIDEr on MS-COCO captioning, and Recall@5 of 86.8/75.8 (I2T/T2I) on MS-COCO and 98.3/93.6 on Flickr30K, which the authors say match state-of-the-art baselines. Qualitative demonstrations of linear interpolation, latent-space arithmetic and zero-shot composed retrieval add evidence beyond the benchmark numbers.
Risks and caveats
The source text names no authors or institutions, gives no date, venue or publication context, and does not identify which state-of-the-art baselines the fine-tuned scores are compared against, which makes the strength of that comparison hard to judge independently. It also gives no details on training data, compute cost or model size, so the practical cost of reproducing or deploying FLAT is unknown from this text alone.