ITSELF: Attention Guided Fine-Grained Alignment for Vision-Language Retrieval

Tien-Huy Nguyen, Huu-Loc Tran, Thanh Duc Ngo
1/3/2026
cs.CVcs.AIcs.IR

No official code found. Best implementation: trhuuloc/itself (4 stars, last commit 2026-05-01).

Abstract

Vision Language Models (VLMs) have rapidly advanced and show strong promise for text-based person search (TBPS), a task that requires capturing fine-grained relationships between images and text to distinguish individuals. Previous methods address these challenges through local alignment, yet they are often prone to shortcut learning and spurious correlations, yielding misalignment. Moreover, injecting prior knowledge can distort intra-modality structure. Motivated by our finding that encoder attention surfaces spatially precise evidence from the earliest training epochs, and to alleviate these issues, we introduceITSELF, an attention-guided framework for implicit local alignment. At its core, Guided Representation with Attentive Bank (GRAB) converts the model's own attention into an Attentive Bank of high-saliency tokens and applies local objectives on this bank, learning fine-grained correspondences without extra supervision. To make the selection reliable and non-redundant, we introduce Multi-Layer Attention for Robust Selection (MARS), which aggregates attention across layers and performs diversity-aware top-k selection; and Adaptive Token Scheduler (ATS), which schedules the retention budget from coarse to fine over training, preserving context early while progressively focusing on discriminative details. Extensive experiments on three widely used TBPS benchmarks showstate-of-the-art performance and strong cross-dataset generalization, confirming the effectiveness and robustness of our approach without additional prior supervision. Our project is publicly available at https://trhuuloc.github.io/itself

View on arXivView PDF

Code Implementations(1)

trhuuloc/itselfCommunity60%

[WACV 2026] ITSELF: Attention Guided Fine-Grained Alignment for Vision–Language Retrieval

40CSS, HTMLDec 1, 20255 months ago

Similar Papers

Cite this paper

@article{nguyen2026itself,
  title  = {ITSELF: Attention Guided Fine-Grained Alignment for Vision-Language Retrieval},
  author = {Tien-Huy Nguyen and Huu-Loc Tran and Thanh Duc Ngo},
  year   = {2026},
  eprint = {2601.01024},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.01024v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.01024.svg)](https://codeofpaper.com/paper/2601.01024)

Discussion