Search papers, labs, and topics across Lattice.
This paper introduces Latent-OPD, an enhancement of On-Policy Distillation (OPD) that incorporates trajectory-level latent distillation to improve video reasoning in Large Multimodal Models (LMMs). By focusing on the hidden states at the end of each trajectory, Latent-OPD captures the accumulated visual evidence and reasoning context more effectively than traditional output-level supervision. Experiments across six benchmarks demonstrate that Latent-OPD significantly outperforms standard OPD, especially in tasks with limited frames or complex evidence aggregation.
Latent-OPD reveals that distilling latent representations at trajectory endpoints can dramatically enhance video reasoning efficiency in LMMs.
Large Multimodal Models (LMMs) for video reasoning have long been hindered by the high computational cost of processing vast amounts of visual information. This dilemma motivates the transfer of the reasoning capabilities of large models to smaller, more efficient ones. On-Policy Distillation (OPD) offers a promising solution by matching output-token distributions along student-generated trajectories. However, video reasoning often depends on evidence accumulated across multiple frames. In this context, output-level supervision only captures information expressed through token predictions and does not directly constrain the latent representations formed during reasoning. To address this limitation, we propose Latent-OPD, which augments OPD with trajectory-level latent distillation. Specifically, our method focuses on the position at the end of each trajectory, where hidden states effectively summarize the accumulated visual evidence and reasoning context. Furthermore, we introduce a progressive teacher-lookahead strategy, which aligns middle-to-late student layers with increasingly deeper teacher layers. Experiments on six video reasoning benchmarks show that Latent-OPD consistently outperforms output-only OPD. Notably, the improvements are particularly pronounced in scenarios with limited frames, long videos, or tasks requiring complex evidence aggregation. These results establish Latent-OPD as a highly effective approach to frame-efficient video reasoning.
