arXiv:2605.11262cs.LG2026-05

通过隐式思维链提升结构化数据模型的推理能力

Latent Chain-of-Thought Improves Structured-Data Transformers

  • 用隐藏状态生成反馈令牌,实现多轮隐式推理
  • 在36个数据集上平均提升12.63%(时间序列)和3.25%(表格数据)
  • 适用于小模型增强,尤其适合资源受限场景

思维链(Chain-of-thought)及测试时计算扩展已被证明能显著提升语言模型的表达与推理能力。受此启发,本文探索了针对时序与表格数据的隐式思维链机制,以及深度与循环对性能的影响。提出一种递归方案:结构化数据Transformer在首次前向传播后,将查询位置的隐藏状态压缩为反馈令牌,附加至输入重新处理,实现多轮隐式计算。在36个时序预测与表格预测数据集上对比实验表明,该方法在9个时序数据集中的7个、27个表格数据集中的23个上优于基线模型,平均提升分别为12.63%和3.25%。同时,该方法在预训练基础模型上也有效:将隐式思维链应用于小型开源表格模型nanoTabPFN,其性能超过更大的TabPFN-v2。结果表明,思维链是扩展结构化数据模型测试时计算的有效途径。

原文摘要 · Abstract (English)

Chain-of-thought and more broadly test-time compute are known to augment the expressive capabilities of language models and have led to major innovations in reasoning. Motivated by this success, this paper explores latent chain-of-thought as well as the impact of depth and looping for time-series and tabular data. We propose a recurrent scheme in which a structured-data transformer, after an initial forward pass, compresses its query-position hidden states into feedback tokens that are appended to the input and processed again, allowing multiple rounds of latent computation before prediction. We compare CoT models against a same-depth no-CoT baseline, a deeper baseline matched to the CoT model in effective depth, and a looped transformer with weight-tied recurrence but no additional chain-of-thought tokens. Across 36 datasets in time-series forecasting and tabular prediction, latent chain-of-thought improves over the baseline on 7/9 time-series datasets (+12.63\% average gain) and 23/27 tabular datasets (+3.25\% average gain), with CoT models performing best on average in both settings. We also show that the benefit of CoT extends to pretrained foundation models: applying latent CoT to nanoTabPFN, a small open-source tabular foundation model, improves its performance above the much larger TabPFN-v2 on TabArena. Together, these results demonstrate that chain-of-thought is a useful axis for scaling test-time compute for structured data.

结构化数据思维链Transformer推理增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。