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

DRIFT: A Residual Flow Adapter for Decoding Continuous Outputs in Vision-Language Models

About

Many modern vision-language models (VLMs) build on autoregressive decoding of discrete tokens. While text-based output interfaces enable scalable pretraining and strong zero-shot generalization across diverse tasks, they are poorly suited for problems that require precise continuous outputs, such as localizing temporal boundaries of events or generating robotic control actions. To address this challenge, we propose DRIFT, a general framework for adapting pretrained VLMs to continuous decoding tasks. DRIFT combines a base predictor, which provides a coarse estimate of the target output, with a generative refinement module based on flow matching that iteratively improves the prediction. This residual formulation transforms the generative modeling problem from learning a global output distribution to modeling a localized residual distribution around a strong prior, substantially simplifying optimization. We evaluate DRIFT on both perception and planning tasks, including visual grounding and robotic control. Across multiple tasks and architectures spanning MLLMs, VLAs, and WAMs, DRIFT consistently outperforms a strong set of regression- and generative-based solutions.

Zhuoming Liu, Jinhong Lin, Kwan Man Cheng, Lin Zhang, Shayok Bagchi, Yin Li• 2026

Related benchmarks

TaskDatasetResultRank
Temporal Video GroundingCharades-STA (test)
Recall@IoU=0.544.8
139
Robotic ManipulationLIBERO
Long Success Rate96.4
108
Temporal Video GroundingActivityNet-Captions (test)
Recall@IoU>0.535.8
46
Robot ManipulationSIMPLER-WidowX
Success Rate61.5
13
Showing 4 of 4 rows

Other info

GitHub

Follow for update