HyperAIHyperAI

Command Palette

Search for a command to run...

스케일링: 친칠라의 지수와 카플란의 결합을 만나다

Mathurin Videau Badr Youbi-Idrissi David Lopez-Paz Kartik Ahuja

초록

신경망 스케일링 법칙은 언어 모델 개발의 기초를 이루지만, 표준적인 공식화는 데이터 부족 및 과도한 학습의 극단 영역에서 손실을 체계적으로 과소 또는 과대 평가한다. 이러한 실패는 모델 크기와 학습 데이터가 손실에 독립적으로 영향을 미친다는 근본적인 가정에서 비롯된다. 이 문제를 해결하기 위해, 우리는 단일 상호작용 지수를 통해 모델 용량과 데이터를 결합하는 일반화된 함수 형태인 스케일링 법칙(Skaling law, /skeIlIN/)을 도입한다. 이 간단한 확장은 보간 및 외삽 체제 전반에 걸쳐 평균 절대 백분율 오차(MAPE)를 1.5~3배 감소시킨다. 낮은 연산량 체제로 제한된 희소 그리드 전략과 결합할 때, 스케일링 법칙은 균일 탐색 대비 약 10배 적은 연산량을 사용하여 정확한 전체 그리드 외삽을 달성한다. 소규모 실험으로부터 신뢰할 수 있는 성능 예측을 가능하게 함으로써, 스케일링 법칙은 차세대 모델 학습에서 연산 예산을 할당하기 위한 보다 강건하고 자원 효율적인 프레임워크를 제공한다.

One-sentence Summary

FAIR at Meta propose the Skaling law, a generalized neural scaling formulation that couples model capacity and data through a single interaction exponent, reducing MAPE by 1.5–3× across regimes and enabling accurate full-grid extrapolation from low-compute sparse grids using approximately 10× less compute than uniform sweeps.

Key Contributions

  • The Skaling law introduces a generalized functional form that couples model capacity and training data through a single interaction exponent, correcting the independence assumption in standard additive scaling laws.
  • This single-parameter extension reduces the Mean Absolute Percentage Error (MAPE) by 1.5–3× across both interpolation and extrapolation regimes and outperforms richer nine-parameter formulations at grid boundaries.
  • Pairing the Skaling law with a sparse, L-shaped grid of low-compute boundary runs enables accurate full-grid extrapolation using approximately 10× less compute than uniform sweeps.

Introduction

The predictable scaling of language model performance with model size and training data is essential for making cost-effective decisions about pretraining budgets and architectures. Widely adopted additive scaling laws, such as the Chinchilla formulation, treat the contributions of model size and data volume as independent terms, which forces their cross-derivative to zero and creates biased loss predictions at the boundaries of the training grid where the two are most imbalanced. The authors introduce the Skaling law, a minimal extension that adds a single coupling exponent to capture the interaction between model size and data, correcting boundary errors and enabling accurate scaling predictions from cheap L-shaped profiling grids that require up to 10× less compute.

Method

The authors begin by analyzing the loss surface to determine if model size (NNN) and training data (DDD) interact. They estimate derivatives using a Moving Least Squares (MLS) estimator. While first-order diagnostics suggest near-separability, the mixed derivative 2L/ND\partial^2 L / \partial N \partial D2L/ND serves as the decisive test. An additive law requires this mixed derivative to be zero. However, the estimated mixed derivative is non-zero across the grid, exhibiting a power-law decay.

As shown in the figure below:

This non-zero mixed derivative indicates a synergy where scaling NNN and DDD together lowers loss more than scaling either alone, motivating a coupled functional form. The authors introduce the Skaling form:

L(N,D)=(ANα+BDβ)k+EL(N, D) = \left(\frac{A}{N^{\alpha}} + \frac{B}{D^{\beta}}\right)^k + EL(N,D)=(NαA+DβB)k+E

This form bridges the additive Chinchilla law (recovered when k=1k=1k=1) and the coupled Kaplan form. By using a single outer exponent kkk while preserving independent inner exponents α\alphaα and β\betaβ, Skaling captures the interaction without tying the per-axis decay rates. It remains strictly decreasing and retains a closed-form compute-optimal allocation similar to the additive law, though the optimal ratio differs due to fitted parameters.

To estimate parameters efficiently, the authors avoid dense full-grid sampling, which is dominated by expensive large-scale runs. Instead, they employ an L-shape sampling strategy. This involves sweeping data volume DDD for the smallest models to isolate data parameters, and sweeping model size NNN for the shortest training horizons to isolate size parameters.

