用线性复杂度实现顶尖多模态理解,让模型更快更省资源。
LinMU: Multimodal Understanding Made Linear
- 用双分支结构替代自注意力,实现线性计算复杂度。
- 在多个基准上性能持平原模型,长视频处理速度提升9倍。
- 适合需要高效推理的边缘设备和长序列多模态任务。
现代视觉-语言模型虽表现优异,但自注意力机制的二次复杂度限制了其在边缘设备上的部署,且对高分辨率图像和长视频的理解成本过高。为此,我们提出LinMU(线性复杂度多模态理解),一种无需任何二次复杂度模块即可实现语言模型解码器线性复杂度的VLM设计,同时保持全局注意力型VLM的性能。LinMU将语言模型解码器中的每个自注意力层替换为M-MATE块:一个双分支模块,结合双向状态空间模型(Flex-MA分支)以捕捉全局上下文,以及类Swin的局部窗口注意力(Local-Swin分支)以建模邻近相关性。为将预训练VLM转换为LinMU架构,我们提出三阶段蒸馏框架:(i) 用自注意力权重初始化双分支并仅训练Flex-MA分支;(ii) 解冻Local-Swin分支,联合微调两个分支;(iii) 解冻其余模块,使用LoRA适配器微调,同时回归冻结教师模型的隐藏状态与词级别逻辑。在MMMU、TextVQA、LongVideoBench、Video-MME等基准上,LinMU性能与教师模型相当,但在分钟级视频上,首次生成时间(TTFT)减少最多2.7倍,吞吐量提升最多9.0倍。消融实验验证了各蒸馏阶段及双分支结构的重要性。该框架表明,无需二次注意力即可实现顶尖多模态推理,为高分辨率图像与长视频处理开辟新路径。
原文摘要 · Abstract (English)
Modern Vision-Language Models (VLMs) achieve impressive performance but are limited by the quadratic complexity of self-attention, which prevents their deployment on edge devices and makes their understanding of high-resolution images and long-context videos prohibitively expensive. To address this challenge, we introduce LinMU (Linear-complexity Multimodal Understanding), a VLM design that achieves linear complexity for the language model decoder without using any quadratic-complexity modules while maintaining the performance of global-attention-based VLMs. LinMU replaces every self-attention layer in the language model decoder with an M-MATE block: a dual-branch module that combines a bidirectional state-space model for global context (Flex-MA branch) with localized Swin-style window attention (Local-Swin branch) for adjacent correlations. To transform a pre-trained VLM into the LinMU architecture, we propose a three-stage distillation framework that (i) initializes both branches with self-attention weights and trains the Flex-MA branch alone, (ii) unfreezes the Local-Swin branch and fine-tunes it jointly with the Flex-MA branch, and (iii) unfreezes the remaining blocks and fine-tunes them using LoRA adapters, while regressing on hidden states and token-level logits of the frozen VLM teacher. On MMMU, TextVQA, LongVideoBench, Video-MME, and other benchmarks, LinMU matches the performance of teacher models, yet reduces Time-To-First-Token (TTFT) by up to 2.7$\times$ and improves token throughput by up to 9.0$\times$ on minute-length videos. Ablations confirm the importance of each distillation stage and the necessity of the two branches of the M-MATE block. The proposed framework demonstrates that state-of-the-art multimodal reasoning can be achieved without quadratic attention, thus opening up avenues for long-context VLMs that can deal with high-resolution images and long videos.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。