AIDB Daily Papers
LLMは出力の残り長さを線形にエンコードしているか
※ 日本語タイトル・ポイントはAIによる自動生成です。正確な内容は原論文をご確認ください。
ポイント
- LLMの隠れ状態から出力の残り長さを予測する線形プローブを構築し、モデル内部の情報を分析した。
- 出力開始前のプロンプトから総トークン数が予測可能であり、モデルが生成計画を内部保持していることを示した。
- モデルが生成を修正する際に予測値が変動することから、単なる位置情報以上の動的な長さ推定を行っていることが判明した。
Abstract
Large language models generate one token at a time, yet their responses show remarkably consistent length structure: step-by-step solutions converge in predictable token counts, retrievals stop after a few sentences, retractions extend responses by measurable amounts. We ask whether the model carries an internal estimate of how much response remains. Training minimal-capacity linear probes on frozen hidden states of three open-weight 7-8B models across seven completion-style datasets, we find three converging pieces of evidence. First, total response length is linearly decodable from the prompt's last hidden state alone, before any output is emitted. Second, probe directions trained on natural-language datasets transfer broadly, including to controlled synthetic completions never seen in training, outperforming a statistical baseline; the converse direction generally fails, and this asymmetry is itself informative. Third, on curated high-loss completions, the probe's per-position estimate shifts upward at the moment the model retracts and restarts a partial solution, a directional behavior no position-only predictor can reproduce (qualitative, not aggregate). We frame this as approximate estimation of remaining generation length, distinct from exact-counting impossibility results for transformers, and interpret it as evidence that LLMs maintain a plan-like internal representation of output length (decodable, not necessarily used causally).
Paper AI Chat
この論文のPDF全文を対象にAIに質問できます。
質問の例: