DRIFT: 视觉语言模型中用于连续输出解码的残差流适配器

Hugging Face Daily Papers 论文

摘要

DRIFT 是一种框架,通过将粗预测与迭代流匹配精化相结合,使预训练的视觉语言模型能够进行连续输出解码,从而提升感知和规划任务的性能。

许多现代视觉语言模型(VLM)基于自回归解码离散令牌。虽然基于文本的输出接口能够实现可扩展的预训练和在多种任务上的强大零样本泛化能力,但对于需要精确连续输出的问题(例如定位事件的时间边界或生成机器人控制动作)却不太适用。为了解决这一挑战,我们提出了 DRIFT,这是一个通用的框架,用于将预训练的 VLM 适配到连续解码任务。DRIFT 结合了一个基础预测器(提供目标输出的粗略估计)和一个基于流匹配的生成式精化模块,通过迭代改进预测。这种残差公式将生成建模问题从学习全局输出分布转变为在强先验周围建模局部残差分布,从而大幅简化了优化过程。我们在感知和规划任务上评估了 DRIFT,包括视觉定位和机器人控制。在涵盖 MLLM、VLA 和 WAM 的多种任务和架构上,DRIFT 始终优于一系列强大的基于回归和基于生成的解决方案。
查看原文
查看缓存全文

缓存时间: 2026/06/11 17:35

论文页面 - DRIFT: 一种用于解码视觉-语言模型中连续输出的残差流适配器

来源: https://huggingface.co/papers/2606.05758

摘要

DRIFT 是一个框架,通过结合粗预测与基于流匹配的迭代精化,适配预训练的视觉-语言模型以进行连续解码任务,从而在感知和规划任务上提升性能。

许多现代视觉-语言模型 (https://huggingface.co/papers?q=vision-language%20models) (VLM) 构建在离散令牌 (https://huggingface.co/papers?q=discrete%20tokens) 的自回归解码 (https://huggingface.co/papers?q=autoregressive%20decoding) 之上。虽然基于文本的输出接口支持可扩展的预训练以及跨多样任务的强零样本泛化能力,但它们并不适合需要精确连续输出 (https://huggingface.co/papers?q=continuous%20outputs) 的问题,例如定位事件的时间边界或生成机器人控制 (https://huggingface.co/papers?q=robotic%20control) 动作。为了解决这一挑战,我们提出了 DRIFT,一个用于适配预训练 VLM 到连续解码任务的通用框架。DRIFT 结合了一个基础预测器 (https://huggingface.co/papers?q=base%20predictor)(提供目标输出的粗估计)和一个基于流匹配 (https://huggingface.co/papers?q=flow%20匹配) 的生成式精化模块 (https://huggingface.co/papers?q=generative%20refinement%20module)(迭代改进预测)。这种残差公式 (https://huggingface.co/papers?q=residual%20formulation) 将生成建模问题从学习全局输出分布 (https://huggingface.co/papers?q=global%20output%20distribution) 转化为在强先验周围建模局部残差分布 (https://huggingface.co/papers?q=localized%20residual%20distribution),从而显著简化优化过程。我们在感知和规划任务上(包括视觉定位 (https://huggingface.co/papers?q=visual%20grounding) 和机器人控制 (https://huggingface.co/papers?q=robotic%20control))评估了 DRIFT。在跨 MLLMs、VLAs 和 WAMs 的多个任务和架构中,DRIFT 始终优于一组强大的基于回归和生成的解决方案。

查看 arXiv 页面 (https://arxiv.org/abs/2606.05758) | 查看 PDF (https://arxiv.org/pdf/2606.05758) | 项目页面 (https://dragonlzm.github.io/DRIFT/) | GitHub0 (https://github.com/dragonlzm/DRIFT) | 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2606.05758)

引用此论文的模型0

无模型链接此论文

请在模型 README.md 中引用 arxiv.org/abs/2606.05758 以从此页面链接。

引用此论文的数据集0

无数据集链接此论文

请在数据集 README.md 中引用 arxiv.org/abs/2606.05758 以从此页面链接。

引用此论文的 Space0

无 Space 链接此论文

请在 Space README.md 中引用 arxiv.org/abs/2606.05758 以从此页面链接。

包含此论文的收藏集0

无收藏集包含此论文

请将此论文添加到一个收藏集 (https://huggingface.co/new-collection) 以从此页面链接。

相似文章

用于优化离散扩散语言模型的漂移目标

arXiv cs.CL

本文提出TokenDrift,一种漂移目标方法,通过将分类预测提升至连续语义空间进行反对称漂移,从而优化离散扩散语言模型。在固定去噪步数下,该方法显著提升了生成质量。

LangFlow:连续扩散在语言建模中可与离散扩散相媲美

Hugging Face Daily Papers

LangFlow提出了首个可与离散扩散方法相媲美的连续扩散语言模型,挑战了长期以来认为连续扩散在语言建模中劣于离散扩散的观点。该工作引入了基于最优Gumbel噪声调度等关键要素,并展示了与离散扩散基线相比具有竞争力的困惑度和迁移学习性能。

注意力漂移:自回归投机解码模型学到了什么

Reddit r/LocalLLaMA

本文指出了自回归投机解码模型中的“注意力漂移”现象,即草稿模型的注意力从提示词转移到了其自身生成的令牌上。作者提出了架构上的改进,例如后归一化(Post-norm)和 RMSNorm,这些改进在各种基准测试中提高了接受率和鲁棒性。