AIDB Daily Papers
LLM推論データ選択におけるステップ長の交絡:自然さは本当に品質を意味するのか?
※ 日本語タイトル・ポイントはAIによる自動生成です。正確な内容は原論文をご確認ください。
ポイント
- 大規模言語モデルの推論データセット構築において、自然さに基づくデータ選択がステップ長の長いサンプルを優先する傾向(ステップ長の交絡)を発見しました。
- ステップ長の交絡は、ステップ最初のトークンの低い確率が長いステップで平均化され、全体の確率を高く見せることで発生し、データ品質の評価を歪めます。
- 最初のトークンの確率を除外するASLEC-DROPと、因果推論に基づくASLEC-CASLという2つの手法を提案し、ステップ長の交絡を軽減できることを実験的に示しました。
Abstract
Large reasoning models have recently demonstrated strong performance on complex tasks that require long chain-of-thought reasoning, through supervised fine-tuning on large-scale and high-quality datasets. To construct such datasets, existing pipelines generate long reasoning data from more capable Large Language Models (LLMs) and apply manually heuristic or naturalness-based selection methods to filter high-quality samples. Despite the proven effectiveness of naturalness-based data selection, which ranks data by the average log probability assigned by LLMs, our analysis shows that, when applied to LLM reasoning datasets, it systematically prefers samples with longer reasoning steps (i.e., more tokens per step) rather than higher-quality ones, a phenomenon we term step length confounding. Through quantitative analysis, we attribute this phenomenon to low-probability first tokens in reasoning steps; longer steps dilute their influence, thereby inflating the average log probabilities. To address this issue, we propose two variant methods: ASLEC-DROP, which drops first-token probabilities when computing average log probability, and ASLEC-CASL, which applies a causal debiasing regression to remove the first tokens' confounding effect. Experiments across four LLMs and five evaluation benchmarks demonstrate the effectiveness of our approach in mitigating the step length confounding problem.
Paper AI Chat
この論文のPDF全文を対象にAIに質問できます。
質問の例: