arXiv:2502.18435cs.CLcs.IT2025-02被引 1

反向生成能提升大模型在选择题上的表现,原因与推理方向有关。

What Makes the Preferred Thinking Direction for LLMs in Multiple-choice Questions?

  • 用从右到左的方式训练模型,替代传统的从左到右生成
  • 在逻辑推理、常识理解等任务上,反向模型性能显著更优
  • 适合研究语言模型推理机制或想提升问答能力的读者

语言模型通常采用从左到右(L2R)的自回归因子分解。然而,这种分解方式未必是最优归纳偏置。本文探究了从右到左(R2L)训练作为替代方案的可能性,聚焦于多项选择题(MCQ)这一知识提取与推理任务。在多种模型规模(2B-8B参数)和训练数据集上进行的广泛实验表明,R2L模型在多个MCQ基准测试中显著优于L2R模型,涵盖逻辑推理、常识理解与真实性评估任务。分析发现,性能差异可能与校准性、可计算性及方向性条件熵等多种因素相关。通过使用算术任务的受控模拟实验,这些因素得以更好解耦。本研究证明,探索文本分布的替代因子分解方式可提升大模型能力,并为逼近人类语言分布的最优因子化提供理论洞见,揭示不同推理顺序在何种情境下更具优势。代码与模型检查点已开源。

原文摘要 · Abstract (English)

Language models usually use left-to-right (L2R) autoregressive factorization. However, L2R factorization may not always be the best inductive bias. Therefore, we investigate whether alternative factorizations of the text distribution could be beneficial in some tasks. We investigate right-to-left (R2L) training as a compelling alternative, focusing on multiple-choice questions (MCQs) as a test bed for knowledge extraction and reasoning. Through extensive experiments across various model sizes (2B-8B parameters) and training datasets, we find that R2L models can significantly outperform L2R models on several MCQ benchmarks, including logical reasoning, commonsense understanding, and truthfulness assessment tasks. Our analysis reveals that this performance difference may be fundamentally linked to multiple factors including calibration, computability, and directional conditional entropy. We analyze the impact of these factors through controlled simulation studies using arithmetic tasks, where the impacting factors can be better disentangled. Our work demonstrates that exploring alternative factorizations of the text distribution can lead to improvements in LLM capabilities and provides theoretical insights into optimal factorization towards approximating human language distribution, and when each reasoning order might be more advantageous. Our code and checkpoints are released at https://github.com/apple/ml-reversal-blessing.

大模型推理语言模型生成方向

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