事実関係
arXiv:2608.06411v1 Announce Type: new Abstract: Multimodal large language models (MLLMs) achieve strong performance across diverse vision-language tasks, but their efficiency is limited by the cost of processing numerou
解説と影響
多模态大语言模型在视觉问答、图像描述等任务上表现优异，但其处理大量视觉令牌（visual tokens）的过程带来了高昂的计算开销。来自 arXiv 的最新研究提出了一种名为“学习预测中间层注意力”的新方法，旨在对视觉令牌进行动态剪枝，从而提升模型效率。该研究的核心思路是，并非所有输入的图像区域都对最终的语言生成任务同等重要，如果能提前识别并移除那些“不重要”的视觉令牌，就能在不显著牺牲性能的前提下，大幅减少模型的计算量 [arXiv:2608.06411]。
具体而言，该方法在模型的中间层引入了一个轻量级的预测模块。这个模块会学习去模仿模型更深层或最终层的注意力分布，从而预判哪些视觉令牌将在后续的处理中被重点关注。基于这一预测结果，模型可以在信息传递到更深、计算成本更高的层之前，就将那些预测为低注意力的视觉令牌剪除。这相当于为模型装上了一道智能“关卡”，只让关键视觉信息通过，从而实现了对计算资源的动态分配 [arXiv:2608.06411]。
这项工作的意义在于，它为多模态大模型的部署，尤其是在资源受限的边缘设备或需要实时响应的应用场景中，提供了一种新的加速思路。与传统的静态剪枝或压缩技术不同，这种基于中间层注意力预测的动态方法，能够根据输入图像的具体内容自适应地调整计算量。例如，对于一张背景简单的物体特写，模型可以剪除大量背景令牌；而对于一张细节丰富的街景，则会保留更多信息。这种内容感知的灵活性，是该方法区别于其他效率优化方案的关键所在。尽管原文未提供具体的性能对比数据，但其技术路径为平衡多模态模型的性能与效率提供了一个值得关注的方向 [arXiv:2608.06411]。
参考資料
出典原文
arXiv:2608.06411v1 Announce Type: new Abstract: Multimodal large language models (MLLMs) achieve strong performance across diverse vision-language tasks, but their efficiency is limited by the cost of processing numerous visual tokens. Visual token pruning can reduce this cost, but requires accurate token importance estimates. Recent studies have demonstrated that text-to-vision attention from middle language model layers can effectively guide visual token pruning, typically using attention from a predefined middle layer to select the visual tokens to retain. Two problems therefore remain. First, our analysis shows that the layer whose attention is most responsive to the question varies substantially across samples, making a fixed layer suboptimal. Second, obtaining attention from the appropriate middle layer requires processing numerous visual tokens through several language model layers, by which point considerable computation has already been spent. To address both problems, we propose Middle-layer Attention Prediction (MAP), which uses Question Contrastive Teacher Selection to identify a sample-specific teacher layer by contrasting attention under the original and