Search papers, labs, and topics across Lattice.
This paper introduces Asymmetric Mutual Variational Learning (AMVL), a novel framework designed to address the train-inference mismatch in Multimodal Large Language Models (MLLMs) by employing a bidirectional calibration objective. By utilizing a forward KL divergence to align the target-agnostic prior with the posterior and a reverse KL divergence to regularize the posterior, AMVL mitigates the issue of "answer leakage" that hampers performance during inference. Experimental results demonstrate that AMVL significantly enhances reasoning capabilities, achieving an average score improvement of +10.83 on the complex BLINK benchmark and up to +32.00 on specific reasoning tasks compared to existing methods.
AMVL eliminates the critical train-inference mismatch in MLLMs, leading to substantial improvements in reasoning performance across multimodal tasks.
Multimodal Large Language Models (MLLMs) are often constrained by a language-space bottleneck, forcing complex visual reasoning into discrete tokens which can lose perceptual nuance. A promising alternative is continuous latent reasoning, where the goal is to discover implicit reasoning pathways that bridge the multimodal query and the final answer. However, this introduces a severe train-inference mismatch: a training-time posterior, conditioned on the ground-truth answer, can exploit answer-dependent shortcuts. Standard variational training then forces the inference-time prior to mimic a posterior that has access to information unavailable at test time, leading to poor performance. To address this, we propose Asymmetric Mutual Variational Learning (AMVL), a framework that resolves this mismatch via a bidirectional calibration objective. A forward KL divergence trains the target-agnostic prior to match the posterior, while a novel reverse KL divergence simultaneously regularizes the posterior, preventing it from collapsing into inference-incompatible regions and mitigating this ``answer leakage''. We provide theoretical analysis formalizing this leakage as prior contamination and prove that our dual-KL objective reduces it. We instantiate AMVL in a latent-integrated MLLM and show that it consistently outperforms strong discrete and latent-reasoning baselines, improving the average score on the complex BLINK benchmark by +10.83 and achieving gains of up to +32.00 on individual reasoning tasks, with analyses confirming improved latent-space stability.
