---
title: Repetition-Aware Mixture Scaling Law
url: https://www.emergentmind.com/topics/repetition-aware-mixture-scaling-law
type: topic
---

# Repetition-Aware Mixture Scaling Law

Searching arXiv for the cited papers and closely related work on repetition-aware mixture scaling laws.
A repetition-aware mixture scaling law is a class of scaling-law formulations in which loss depends not only on model size and consumed tokens, but also on how data are mixed across domains, qualities, or components, and on how often limited subsets are replayed. In this literature, repetition is modeled in several mathematically distinct ways: as a target-domain repetition factor such as \(r = h D_\text{total} / D_\text{target}\), as a spectral redundancy index \(\rho_{\mathrm{red}} = 1/\beta\), as a bucketwise repetition factor \(R_d\) inside an information-equivalent token count, or as an additive overfitting penalty on top of a Chinchilla-style base law. The common departure from classical scaling laws is that repeated tokens are not treated as equivalent to fresh tokens, so mixture composition and repetition jointly determine effective data, scaling exponents, and compute-optimal training configurations [2509.20721], [2605.12715], [2605.02364].

## 1. Scope and historical setting

Classical Chinchilla-style laws model final validation loss as
\[
L(N, D) = E + \frac{A}{N^\alpha} + \frac{B}{D^\beta},
\]
with \(N\) the number of parameters and \(D\) the number of training tokens. In the repetition-aware literature, this form is regarded as inadequate whenever high-quality or target-domain data are scarce and must be reused, because it assumes that every training token is unique or, equivalently, that raw token count is a sufficient proxy for usable information [2605.01640].

Mixture-aware scaling work already generalized the Chinchilla form to target-domain loss under a training mixture \(h \in \Delta_k\). One representative formulation predicts
\[
\mathcal{L}(N,D,h)
= E + \left(\sum_{i=1}^k C_i\,h_i^{\gamma_i}\right)^{-1}
+ \frac{A^h}{N^\alpha}
+ \frac{B^h}{D^\beta},
\]
with mixture-dependent coefficients \(A^h\) and \(B^h\), and uses mirror descent on the simplex to derive optimal domain weights for a target domain under a given training budget \((N,D)\) [2507.09404]. That framework is mixture-aware but not repetition-aware: it treats the data term through raw \(D\), assumes i.i.d. draws from \(\mathrm{mix}(h)\), and does not explicitly model repeated or duplicated data.

The explicit repetition-aware turn arises in settings where a finite target corpus is mixed with an effectively unlimited generic corpus, or where a fixed unique-token budget \(U_D\) is traversed for multiple epochs. In the target-plus-generic setting, training uses a target fraction \(h\), total training tokens \(D_\text{total}\), and target pool size \(D_\text{target}\), which induces the repetition factor
\[
r = h \cdot \frac{D_\text{total}}{D_\text{target}}.
\]
In the data-constrained setting, the same structure is written as \(D_\text{total} = U_D(1+R_D)\), where \(R_D = \text{epochs} - 1\) counts repetitions beyond the first pass [2605.12715], [2605.01640]. The central problem is then no longer only compute allocation across \(N\) and \(D\), but allocation across fresh tokens, repeated tokens, and model capacity.

## 2. Spectral redundancy laws and the mixture bottleneck

