モデルのしくみ 中級
推論時スケーリング
Test-time Compute
回答を出す段階(推論時)で、モデルにより多くの計算を使わせて精度を高める考え方です。長く考えさせるほど、難しい問題を解ける確率が上がります。一方で、費用と待ち時間は増えます。
別の呼び方テストタイムコンピュート、推論時計算、推論時の計算量
例推論時スケーリングが効く課題だったので、思考の予算を増やして数学問題の正答率を試した。
詳しく
推論時スケーリング(テストタイムコンピュート)とは、モデルが回答を作る段階で使う計算量を増やすことで、答えの質を上げようとする考え方です。モデルを大きくしたり、学習データを増やしたりする代わりに、「答える前によく考える時間」を増やします。
なぜ注目されたのか
これまでのAIの性能向上は、主に学習時の投資、つまりモデルの大きさ・データ量・学習の計算量を増やすことで進んできました(スケーリング則)。ところが、学習にかかるコストが非常に大きくなるにつれ、別の伸ばし方が注目されます。それが、使うときの計算量を増やす方法です。
人間でも、難しい問題をパッと答えるより、紙に途中式を書いて検算したほうが正確になります。AIでも同じ発想です。
主な方法
- 長く考えさせる:思考の連鎖(chain-of-thought)のように、結論の前に長い途中経過を書かせます。推論モデルは、これを強化学習などで鍛えたものです。
- 複数回答えさせて選ぶ:同じ問題を何度も解かせ、多数決をとったり、検証役のモデルが最も良いものを選んだりします。
- 探索する:途中の案を枝分かれさせ、見込みのない案を捨てながら進めます。
- 自分で見直させる:一度出した答えを検証し、誤りがあれば修正させます。
具体例
数学の証明や、複雑なバグの原因調査では、思考量を増やすと正答率が上がる傾向があります。一方、「この文を敬語に直して」のような単純な作業では、長く考えさせても効果はほとんどなく、待ち時間と費用だけが増えます。
使うときの注意
- 考えた分のトークンが出力として課金されるサービスが多く、費用と応答時間が増えます。
- 効果は課題による差が大きく、考える量を増やし続ければ、いくらでも賢くなるわけではありません。
- 長い思考過程が、必ずしも正しい理由づけを表しているとは限りません。
- 利用側から思考の量(予算)を調整できるサービスもあります。難しい課題だけ多めに割り当てる使い分けが現実的です。
実務での使いどころ
推論時の計算を増やす価値があるのは、「間違えると損失が大きく、人が検算する手間のほうが高くつく」課題です。複雑な計画づくり、コードの設計判断、数学的な推論などが当てはまります。まず軽い設定で試し、足りないときだけ思考量を上げる、という順序にすると、費用を抑えながら効果を確かめられます。
関連する考え方
学習時にかける計算量を扱うのがスケーリング則で、推論時スケーリングは、その「もう1つの軸」と位置づけられます。実際に利用できる形になったものが推論モデルです。
あわせて知りたい言葉
- 推論モデル答えを出す前に内部で段階的に考える過程(思考)を生成してから回答するモデルや機能のことです。数学・コーディング・複雑な分析など、手順の多い問題で精度が上がりやすくなります。
- 思考の連鎖答えを出す前に、途中の考え方を順番に書き出させる方法です。複雑な計算や論理の問題で答えの正確さが上がりやすく、推論モデルの考え方の原型にもなっています。この手法は広く使われています。
- スケーリング則モデルの大きさ、学習データの量、計算量を増やすほど、性能が規則的に向上するという経験則です。大規模言語モデルが巨大化してきた理由を説明する、有力な考え方の一つです。
- 推論(インファレンス)学習を終えた AI モデルに、実際に入力を与えて答えや生成物を出させる段階のことです。チャットで質問したり、画像を生成したりするときに裏で行われている処理を指します。
- トークン単価LLM の API で、処理したトークン数に応じて料金が決まる仕組みと、その単価のことです。入力と出力で単価が分かれ、「100万トークンあたり何ドル」と表示されるのが一般的です。