Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

T-REN: Learning Text-Aligned Region Tokens Improves Dense Vision-Language Alignment and Scalability

About

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.

Savya Khosla, Sethuraman T V, Aryan Chadha, Alex Schwing, Derek Hoiem• 2026

Related benchmarks

TaskDatasetResultRank
Single-needle retrievalVisual Haystacks single-needle challenge
Recall@188.5
130
Video Semantic SegmentationVSPW
mIoU38.3
55
Open Vocabulary Semantic SegmentationCityscapes (val)
mIoU58.7
48
Open Vocabulary Semantic SegmentationADE20K (val)
mIoU32
11
Video Query LocalizationEgo4D (test)
Recall@152.4
3
Showing 5 of 5 rows

Other info

Follow for update