用两阶段强化学习让30亿参数多模态模型具备强推理能力
LMM-R1: Empowering 3B LMMs with Strong Reasoning Abilities Through Two-Stage Rule-Based RL
- 先用纯文本数据增强基础推理,再迁移到多模态任务
- 在足球复杂任务上提升3.63%,多模态与纯文本基准均优于基线
- 无需大量多模态标注数据,适合资源受限的轻量级模型优化
大型多模态模型(LMMs)的推理能力提升面临视觉感知与逻辑推理耦合的挑战,尤其在30亿参数等紧凑架构中,结构限制导致推理能力不足和模态对齐困难。尽管基于规则的强化学习(RL)在纯文本领域表现优异,其多模态扩展却受两大障碍制约:(1)因答案模糊和复杂推理样本稀缺导致的数据不足;(2)多模态预训练引发的基础推理能力退化。为此,我们提出LMM-R1,一种两阶段框架,通过 extbf{基础推理增强(FRE)}和 extbf{多模态泛化训练(MGT)}实现多模态推理。FRE阶段利用纯文本数据进行规则化强化学习以强化推理能力,随后在MGT阶段将该能力推广至多模态场景。在Qwen2.5-VL-Instruct-3B上的实验表明,LMM-R1在多模态与纯文本基准上分别较基线提升4.83%和4.5%,在复杂足球游戏任务中提升3.63%。结果验证了基于文本的推理增强可有效实现多模态泛化,提供了一种高效且节省成本的训练范式,避免依赖昂贵的高质量多模态数据。
原文摘要 · Abstract (English)
Enhancing reasoning in Large Multimodal Models (LMMs) faces unique challenges from the complex interplay between visual perception and logical reasoning, particularly in compact 3B-parameter architectures where architectural constraints limit reasoning capacity and modality alignment. While rule-based reinforcement learning (RL) excels in text-only domains, its multimodal extension confronts two critical barriers: (1) data limitations due to ambiguous answers and scarce complex reasoning examples, and (2) degraded foundational reasoning induced by multimodal pretraining. To address these challenges, we propose \textbf{LMM-R1}, a two-stage framework adapting rule-based RL for multimodal reasoning through \textbf{Foundational Reasoning Enhancement (FRE)} followed by \textbf{Multimodal Generalization Training (MGT)}. The FRE stage first strengthens reasoning abilities using text-only data with rule-based RL, then the MGT stage generalizes these reasoning capabilities to multimodal domains. Experiments on Qwen2.5-VL-Instruct-3B demonstrate that LMM-R1 achieves 4.83\% and 4.5\% average improvements over baselines in multimodal and text-only benchmarks, respectively, with a 3.63\% gain in complex Football Game tasks. These results validate that text-based reasoning enhancement enables effective multimodal generalization, offering a data-efficient paradigm that bypasses costly high-quality multimodal training data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。