Papers

PILOT 框架:将推理建模为隐变量以改进多模态嵌入

Beyond Post-Hoc Rewards: Privileged Reasoning Exploration for Multimodal Embeddings

Why we picked it

一篇做多模态嵌入的论文,把推理当成隐变量来训练,MMEB-V2 上 78 个任务全赢了基线,视觉文档检索涨了 6 分多,做检索的可以看看。

论文提出 PILOT 框架,把推理过程建模为隐变量,用同一个 MLLM 的先验和后验两种推理模式来探索对检索有用的推理轨迹。后验在训练时利用关系证据引导探索,通过重要性加权的变分目标把探索成果迁移给先验。在包含 78 个任务的 MMEB-V2 基准上,PILOT 在 2B 和 7B 参数下均超过主基线,总分别提升 3.2 和 3.3 分,其中视觉文档检索提升最大,达 +6.1 和 +6.4 分。

Original · arXiv cs.AI

Beyond Post-Hoc Rewards: Privileged Reasoning Exploration for Multimodal Embeddings

Advances in Multimodal Large Language Models (MLLMs) have driven substantial progress in multimodal embeddings. Recent methods exploit generative capabilities by augmenting embedded inputs with self-generated reasoning to produce more informative embeddings. However, conventional retrieval supervision for training embedding models specifies pairwise relevance relations without explicitly guiding the reasoning used for input augmentation. Such reasoning is neither directly observed nor uniquely determined by the input, allowing multiple plausible traces to introduce semantic cues of varying retrieval utility. Existing approaches largely optimize reasoning conditioned on the input alone, limiting the exploration of retrieval-relevant traces. We introduce PILOT, a framework that models reasoning as a latent variable and instantiates its prior and posterior distributions as distinct reasoning modes of a shared MLLM. The prior conditions only on the input, whereas the posterior uses additional training-time relational evidence to guide reasoning exploration. To translate the posterior's broader exploration into inference-time improvements, we derive an importance-weighted variational objective and instantiate it as a prior-posterior co-evolution framework that improves both modes under a shared inference-compatible reward while selectively transferring posterior trajectories to the prior according to their utility and prior reachability. On the 78-task MMEB-V2 benchmark, PILOT outperforms the primary baseline across all 12 task groups at both 2B and 7B parameters, improving overall scores by 3.2 and 3.3 points, respectively, with the largest gains in visual-document retrieval (+6.1 and +6.4 points). Our results show that relational supervision is most valuable not as a post-hoc reward but as a signal for shaping how reasoning is explored in training reasoning-enhanced embedding models.