A theoretical route to repetition-aware mixture laws is given by the redundancy interpretation of scaling laws in kernel ridge regression. In this setting, the covariance or kernel integral operator
\[
T_K f = \mathbb{E}_x[f(x)K(\cdot,x)]
\]
has eigenpairs \((\lambda_i, u_i)\), and the effective dimension is
\[
N_{\mathrm{eff}}(\lambda)
:= \sum_{i=1}^\infty \frac{\lambda_i}{\lambda_i+\lambda}
= \mathrm{Tr}\big((T_K+\lambda I)^{-1}T_K\big).
\]
Under a polynomial spectral tail
\[
\lambda_i \asymp i^{-1/\beta}, \qquad \beta > 1,
\]
and a source condition
\[
\langle f^*,u_i\rangle = \lambda_i^{\,s}\,\langle g,u_i\rangle,
\]
the bias-variance decomposition yields
\[
\mathbb{E}\,\mathcal{E}(f_{\lambda,n})
\lesssim
\lambda^{2s}
+
\frac{\sigma^2}{n} N_{\mathrm{eff}}(\lambda)
\asymp
\lambda^{2s} + \frac{\sigma^2}{n}\lambda^{-1/\beta}.
\]
Optimizing over \(\lambda\) gives
\[
\lambda^\star \asymp n^{-\frac{1}{2s+1/\beta}},
\qquad
\mathbb{E}\,\mathcal{E}(f_{\lambda^\star,n})
\asymp n^{-\alpha},
\qquad
\alpha = \frac{2s}{2s + 1/\beta}.
\]
The redundancy index is \(\rho_{\mathrm{red}} := 1/\beta\), so the exponent can also be written as
\[
\alpha = \frac{2s}{2s+\rho_{\mathrm{red}}}.
\]
This identifies the scaling exponent with redundancy in the representation: a steeper spectrum, corresponding to larger \(\beta\) and smaller \(\rho_{\mathrm{red}}\), yields larger \(\alpha\); a flatter spectrum yields smaller \(\alpha\) [2509.20721].

The mixture result is the key step that turns this redundancy law into a repetition-aware mixture law. If a mixture distribution has component operators
\[
T = \sum_{k=1}^K w_k T_k, \qquad \sum_k w_k = 1,
\]
and each component satisfies
\[
\lambda_i^{(k)} \asymp i^{-1/\beta_k},
\]
then the mixture operator has tail index
\[
\beta = \min_k \beta_k.
\]
Consequently,
\[
\alpha_{\mathrm{mix}}
=
\frac{2s}{2s + 1/\beta_{\mathrm{mix}}}
=
\frac{2s}{2s + 1/\min_k \beta_k}.
\]
The heaviest tail, equivalently the smallest \(\beta_k\), dominates the global exponent. In the paper’s interpretation, smaller \(\beta_k\) corresponds to higher redundancy, so the most repetitive component that has non-zero weight bottlenecks the asymptotic learning curve [2509.20721].

This framework also provides a spectral reading of repetition. A highly repetitive corpus is associated with a flatter covariance tail, hence smaller \(\beta\), larger effective dimension \(N_{\mathrm{eff}}(\lambda) \asymp \lambda^{-1/\beta}\), and slower risk decay. A more diverse or deduplicated corpus is associated with a steeper tail and a larger \(\alpha\). The proposition that “Redundancy reduction improves \(\alpha\)” formalizes this monotonicity: if a transformation produces a steeper tail \(\beta' \ge \beta\), then
\[
\alpha' = \frac{2s}{2s+1/\beta'} \ge \alpha.
\]

## 3. Scarce-target pretraining with generic regularization

The most explicit use of the term “repetition-aware mixture scaling law” appears in a pretraining setting with two data sources: a scarce target dataset of size \(D_\text{target}\) tokens and an effectively unlimited generic dataset. Training consumes \(D_\text{total}\) tokens, draws a target token with probability \(h\), and therefore induces the repetition factor
\[
r = h \cdot \frac{D_\text{total}}{D_\text{target}}.
\]
The empirical program comprises more than 2,000 language-model training runs with GPT-2–style decoder-only Transformers of size 101M, 143M, 192M, 340M, 539M, and 805M non-embedding parameters, spanning multilingual, domain-specific, and quality-filtered mixtures. Across all settings, the paper reports that repetition is a central driver of target-domain performance, that mixture training tolerates much higher repetition than single-source training, and that scarce target corpora can be reused \(15\)–\(20\) times, with the optimal number of repetitions depending on target data size, compute budget, and model scale [2605.12715].

