- 大規模言語モデルにおける **masked diffusion model** と **autoregressive (AR) model** を、データ制約下(限られたデータを繰り返し学習する設定)で体系的に比較した研究。
- 計算資源が豊富でデータが乏しい状況では、diffusion model が AR model を大きく上回ることを示す。
- diffusion model の新しい **scaling law** を提示し、diffusion が AR を逆転する臨界計算量の閉形式表現を導出。
- 優位性の理由として、ランダムマスキング目的が多様な token ordering 上の分布を暗黙的に学習する **implicit data augmentation** 効果を挙げる。
2. 先行研究と比べてどこがすごい?
- AR model が長く LLM の主流であり、diffusion-based language model は代替候補として登場したが、AR に対する優位性は十分に検討されていなかった。
- 本研究はデータ制約下という特定の regime に焦点を当て、diffusion が AR を上回る条件を体系的に示した点が新しい。
- 単なる性能比較にとどまらず、diffusion model の **scaling law** と AR を逆転する臨界計算量の閉形式を導出した点が先行研究と異なる。
- 性能差の機序を **implicit data augmentation** として説明した点も新規性として挙げられている。
3. 技術・手法の肝は?
- **masked diffusion model** を対象に、限られたデータを繰り返し学習するデータ制約設定で AR model と比較。
- 計算量とデータ量を変化させて検証し、diffusion model の **scaling law** を新たに定式化。
- その scaling law から、diffusion が AR を上回り始める臨界計算量の閉形式表現を導出。
- 優位性の機序として、ランダムマスキング目的が多様な token ordering 上の分布を暗黙的に学習し、AR の固定された left-to-right factorization に欠ける **implicit data augmentation** として働くと説明。
4. どうやって有効だと検証した?
- データ制約設定で繰り返し学習を行い、diffusion model と AR model の validation loss および downstream performance を比較。
- diffusion model が繰り返しデータをより有効に活用し、より低い validation loss と優れた downstream performance を達成することを示した。
- 計算量・データ量を変えた実験から **scaling law** を導出し、臨界計算量の閉形式表現の妥当性を検証。
- コードは https://diffusion-scaling.github.io で公開。
5. 議論はある?
- データがボトルネックで計算資源が豊富な場合、diffusion model が標準的な AR paradigm の有力な代替になり得ると主張。
- 優位性の説明として **implicit data augmentation** 効果を提示しているが、その詳細な分析範囲は要旨からは不明。
- 臨界計算量の閾値の一般性や、他の diffusion 手法・データ分布への拡張性については要旨からは不明。
- 実運用規模での検証や限界については要旨からは不明。
Autoregressive (AR) models have long dominated the landscape of large language models, driving progress across a wide range of tasks. Recently, diffusion-based language models have emerged as a promising alternative, though their advantages over AR models remain underexplored. In this paper, we systematically study masked diffusion models in data-constrained settings where training involves repeated passes over limited data and find that they significantly outperform AR models when compute is abundant but data is scarce. Diffusion models make better use of repeated data, achieving lower validation loss and superior downstream performance. We find new scaling laws for diffusion models and derive a closed-form expression for the critical compute threshold at which diffusion begins to outperform AR. Finally, we explain why diffusion models excel in this regime: their randomized masking objective implicitly trains over a rich distribution of token orderings, acting as an implicit data augmentation that AR's fixed left-to-right factorization lacks. Our results suggest that when data, not compute, is the bottleneck, diffusion models offer a compelling alternative to the standard AR paradigm. Our code is available at: https://diffusion-scaling.github.io.