ChartM$^3$:用于构建图表理解中多维多步视觉推理数据的多阶段代码驱动流水线
ChartM$^3$: A Multi-Stage Code-Driven Pipeline for Constructing Multi-Dimensional and Multi-Step Visual Reasoning Data in Chart Comprehension
AI总结:
针对复杂图表理解任务,本文提出一种自动化多阶段代码驱动流水线,结合RAG与CoT生成高质量视觉推理数据集ChartM$^3$,通过SFT和RL训练显著提升了小模型在复杂图表理解中的推理与泛化能力。
摘要:
复杂的图表理解任务要求多模态大语言模型(MLLMs)具备高级的视觉识别与推理能力。然而,当前研究对现实应用中普遍存在的复杂图表场景和计算密集型推理任务的覆盖十分有限。为解决这些局限性,本研究提出了一种自动化多阶段代码驱动流水线,用于系统性地生成视觉推理数据集。该流水线集成了检索增强生成(RAG)以检索专业图表模板,并采用思维链(CoT)策略生成模拟真实数据分布的推理代码,从而驱动图表渲染和与问题相关的统计计算。通过基于模型的评估,该流水线提升了图表多样性和数据质量。利用该框架,我们构建了ChartM$^3$,这是一个多维多步数据集,包含用于训练的38K张图表和142K个问答对,以及2,871个用于实现实际性能评估的高质量评估样本。监督微调(SFT)和强化学习(RL)实验表明,我们的数据集显著提升了推理能力和跨领域泛化性能,使较小模型在复杂图表理解中能够达到与较大规模模型相当的性能。
英文摘要:
Complex chart understanding tasks demand advanced visual recognition and reasoning capabilities from multimodal large language models (MLLMs). However, current research provides limited coverage of complex chart scenarios and computation-intensive reasoning tasks prevalent in real-world applications. This study proposes an automated multi-stage code-driven pipeline for systematically generating visual reasoning datasets to address these limitations. The pipeline integrates retrieval-augmented generation (RAG) to retrieve professional chart templates and employs chain-of-thought (CoT) strategies to generate reasoning codes that simulate real data distributions, thereby driving chart rendering and question-related statistical computations. Through model-based evaluation, the pipeline enhances chart diversity and data quality. Using this framework, we construct ChartM$^3$, a multi-dimensional and multi-step dataset containing 38K charts and 142K Q&A pairs for training, along with 2,871 high-quality evaluation samples for enabling practical performance assessment. Supervised fine-tuning (SFT) and reinforcement learning (RL) experiments demonstrate that our dataset significantly improves reasoning capabilities and cross-domain generalization performance, enabling smaller models to achieve performance comparable to larger-scale models in complex chart comprehension.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。