The law begins by defining an effective target data size
\[
D_T = D_\text{target}\bigl(1 + \rho(r)\bigr),
\qquad
\rho(r) = r_1\!\left(1 - e^{-(r-1)/r_1}\right).
\]
For small repetition, \(\rho(r) \approx r-1\), so early passes count almost fully; for large repetition, \(\rho(r) \to r_1\), so the target contribution saturates. Total effective data is then
\[
D_\mathrm{eff}
=
(1-h)\,D_\text{total} + \tau\,D_T,
\]
where \((1-h)D_\text{total}\) is the generic-token count, \(D_T\) is the saturating target contribution, and \(\tau\) weights the relative value of target tokens versus generic tokens for target loss.

At fixed model size, the proposed loss law is
\[
L_{\mathrm{fix}}(D_\text{total}, h, D_\text{target})
=
E + \frac{A}{D_\mathrm{eff}^{\,\alpha}} + \gamma h.
\]
Across model sizes, it is generalized to
\[
L_{\mathrm{size}}(N, D_\text{total}, h, D_\text{target})
=
E + \frac{C}{N^{\beta}} + \frac{B}{N^{\delta} D_\mathrm{eff}^{\,\alpha}} + \gamma h.
\]
The \(\gamma h\) term is a weight penalty, motivated by the empirical observation that high target fractions tend to hurt target loss at fixed \(D_\mathrm{eff}\) because they increase repetition and reduce generic regularization. Parameters are fit by minimizing a weighted Huber loss
\[
\hat\theta
=
\arg\min_\theta
\sum_i \omega_i \,\mathcal{H}\!\bigl(\ell_i - L_\theta(\cdot)\bigr),
\qquad
\omega_i = \max(r_i \cdot h_i, \epsilon),
\quad \epsilon=0.01,
\]
using basin-hopping with 100 random restarts.

The principal empirical claim is that generic data acts as an implicit regularizer. Because only target tokens repeat while generic tokens stay fresh, mixture pretraining can sustain much higher target repetition than single-source training. The paper reports that, for realistic budgets, the best target performance often occurs at \(r \approx 15\)–\(20\), and for some smaller pools even higher, up to \(\sim 30+\) in a multi-domain setting with very small pools. Held-out weighted \(R^2\) values are \(0.95\) for German, \(0.88\) for Maths, \(0.71\) for Quality-filtered mixtures, and \(0.80\) for Wiki/peS2o under the fixed-size law, while the multi-size law gives test \(wR^2 = 0.65\) for German at 539M and \(0.73\) for Maths at 936M. For German mixture optimization, the median wasted token fraction is \(26\%\) for the repetition-aware law, compared with \(88\%\) for the repetition-agnostic baseline [2605.12715].

## 4. Information-based and penalty-based variants

Two later strands modify the token-count axis itself. One retains a base scaling law but adds an overfitting term for repetition; the other replaces raw tokens with an information-equivalent variable.

The mixture-only precursor remains the optimal-data-mixtures law
\[
\mathcal{L}(N,D,h)
=
E + \left(\sum_{i=1}^k C_i\,h_i^{\gamma_i}\right)^{-1}
+ \frac{A^h}{N^\alpha}
+ \frac{B^h}{D^\beta},
\]
where
\[
A^h = \left(\sum_{i=1}^k C_i^A\, h_i\right)^{\gamma^A},
\qquad
B^h = \left(\sum_{i=1}^k C_i^B\, h_i\right)^{\gamma^B}.
\]
This framework is mixture-aware and prescriptive, but it leaves repetition outside the model and therefore treats raw \(D\) as the effective data scale [2507.09404].

