用线性复杂度RNN实现多模态大模型,训练更快效果不差。
ModRWKV: Transformer Multimodality in Linear Time
- 基于RWKV7架构设计轻量级多模态模块,动态适配不同输入
- 在多个数据集上达到SOTA性能,推理速度比Transformer快3倍以上
- 适合需要低延迟部署的多模态应用,如实时对话系统
当前多数多模态研究基于二次复杂度的Transformer架构。尽管RNN等线性模型具有低推理成本优势,但其应用长期局限于纯文本。本文探索现代RNN在多模态场景下的能力,提出ModRWKV——一种基于RWKV7 LLM主干的解耦式多模态框架,通过动态可调的异构模态编码器实现多源信息融合。该框架采用极轻量级模块设计,经大量实验验证出性能与效率的最佳平衡配置。利用RWKV7预训练权重初始化,显著加速多模态训练过程;对比不同预训练检查点的实验进一步证明,该初始化对提升模型理解多模态信号的能力至关重要。实验证明,现代RNN架构可成为多模态大语言模型领域中Transformer的有效替代方案。此外,通过系统探索确定了ModRWKV的最优配置。
原文摘要 · Abstract (English)
Currently, most multimodal studies are based on large language models (LLMs) with quadratic-complexity Transformer architectures. While linear models like RNNs enjoy low inference costs, their application has been largely limited to the text-only modality. This work explores the capabilities of modern RNN architectures in multimodal contexts. We propose ModRWKV-a decoupled multimodal framework built upon the RWKV7 architecture as its LLM backbone-which achieves multi-source information fusion through dynamically adaptable heterogeneous modality encoders. We designed the multimodal modules in ModRWKV with an extremely lightweight architecture and, through extensive experiments, identified a configuration that achieves an optimal balance between performance and computational efficiency. ModRWKV leverages the pretrained weights of the RWKV7 LLM for initialization, which significantly accelerates multimodal training. Comparative experiments with different pretrained checkpoints further demonstrate that such initialization plays a crucial role in enhancing the model's ability to understand multimodal signals. Supported by extensive experiments, we conclude that modern RNN architectures present a viable alternative to Transformers in the domain of multimodal large language models (MLLMs). Furthermore, we identify the optimal configuration of the ModRWKV architecture through systematic exploration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。