Search papers, labs, and topics across Lattice.
T-REN addresses weak alignment and high token counts in vision-language encoders by learning text-aligned region tokens. A lightweight network is trained on top of a frozen vision backbone to pool patch-level representations into region tokens, aligning them with region-level text annotations. T-REN achieves state-of-the-art results on dense cross-modal understanding tasks while significantly reducing token counts for both images and videos.
Forget slow, fine-grained vision-language models: T-REN slashes token counts by 24x (images) and 187x (videos) while boosting performance on segmentation, retrieval, and localization.
Despite recent progress, vision-language encoders struggle with two core limitations: (1) weak alignment between language and dense vision features, which hurts tasks like open-vocabulary semantic segmentation; and (2) high token counts for fine-grained visual representations, which limits scalability to long videos. This work addresses both limitations. We propose T-REN (Text-aligned Region Encoder Network), an efficient encoder that maps visual data to a compact set of text-aligned region-level representations (or region tokens). T-REN achieves this through a lightweight network added on top of a frozen vision backbone, trained to pool patch-level representations within each semantic region into region tokens and align them with region-level text annotations. With only 3.7% additional parameters compared to the vision-language backbone, this design yields substantially stronger dense cross-modal understanding while reducing the token count by orders of magnitude. Specifically, T-REN delivers +5.9 mIoU on ADE20K open-vocabulary segmentation, +18.4% recall on COCO object-level text-image retrieval, +15.6% recall on Ego4D video object localization, and +17.6% mIoU on VSPW video scene parsing, all while reducing token counts by more than 24x for images and 187x for videos compared to the patch-based vision-language backbone. The code and model are available at https://github.com/savya08/T-REN.