InfoLaw instead formulates pretraining as information accumulation with bucketwise repetition. Documents are grouped into quality buckets \(d \in \{0,\dots,5\}\), with mixture weights \(w_d\), source proportions \(B_d\), source tokens \(S\), and training tokens \(K\). For each bucket,
\[
K_d = w_d K, \qquad
S_d = B_d S, \qquad
M_d = \min(K_d, S_d), \qquad
R_d = \frac{K_d}{M_d}.
\]
Quality density is parameterized as
\[
f_d(\theta) = e^{-\theta d},
\]
so total intrinsic information in bucket \(d\) is \(I_d = f_d M_d\). The total learned information is then
\[
\text{info}(w, K, S, f, \lambda(N))
=
\sum_d f_d M_d \log(K)
\left(1 - e^{-\lambda(N) R_d / \log(K)}\right),
\]
and loss follows the power law
\[
L = \alpha \cdot \text{info}^{-\beta}.
\]
The fitted parameters reported in the main experiments are \(\alpha = 3.7373\), \(\beta = 0.0441\), and
\[
\lambda(N; a, b) = a \cdot \ln(N) + b,
\qquad
a^\* = 0.140,\; b^\* = 0.018.
\]
InfoLaw predicts unseen recipes and larger-scale runs up to \(7\)B and \(425\)B tokens with \(0.15\%\) mean and \(0.96\%\) max absolute error in loss, and it extrapolates across overtraining levels [2605.02364].

A distinct line of work starts from a Chinchilla base fit on single-epoch runs and then models repetition as an additive overfitting penalty. Using a unique-token budget \(U_D\), repetitions \(R_D=\text{epochs}-1\), and \(D_\text{total}=U_D(1+R_D)\), the most general proposed form is
\[
L(N, U_D, R_D)
=
E + \frac{A}{N^\alpha}
+ \frac{B}{(U_D(1+R_D))^\beta}
+ P \cdot R_D^{\delta} \cdot \left(\frac{N}{U_D^\gamma}\right)^{\kappa}.
\]
Its simplest one-parameter version is
\[
L(N, U_D, R_D)
=
E + \frac{A}{N^\alpha}
+ \frac{B}{(U_D(1+R_D))^\beta}
+ P \cdot R_D \cdot \frac{N}{U_D}.
\]
The additive term is designed to let validation loss increase with further training tokens in high-repetition regimes, which effective-data saturation laws cannot do. The paper reports that the one-parameter penalty already fits multi-epoch data much better than effective-data baselines, and that strong weight decay \((\lambda = 1.0)\) reduces the one-parameter coefficient \(P\) by approximately \(70\%\), providing a scaling-law explanation for why optimal weight decay in data-constrained regimes can be much larger than standard practice [2605.01640].

## 5. Architectural generalizations: Transformers, MoE, and upcycling

Mixture-of-Experts theory introduces a different notion of repetition-awareness by separating active capacity from routing combinatorics. For a sparse MoE Transformer with hard top-\(k\) routing, the active parameter count is
\[
N_{\mathrm{act}}
=
L_T \Pi_{\mathrm{attn}}
+
L_T k \Pi_{\mathrm{exp}},
\]
and the worst-case generalization bound under a \(d\)-dimensional manifold data model and \(C^\beta\) targets is
\[
\mathbb{E}\| \hat T_n - f\|_{L^2(Q)}^2
\lesssim
N_{\mathrm{act}}^{-2\beta/d}
+
\frac{N_{\mathrm{act}}}{n}
+
\frac{L_T \ell k \log(eM/k)}{n}.
\]
The covering-number decomposition shows a clean separation between an active-capacity term and a MoE-specific routing overhead
\[
R_{\mathrm{route}} := L_T \ell k \log\!\Big(\frac{eM}{k}\Big).
\]
The paper explicitly identifies this routing term as conservative and argues that any improvement beyond the worst-case routing overhead must come from data-dependent routing stability and expert specialization. In repetition-aware reinterpretations, that suggests replacing worst-case \(\log|\Pi|\) by an effective routing entropy based on expert usage frequencies or realized routing patterns [2604.09175].

