用蒙特卡洛树搜索提升大模型表格推理准确率与效率
TabTracer: Monte Carlo Tree Search for Complex Table Reasoning with Large Language Models
- 通过分步验证和状态回滚,防止错误传播
- 在多个数据集上准确率提升最高6.7%,耗 token 减少59%~84%
- 适合需要高精度、低资源消耗的复杂表格推理任务
大型语言模型在自然语言表格推理中表现突出,现有方法分为提示驱动和代理驱动两类。前者依赖单次程序生成,缺乏步骤验证;后者虽有闭环工具调用,但验证局部、回溯有限,易导致错误累积且成本高。此外,它们通常采用链式或束搜索轨迹,存在组合冗余问题。本文提出 TabTracer,一种基于代理的框架,通过跟踪中间表格状态,实现多步工具调用的协调。首先,引入类型化操作与轻量级数值及格式检查,实现步骤级验证并提供可靠奖励。其次,利用执行反馈的蒙特卡洛树搜索,构建候选状态搜索树,通过反向传播的反思分数指导UCB1选择与版本化快照回滚。第三,采用预算感知剪枝、去重与状态哈希结合单调性门控,显著减少冗余。在TabFact、WikiTQ和CRT数据集上的综合评估显示,TabTracer相比最先进基线最高提升6.7%准确率,同时降低59%至84%的令牌消耗。
原文摘要 · Abstract (English)
Large language models (LLMs) have emerged as powerful tools for natural language table reasoning, where there are two main categories of methods. Prompt-based approaches rely on language-only inference or one-pass program generation without step-level verification. Agent-based approaches use tools in a closed loop, but verification is often local and backtracking is limited, allowing errors to propagate and increasing cost. Moreover, they rely on chain- or beam-style trajectories that are typically combinatorially redundant, leading to high token costs. In this paper, we propose TabTracer, an agentic framework that coordinates multi-step tool calls over intermediate table states, with explicit state tracking for verification and rollback. First, it enforces step-level verification with typed operations and lightweight numeric and format checks to provide reliable rewards and suppress hallucinations. Second, execution-feedback Monte Carlo Tree Search maintains a search tree of candidate table states and uses backpropagated reflection scores to guide UCB1 selection and rollback via versioned snapshots. Third, it reduces redundancy with budget-aware pruning, deduplication, and state hashing with a monotonicity gate to cut token cost. Comprehensive evaluation on TabFact, WikiTQ, and CRT datasets shows that TabTracer outperforms state-of-the-art baselines by up to 6.7% in accuracy while reducing token consumption by 59--84%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。