用多粒度强化学习突破图表转代码的性能瓶颈
Breaking the SFT Plateau: Multimodal Structured Reinforcement Learning for Chart-to-Code Generation
- 设计多粒度奖励机制,融合文本规则与视觉结构相似性反馈
- 在300万真实图表数据上训练,性能较SFT提升6.2%~9.9%
- 适合需要结构化输出生成的研究者和工业界开发者
尽管强化学习在视觉-语言模型的一般推理中表现优异,但在需深度理解信息密集型图像并生成结构化输出的任务中仍研究不足。图表转代码是典型挑战,要求对视觉图表进行复杂推理以生成代码。仅靠监督微调(SFT)往往不足,亟需针对结构化输出的高效强化学习策略。本文通过大规模实验系统分析了SFT的性能瓶颈,提出多模态结构化强化学习(MSRL)用于图表转代码。构建迄今最大的训练语料库,包含300万来自arXiv论文的真实表格-代码对,克服以往合成数据集的局限。尽管已达顶尖水平,但增加SFT数据最终导致收益递减。为此,MSRL采用多粒度奖励机制:文本层面使用规则奖励验证代码细节,视觉层面通过模型生成的渲染结果评估与真实图表的结构相似性。实施两阶段课程训练:先优化文本奖励,再引入视觉信号进一步提升。实验表明,MSRL显著突破SFT瓶颈,在ChartMimic和ReachQA基准上分别提升6.2%和9.9%。该方法在图表领域优于所有现有方案,并达到与先进闭源模型相当的性能。
原文摘要 · Abstract (English)
While reinforcement learning (RL) has proven highly effective for general reasoning in vision-language models, its application to tasks requiring deep understanding of information-rich images and structured output generation remains underexplored. Chart-to-code generation exemplifies this challenge, demanding complex reasoning over visual charts to produce structured code. Supervised fine-tuning (SFT) alone is often insufficient, highlighting the need for effective RL strategies tailored to structured outputs. In this paper, we systematically investigate the performance plateau of SFT through large-scale experiments and propose Multimodal Structured Reinforcement Learning (MSRL) for chart-to-code generation. We construct the largest training corpus to date, with 3 million chart-code pairs curated from real-world tables in arXiv papers, addressing the limitations of previous synthetic datasets. Despite achieving state-of-the-art performance, our experiments show that simply increasing SFT data eventually leads to diminishing improvements. To break this plateau, MSRL employs a multi-granularity reward system that integrates both textual and visual feedback. At the textual level, rule-based rewards validate fine-grained code details, while at the visual level, a model-based reward assesses the structural similarity between rendered code and ground-truth charts. We implement a two-stage curriculum training strategy, first optimizing the model with textual rewards and then incorporating visual signals for further enhancement. Experimental results demonstrate that MSRL substantially breaks the SFT plateau, improving high-level metrics by 6.2% and 9.9% on ChartMimic and ReachQA benchmarks, respectively. Notably, our method outperforms all existing approaches in the chart domain and achieves competitive results with advanced closed-source models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。