用强化学习提升大模型处理复杂表格推理的能力
TableGPT-R1: Advancing Tabular Reasoning Through Reinforcement Learning
- 通过强化学习构建可闭环执行代码的智能体轨迹数据
- 在多个基准上超越现有模型,推理准确率显著提升
- 适合需要精准表格分析与多步推理的研究者使用
表格数据是现代数据分析和科研的核心。尽管经过监督微调的大型语言模型在自然语言交互方面已有显著进步,但在处理复杂多步推理与鲁棒代码执行时仍显不足。强化学习虽具潜力,却面临三大挑战:高质量智能体轨迹稀缺、反馈信号差异巨大(从严格SQL执行到开放数据解读),以及垂直专业化导致通用知识遗忘。为此,我们提出基于系统化强化学习框架的TableGPT-R1模型。其整合了分层难度的数据工程管道、任务自适应奖励机制(结合规则验证与注入式奖励模型)及过程级奖励设计,并采用多阶段训练流程,在稳定推理能力后逐步专精于表格任务。大量实验表明,TableGPT-R1在权威基准上达到当前最佳表现,显著优于基线模型,同时保持强泛化能力。
原文摘要 · Abstract (English)
Tabular data serves as the backbone of modern data analysis and scientific research. While Large Language Models (LLMs) fine-tuned via Supervised Fine-Tuning (SFT) have significantly improved natural language interaction with such structured data, they often fall short in handling the complex, multi-step reasoning and robust code execution required for real-world table tasks. Reinforcement Learning (RL) offers a promising avenue to enhance these capabilities, yet its application in the tabular domain faces three critical hurdles: the scarcity of high-quality agentic trajectories with closed-loop code execution and environment feedback on diverse table structures, the extreme heterogeneity of feedback signals ranging from rigid SQL execution to open-ended data interpretation, and the risk of catastrophic forgetting of general knowledge during vertical specialization. To overcome these challenges and unlock advanced reasoning on complex tables, we introduce \textbf{TableGPT-R1}, a specialized tabular model built on a systematic RL framework. Our approach integrates a comprehensive data engineering pipeline that synthesizes difficulty-stratified agentic trajectories for both supervised alignment and RL rollouts, a task-adaptive reward system that combines rule-based verification with a criteria-injected reward model and incorporates process-level step reward shaping with behavioral regularization, and a multi-stage training framework that progressively stabilizes reasoning before specializing in table-specific tasks. Extensive evaluations demonstrate that TableGPT-R1 achieves state-of-the-art performance on authoritative benchmarks, significantly outperforming baseline models while retaining robust general capabilities. Our model is available at https://huggingface.co/tablegpt/TableGPT-R1.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。