arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.00574cs.AI

通过潜在嵌入学习模态混合的思考链推理

Learning Modal-Mixed Chain-of-Thought Reasoning with Latent Embeddings

University of California, San Diego(加州大学圣地亚哥分校)

Yifei Shao, Kun Zhou, Ziming Xu, Mohammad Atif Quamar, Shibo Hao, Zhen Wang, Zhiting Hu, Biwei Huang

AI总结:

本文提出模态混合CoT方法,通过潜在嵌入结合视觉与文本信息,提升多模态推理性能。

摘要:

我们研究如何将思考链(CoT)扩展到语言之外,以更好地处理多模态推理。虽然CoT帮助LLM和VLM表达中间步骤,但其纯文本形式在需要视觉关键中间状态的问题上往往失效。我们引入了模态混合CoT,该方法将文本标记与紧凑的视觉草图(以潜在嵌入表示)交织在一起。为了在不损害VLM原始知识和能力的前提下弥合模态差距,我们使用VLM本身作为编码器,并训练语言主干来重建其自身的中间视觉嵌入,以保证视觉潜在空间的语义一致性。我们进一步附加了一个基于扩散的潜在解码器,该解码器由特殊控制标记触发,并且基于VLM的隐藏状态进行条件化。这样,扩散头负责细粒度的感知细节,而VLM则指定高层次的意图,这清晰地分离了角色并减少了VLM的优化压力。训练分为两个阶段:首先在交织文本和潜在状态的轨迹上进行监督微调,采用联合下一个标记和潜在重构目标;随后进行强化学习,教导何时切换模态以及如何构建长推理链。在11种多样化的多模态推理任务上的广泛实验表明,我们的方法在语言-only和其它CoT方法上表现更优。我们的代码将公开发布。

英文摘要:

We study how to extend chain-of-thought (CoT) beyond language to better handle multimodal reasoning. While CoT helps LLMs and VLMs articulate intermediate steps, its text-only form often fails on vision-intensive problems where key intermediate states are inherently visual. We introduce modal-mixed CoT, which interleaves textual tokens with compact visual sketches represented as latent embeddings. To bridge the modality gap without eroding the original knowledge and capability of the VLM, we use the VLM itself as an encoder and train the language backbone to reconstruct its own intermediate vision embeddings, to guarantee the semantic alignment of the visual latent space. We further attach a diffusion-based latent decoder, invoked by a special control token and conditioned on hidden states from the VLM. In this way, the diffusion head carries fine-grained perceptual details while the VLM specifies high-level intent, which cleanly disentangles roles and reduces the optimization pressure of the VLM. Training proceeds in two stages: supervised fine-tuning on traces that interleave text and latents with a joint next-token and latent-reconstruction objective, followed by reinforcement learning that teaches when to switch modalities and how to compose long reasoning chains. Extensive experiments across 11 diverse multimodal reasoning tasks, demonstrate that our method yields better performance than language-only and other CoT methods. Our code will be publicly released.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