Upcycling laws introduce repetition-awareness through prior training rather than corpus replay. In MoE upcycling, a dense model pretrained on \(D_1\) tokens is converted into an MoE by replicating dense MLP weights into multiple experts and then further trained for \(D_2\) tokens. The empirical two-stage law is
\[
L(D_1,D_2)
=
A\, D_1^{-\alpha_1}\,
D_2^{-\alpha_2 + \alpha_3 \log D_1}
+ E,
\]
and the joint law for a fixed architecture is
\[
L(D_1, D_2, N_1)
=
A D_1^{-\alpha_1} D_2^{-\alpha_2 + \alpha_3 \log D_1}
+ B N_1^{-\beta_2} + E.
\]
Because \(\alpha_3>0\), the effective exponent for additional upcycling data,
\[
\alpha_{\mathrm{eff}}(D_1) = \alpha_2 - \alpha_3 \log D_1,
\]
decreases as dense sunk cost grows: larger \(D_1\) gives a better head start but slower progress with further \(D_2\). The paper reports that using disjoint subsets for dense pretraining and upcycled training makes little difference, so the interaction is not primarily due to literal sample overlap. It also derives a threshold
\[
D^* \simeq 4\left(\frac{N_1}{10^9}\right)^{-0.7 + 0.04 \log (N_1/10^9)} \text{B tokens},
\]
below which upcycling is more efficient than from-scratch MoE training and above which from-scratch becomes preferable [2502.03009].

## 6. Empirical pathologies, misconceptions, and open questions

A controlled study of exact document repetition shows that repetition damage is not monotone in repeat count. Using Qwen3-style decoder-only transformers of size \(34, 48, 63, 93, 153,\) and \(344\) million parameters, a fixed repeated-token fraction \(f = 0.1\), and a no-repetition reference frontier
\[
L(C) = 2.365 + 6.647 \times 10^{5} \, C^{-0.317},
\]
the paper finds that evaluation loss peaks at an intermediate repeat count \(R\), not at maximal repetition. Peak locations obey
\[
R^{\mathrm{peak}} = 2.31 \times 10^{10} \, N^{-0.96},
\qquad
D_r^{\mathrm{peak}} = 7.58 \times 10^{-10} \, N^{1.84},
\]
and the compute-equivalent loss can be large: for the most damaging repeat count on FineWeb-Edu-Dedup, a \(344\)M-parameter model at \(OT=1\) has \(CEG \approx 0.67\), hence \(CEL \approx 0.33\). This directly contradicts the common intuition that the harmfulness of repetition should simply increase with repeat count [2606.24998].

A second misconception is that single-source heuristics transfer unchanged to mixture pretraining. Earlier single-source work supplied a rule-of-thumb that “up to ~4 repetitions is safe; benefits to ~16; overfitting by ~40,” but the target-plus-generic mixture experiments report a distinct regime in which scarce target corpora can be reused \(15\)–\(20\) times and sometimes \(\sim 30+\), because generic data remains fresh and acts as an implicit regularizer. This means that a repetition-aware mixture law is not merely a correction to single-source repetition laws; it is a different scaling object in which mixture composition changes the tolerated repetition frontier [2605.12715].

The current literature also has pronounced scope conditions. The redundancy-theoretic work assumes clean polynomial tails \(\lambda_i \asymp i^{-1/\beta}\), a source condition, and finite convex mixtures with fixed \(\beta_k\); it leaves non-polynomial tails, explicit duplication models, and practical estimation of \(\beta\) open. The empirical pretraining laws are descriptive rather than mechanistic, are fitted on finite architecture families, and report weaker fits in some regimes, such as quality-filtered mixtures. Open directions therefore include more explicit models of duplication and near-duplicate clusters, data-dependent routing analyses for MoE, joint optimization of data mixture, model size, and learning schedule, and extension to other modalities such as vision or code [2509.20721], [2605.02364], [2605.12715].

Taken together, these works define a coherent research area rather than a single canonical equation. The unifying principle is that repetition, redundancy, and mixture composition alter the effective data axis itself. In theoretical formulations, the decisive quantity is the tail index or redundancy index. In empirical pretraining laws, it is an effective target size, an information-equivalent token count, or an additive overfitting penalty. In architectural variants, it can be the entropy of reused routing patterns or the diminishing marginal value of further training after large sunk cost. A repetition-aware mixture scaling law is therefore best understood as a family of scaling relations that endogenize repeated data, instead of absorbing it into raw token count.

Source: https://www.emergentmind.com/topics/repetition-aware-mixture-scaling-law