3B轻量模型通过两阶段优化实现多模态推理高效高准
Towards Efficient Multimodal Unified Reasoning Model via Model Merging
- 用强化学习动态优化响应长度与质量,生成更短更准结果
- 在10个基准上超越同类小模型,推理速度提升显著
- 适合资源受限场景下需要多任务统一推理的用户
尽管多模态大语言模型在各类任务中表现卓越,但面临推理效率低、模型过大和过度思考等问题。现有轻量级模型难以在小规模下兼顾效率与性能。为此,我们提出Tiny-R1V,一种新型3B参数轻量模型,通过两阶段优化实现更快推理和更高准确率,同时统一多任务多模态推理,减少推理标记数。第一阶段引入长度感知相对策略优化(LIPO),一种新强化学习方法,训练数学推理、图表推理和OCR等专用模型;LIPO通过优先选择简洁高质量响应,动态调整组内响应优势,促进生成更短更准确内容。第二阶段提出无需训练的自适应模型融合(AMM)方法,将多个专家模型融合为统一架构;AMM通过新型梯度投影正则化损失函数,自适应调整任务向量权重,缓解任务间冗余冲突。在涵盖数学、结构化数据(图表、表格、文档)、OCR及通用能力的十个主流推理基准上的广泛评估显示,Tiny-R1V表现出色,使轻量模型在多种多模态推理任务中达到领先水平。代码将公开于 https://github.com/buptyqx/Tiny-R1V
原文摘要 · Abstract (English)
Although Multimodal Large Language Models (MLLMs) have demonstrated remarkable capabilities across diverse tasks, they encounter challenges in terms of reasoning efficiency, large model size and overthinking. However, existing lightweight MLLMs lack the capability to balance high efficiency and performance at a small scale. To this end, we propose Tiny-R1V, a novel lightweight 3B model that achieves faster inference and higher accuracy via a two-stage optimization, while unifying multimodal reasoning across multiple tasks with fewer inference tokens. In the first stage, Tiny-R1V introduces Length-Informed Relative Policy Optimization (LIPO), a new reinforcement learning method, to train each reasoning model, including mathematical reasoning, chart reasoning, and OCR capability. The LIPO dynamically adjusts the advantages of responses within groups by prioritizing concise yet high-quality responses to encourage the generation of shorter and more accurate responses. In the second stage, we propose Adaptive Model Merging (AMM), a training-free model merging method that merges multiple specialist models into a unified architecture. Specifically, AMM adaptively adjusts the weights of task vectors via a novel gradient projection regularization loss function, thus mitigating redundant conflicts between them. Extensive evaluations on ten widely-used reasoning benchmarks covering mathematics, structured data (charts, tables, documents), OCR, and general capabilities showcase the superior performance of Tiny-R1V, enabling lightweight models to excel in diverse multimodal reasoning tasks. Code will be available at \href{https://github.com/buptyqx/Tiny-R1V}{https://github.com/buptyqx/Tiny-R1V}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。