ReToken: Improving Long-Context VLMs with Visual Retrieval Token

RETOKEN introduces a single learnable embedding trained as an explicit retrieval target to select relevant visual tokens from a pre-filled cache. This method uses a two-pass strategy: first, retrieving query-relevant frames using the learned token, and second, answering only with those retrieved tokens. It significantly boosts performance on long video understanding and image QA benchmarks.

Open the full episode

More from AI Radio