For evaluation, a comprehensive cross-validation framework is used. The data is partitioned into validation (interpolation), Extrapolation N (larger models), Extrapolation D (more data), and Far Extrapolation (largest models and data). Performance is measured using Mean Absolute Percentage Error (MAPE) across all sets and the coefficient of determination (R2R^2R2) for interpolation.

To test predictive capability for massive pretraining runs, which typically scale along a fixed token-to-parameter ratio, the authors perform compute extrapolation along iso-ratio slices.

Refer to the framework diagram:

Runs are grouped into slices of constant D/ND/ND/N. Within each slice, the highest-compute points are held out. Scaling laws are fitted on the pooled low-compute points across all slices and evaluated on the held-out high-compute runs. This setup tests if a globally fitted law can predict expensive runs using cheap data, comparing against a per-ratio baseline that fits a one-dimensional power law within each slice.

Experiment

The evaluation uses two pretraining loss grids (Farseer and SK-Grid) and a cross-validation protocol that tests interpolation, single-axis extrapolation, and far extrapolation beyond the training boundaries. The Skaling law consistently reduces boundary errors compared to additive baselines, especially in imbalanced corners and sparse profiling regimes, because its coupled functional form captures the saddle-shaped interaction between model size and data. Compute extrapolation along fixed token-to-parameter ratios further shows that Skaling predicts high-cost runs from low-cost data more accurately and stably than global baselines, while the direction of the optimal allocation trend depends on the dataset and architecture. Overall, the results demonstrate that a multiplicative coupling between model and data scales provides better predictive accuracy and more reliable extrapolation than purely additive formulations.

The Skaling law dramatically reduces extrapolation error compared to the additive Chinchilla law, especially at grid boundaries and in far-extrapolation regimes, while achieving comparable or better interpolation fit. Sparse training on an L-shaped grid preserves Skaling’s accuracy but causes Chinchilla’s error to surge, demonstrating that high interpolation R² alone is insufficient to validate a scaling law. On the full SK-Grid, Skaling lowers far-extrapolation MAPE from 5.17% to 0.70%, and on the L-shape grid from 14.63% to 1.15%. Chinchilla attains strong interpolation R² (0.992–0.995) yet its extrapolation errors are several times larger than Skaling’s, showing that interior fit quality does not guarantee boundary accuracy.

Fitting results show the Skaling law consistently recovers a sub-unit coupling exponent (k ≈ 0.31–0.45) instead of collapsing to the additive Chinchilla case (k=1), and its fitted irreducible loss is systematically lower, nearly vanishing on Farseer data. The coupling exponent and irreducible loss trade off against each other, allowing the coupled form to absorb curvature that the additive law would otherwise represent through a larger floor. This stable coupling explains why Skaling improves boundary predictions while remaining close to Chinchilla when the data are nearly additive. Across all grids the Skaling law fits a coupling exponent k between roughly 0.31 and 0.45, far from the additive value of 1. On Farseer data the fitted irreducible loss E drops from 0.45–0.59 under Chinchilla to 0.03–0.05 under Skaling, while on SK-Grid it remains substantial (1.14–1.75). With k < 1 the concave outer map makes the coupled reducible term decay more slowly at large scale, absorbing curvature that the additive law can only capture through a larger E.

The Skaling law consistently achieves the lowest prediction error across all training regimes when extrapolating to high-compute runs, with a pooled MAPE of 0.60%. Chinchilla’s error hides a severe drop in the optimal regime (3.47%), while the per‑ratio power law is competitive only near the optimum but cannot guide joint parameter–data allocation because it is fit separately within each slice. The more heavily parameterized Farseer law remains less accurate than Skaling overall. Skaling reduces extrapolation error nearly fourfold over Chinchilla and keeps MAPE below 0.9% in every regime, whereas Chinchilla exceeds 3.4% in the optimal band. The per‑ratio power law edges Skaling only in the optimal slice (0.77% vs 0.88%), but it is fit per recipe and cannot inform joint scaling decisions.

The Skaling law substantially reduces extrapolation error relative to the additive Chinchilla law, particularly at grid boundaries and in far-extrapolation regimes, while maintaining comparable interpolation fit. Sparse L-shaped training reveals that high interpolation R² is insufficient for validating a scaling law, as Chinchilla’s error surges under these conditions whereas Skaling remains accurate. Fitting results show that Skaling consistently recovers a sub-unit coupling exponent (k ≈ 0.31–0.45) and a systematically lower irreducible loss, allowing the coupled form to absorb curvature that the additive law would otherwise represent through a larger floor. Across all training regimes, Skaling achieves the lowest prediction error when extrapolating to high-compute runs, reducing error nearly fourfold over Chinchilla and keeping MAPE below 0.9% in every regime.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp