Tier 1: 世界のルールを不可逆に変えた原典スケーリング則HF 185 votes
スケーリング則大規模言語モデル事前学習

ニューラル言語モデルにおけるスケーリング則:計算量・データ・パラメータのべき乗則

モデル規模・データ量・計算量に対し4桁以上のスケールで滑らかなべき乗則に従うことを立証し、巨額コンピュート投資の科学的根拠を確立

Scaling Laws for Neural Language Models

論文の書誌情報と関連リンク

読み方の2つの軸

現在:しくみ × 全体像

語彙 使うことば・数式・例え方

直感数式なし
しくみ基礎的な数式
原論文原論文の表現

深さ 研究のどこまで読むか

核心肝だけ
全体像背景から評価
読み解く前提から限界

この2本は、本文下のスライダーからいつでも変更できます。

概要

この研究は、AIの大きさ、学習データ、計算量を増やすと、予測の誤りがどのような規則で減るかを数多くの実験から調べました。

著者らは、大きさや形の異なる言語モデルを学習し、モデル規模・データ量・計算量と予測誤差の関係を測りました。一定の範囲では、誤差が滑らかな規則に沿って下がると報告しています。

この論文は、モデル、データ、計算の各量と学習後の誤差を結ぶ経験的な規則を示します。同時に、後の研究でデータ配分が修正されたことや、個別課題の能力を誤差だけでは説明できない点も重要です。

Kaplanらは、自己回帰言語モデルのlossがparameters、dataset size、training computeに対してpower lawを示すことを系統的に調べました。

本研究はTransformer decoder群を学習し、テストlossをモデル規模、token数、FLOPsへ個別にfitします。形状依存性とcompute-optimalな早期停止も分析対象です。

本論文は、個別・結合scaling law、compute allocation、sample efficiency、critical batch sizeを扱います。観測範囲内のfitと外挿、下流能力、後のChinchillaによる配分修正を区別して読みます。

本論文はautoregressive language modeling lossと(N)、(D)、(C)の経験的power-law relationを報告します。

著者らはnon-embedding parameter count (N)、dataset tokens (D)、training compute (C)に対するcross-entropy lossをfitし、architecture shapeの影響とcompute-efficient frontierを推定します。

本論文の中心はjoint loss model、optimal allocation、early stopping、gradient noise scaleです。推定指数の適用範囲、downstream metricとのずれ、後続研究で修正されたtoken allocationを検討します。

Jared KaplanJohns Hopkins University / OpenAI / Anthropic
Dario AmodeiOpenAI / Anthropic
著者をもっと詳しく知る

論文と確認可能な公式情報に基づき、著者の所属と研究背景を掲載しています。

所属
: Johns Hopkins University / OpenAI / Anthropic
学歴
: Harvard 博士(理論物理学)、スタンフォード大学 学士
経歴
: Johns Hopkins University、OpenAI、Anthropic 共同創業者
研究の系譜
: 素粒子物理学・場の量子論出身
代表的な論文
: Scaling Laws for Autoregressive Generative Modeling、Claude
関連情報
: 物理学の統計力学的アプローチを深層学習のスケーリング解析に持ち込みAnthropicを共同創業
所属
: OpenAI / Anthropic
学歴
: Princeton University 博士(生物物理学)、Stanford Postdoc
経歴
: Baidu、Google Brain、OpenAI VP of Research、Anthropic CEO
研究の系譜
: Andrew Ng(Baidu時代の共同研究者)
代表的な論文
: Deep Speech 2、Concrete Problems in AI Safety、GPT-2、GPT-3
関連情報
: OpenAIの研究担当副社長としてGPT-2/GPT-3を指揮、後にAnthropicを設立

なぜ歴史的イノベーションなのか

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

モデル規模・データ量・計算量と精度の普遍的べき乗則を実証した巨大モデル投資の理論的支柱

コミュニティの評価・歴史的インパクト(1件)
  • モデル規模・データ量・計算量に対し4桁以上のスケールで滑らかなべき乗則に従うことを立証し、巨額コンピュート投資の科学的根拠を確立

    原文を見る ↗
この読み方に出てくる言葉(6語)
推論

学習済みモデルから出力を生成する処理。

スケーリング則

モデル・データ・計算量と損失の関係を表す経験則。

パラメータ

モデルがデータから学習する重み。

ゼロショット

対象タスクの例示や追加学習なしで解く設定。

損失

予測と目標のずれを表す最適化対象。

Transformer

注意機構を中心に系列を処理するニューラルネットワーク。

どんな問いに向き合ったか

モデルを大きくしたとき、損失はどの程度予測可能に下がるのか。著者らは、パラメータ数 NN、学習トークン数 DD、計算量 CC を分けて変え、それぞれとテスト損失 LL の関係を調べました。

従来の方法と課題

従来は、特定規模のモデルごとに深さ、幅、ヘッド数などを試行錯誤する設計が中心でした。この研究は個別構成の比較から一歩引き、総パラメータ数、データ量、計算量という三つの尺度で損失を予測できるかを問いました。

肝のアイデア

中心的な発見は、ほかの要因がボトルネックでない範囲では、損失が各尺度のべき乗に沿って低下したことです。また、総パラメータ数が同じなら、深さや幅といったモデル形状の違いが損失へ与える影響は比較的小さいと報告されました。

どういうしくみか

著者らは規模と形状の異なるTransformerデコーダを学習し、得られたテスト損失を対数軸で比較しました。その観測値へべき乗曲線を当てはめ、モデル、データ、計算の各軸について指数を推定しました。計算量は概ねパラメータ数と学習トークン数の積として扱います。

どう確かめたか

評価の中心は、規模を数桁にわたって変えたときのクロスエントロピー損失です。著者らは観測点が対数軸上でほぼ直線に並ぶか、モデル形状を変えても関係が保たれるか、学習をどこで止めると計算効率がよいかを比較しました。

何が分かったか

著者らは、対象範囲で損失のべき乗関係が滑らかに続き、モデル形状より総パラメータ数の影響が大きいと報告しました。下流のゼロショット評価にも改善傾向は見られましたが、個別能力が損失と同じ滑らかさで伸びることまで保証する結果ではありません。

どこまで使えるか

この結果が直接扱うのは、自己回帰言語モデルの事前学習損失と、論文内で測ったゼロショット課題です。固定した計算予算の配分や学習停止点を考える手がかりにはなりますが、推論時の費用や特定用途の正解率を直接最適化する式ではありません。

限界と未解決の問い

・コサインスケジューラの学習率リセット効果を考慮しておらず、データ量を低く見積もっていた点

・推論時のスループットやメモリ要件を考慮していない点

この研究から考える

ここからは、論文の結果を踏まえた編集上の考察です。

著者らは言語モデルの損失がモデル・データ・計算量に対して滑らかなべき乗則を示す範囲を観測しました。この観察が自分のデータと計算条件でも保たれるなら、次の学習実験の規模と予算を決める際に、経験的な予測曲線を使う価値があります。データ配分は後続研究で修正され、下流能力や推論費用を同じ式から直接は予測できませんので、同じbaselineとmetricでの比較が前提です。