AIのスケーリング則とは何ですか?
スケーリング則は、パラメータ、学習データ、計算量を増やすと、言語モデルの損失が予測どおりに下がることを示すパターンです。決まった計算量の予算をどう分けるかも教えてくれます。2022年の有名な研究では、モデルとデータをいっしょに大きくして、1パラメータあたり約20トークンで学習させると、いちばんよい結果になることがわかりました。
▶ やってみようなぜ大切なのか
大きなモデルの学習には、チップと電気代でばく大なお金がかかるので、研究所はスケーリング則で前もって計画します。意外なことに、予算が決まっていれば、いちばん大きなモデルがいちばんよいとはかぎりません。
6分後にできるようになること
- モデルの質は、大きくできる3つの材料で決まります。データ、パラメータ、計算量です。
- 予算が決まっていれば、データが少なすぎる巨大モデルは、バランスのとれたモデルに負けます。
- スケーリング則があれば、小さく安い実験から大きな学習の結果を予想できます。
練習する内容
大きくなる3つのもの
Vireoの学習計画の各行を、どの材料を増やすものか仕分けましょう。
- 15兆トークンの文章 → データ
- 層を広く、しかも数を多く → パラメータ
- 何か月も動き続ける何千ものチップ → 計算量
- 700億個の重み → パラメータ
- 本、コード、Webページを追加する → データ
- 過去最高の電気代 → 計算量
計算量の予算を使おう
スライダーを動かして、予算をモデルの大きさと学習データに分けましょう。損失がいちばん低いところを見つけたら、決定します。
予算は同じで、使い方は2つ。大きなモデルにするか、学習させる文章を増やすか。スライドしていろいろな分け方を試し、予測される損失を見ましょう。低いほどよいです。
- 予算 1倍: ここでのいちばんよい分け方:10億パラメータを200億トークンで。
- 予算 10倍: ここでのいちばんよい分け方:32億パラメータを630億トークンで。
- 予算 100倍: ここでのいちばんよい分け方:100億パラメータを2000億トークンで。
≈ 1パラメータあたり20トークン 説明のための例:Chinchilla型の式で、実際の測定値ではありません
Vireoの次の学習を計画しよう
予算は同じで、計画が2つ。よいモデルになりそうなのはどっち?
- 4分の1の大きさのモデルを、4倍の文章で学習させる (答え)
- 予算いっぱいの最大のモデルを、残った文章で学習させる
Vireo Labsは、小さなモデルを1パラメータあたり20トークンよりずっと多くで学習させます。なぜかしこいやり方かもしれない?
- 小さなモデルは何百万人に使ってもらうとき動かすのが安いので、学習を増やす価値がある (答え)
- まちがい。1パラメータあたり20トークンより多くても役に立たない
何か月分もの計算量を使う前に、スケーリング則で研究所は何ができる?
- 小さなテストの結果から、大きなモデルがだいたいどれくらいよくなるか予測する (答え)
- モデルがどんな能力を持つか、正確に保証する
覚えておこう
計算量の予算が決まっているなら、モデルとデータをいっしょに大きくします。昔からの目安は、1パラメータあたり約20トークンです。
予算が決まっているなら、モデルとデータをいっしょに大きくします。目安は1パラメータあたり約20トークンです。
押さえておきたい事実
- 言語モデルの損失は、モデルの大きさ、データ、計算量が増えるにつれて、べき乗則に従ってなめらかに予測どおり下がります。 · 出典: Kaplan et al., Scaling Laws for Neural Language Models, OpenAI, 2020
- 計算量の予算が決まっているとき、いちばんよい結果になったのは、モデルの大きさと学習トークンを同じ割合で増やす、1パラメータあたり約20トークンのときでした。 · 出典: Hoffmann et al., Training Compute-Optimal Large Language Models, DeepMind, 2022
- 1.4兆トークンで学習した700億パラメータのChinchillaは、同じ計算量の予算を使った2800億パラメータのGopherより高い性能を出しました。 · 出典: Hoffmann et al., DeepMind, 2022
- 1750億パラメータのGPT-3は、約3000億トークンで学習しました。1パラメータあたり2トークン未満です。 · 出典: Brown et al., Language Models are Few-Shot Learners, OpenAI, 2020
- Llama 3は15兆トークン以上で事前学習され、いちばん大きなモデルでも1パラメータあたり20トークンをはるかに上回りました。 · 出典: Meta, The Llama 3 Herd of Models, 2024
更新日:
よくある質問
Chinchillaのスケーリング則とは何ですか?
2022年の研究成果で、学習に使う計算量が決まっているなら、モデルの大きさと学習データをいっしょに増やすべきだというものです。そこから生まれた目安が、1パラメータあたり約20トークンです。
AIモデルは大きいほど、いつも性能が高いのですか?
同じ予算ならそうとはかぎりません。大きなモデルが役立つのは、十分な学習データももらえたときだけです。より多くの文章で学習した小さなモデルが、学習不足の大きなモデルに勝つこともあります。
AIの学習でいう「計算量」とは何ですか?
モデルの学習に使う計算の合計です。専用のチップ、それを動かす時間、使う電気で支払います。
研究所はなぜ小さなモデルをそんなに多くのデータで学習させるのですか?
小さなモデルは、利用者1人ひとりに動かすのが安くて速いからです。計算量の面で最適な量よりずっと多くのデータで学習させると、質が少し上がり、それが何百万回もの利用で元がとれます。
関連用語
