WorksComputer Science
Approximate Solomonoff Induction: On the Robustness of Learned Universal Predictors
Abstract
Why large language models trained via log-loss minimization over massive and diverse datasets exhibit broad capabilities and generalizable skills is not yet fully understood. Grau-Moya et al. (ICML 2024) offer a conceptual explanation through the lens of meta-learning: if sequence-generating environments are drawn from a prior , minimizing population log-loss recovers the corresponding Bayesian mixture predictor . In the idealized limit in which is a universal prior, the optimal predictor corresponds to Solomonoff universal induction. Practical machine learning algorithms, however, generally produce only approximately optimal predictors. This raises a natural question: Does the prediction performance remain robust under approximation? In this paper, we first provide a tight bound on the worst-case effect of approximation on prediction error. We show any predictor satisfying retains bounded cumulative-expected-squared prediction error. However, while the Bayesian mixture predictor has error in an environment , we establish a tight bound on the excess error of relative to . Thus, approximation can substantially degrade performance in environments assigned small prior probability. In sharp contrast, when environments are averaged according to prior , we prove that the excess prediction error is only . Furthermore, we establish a distribution-shift theorem for dominated shifts from a training prior to a test prior , interpolating our average-case and worst-case guarantees. Finally, we extend our robustness analysis to a finite-sampling setting relevant to learning universal predictors with neural networks and Transformers, where could be unbounded. [For the full abstract, see the paper.]
Introduction
Large language models (LLMs) trained for sequential prediction have acquired a remarkable range of capabilities from vast and diverse data. In recent years, scaling model capacity, expanding the volume and breadth of training data, increasing computational resources [KMI+20], and incorporating real-world human interaction [13] have driven substantial progress, transforming LLMs into increasingly general-purpose systems for problem solving and induction. Given the substantial and rapidly growing resources required to train frontier models, understanding the benefits and limitations of further scaling has become more important than ever, raising a range of fundamental research questions.
Classical learning frameworks, such as semi-supervised learning, PAC learning, and reinforcement learning, typically analyze learning relative to a specified task, data distribution, and hypothesis class. In contrast, a single LLM trained on heterogeneous data can perform a wide range of tasks—including translation, writing, coding, and mathematical reasoning—without task-specific retraining, guided only by instructions or a few demonstrations provided in context. Although pretraining is largely driven by minimizing next-token log loss, the resulting models exhibit capabilities involving generation, abstraction, reasoning, and planning, often without explicit supervision for each capability. Moreover, once trained, such a model can be deployed at a comparatively low marginal cost per inference [2]. LLM training thus distills enormous amounts of data and computation into a reusable neural network whose fixed parameters support adaptive behavior across contexts and tasks. This phenomenon suggests that learning to predict sufficiently rich and heterogeneous sequences can give rise to broad knowledge and generalizable computational abilities. Precisely how such capabilities emerge from a seemingly simple prediction objective, however, remains incompletely understood, and is central to characterizing the fundamental power and limitations of the current LLM paradigm.
Motivating Prior Work. Meta-learning, or learning to learn, provides a conceptual framework for understanding this phenomenon [4]. Meta-learning treats the learning procedure itself as an object of optimization, using experience accumulated across tasks to improve the learning of new tasks [HAMS21]. It involves two levels of learning, either explicitly or implicitly: an inner process learns a particular task from a specified dataset and objective, while an outer process optimizes aspects of the inner process—such as the model class, initialization, or optimization rule—with the aim of reducing the data or computation required to solve new tasks.
One approach is memory-based meta-learning (MBML) [SBB+16], in which a memory-augmented model—an LLM whose context serves as memory, for example—is trained on data drawn from a distribution over tasks using a single optimization loop. The two levels of learning are realized through different mechanisms: the outer process updates the model’s parameters across training tasks, whereas the inner process adapts to an individual task through changes in the model’s memory state. At test time, the parameters remain fixed, and adaptation proceeds through memory dynamics without task-specific parameter updates. In-context learning thus emerges as a learned adaptation procedure implemented through the model’s forward computation.
This MBML framework admits a precise Bayesian characterization [OWR+19, XRLM22]. For a fixed horizon , suppose that each training sequence is generated by first sampling a task from a prior distribution , then sampling data from . Suppose a sufficiently expressive model is trained to minimize the population log-loss objective
Then behaves as the Bayesian mixture predictor . In particular,
where is the posterior probability of . Thus, conditioned on an observed prefix , the Bayesian predictor combines the predictions of the candidate tasks according to their posterior probabilities, reflecting both their prior weights and their consistency with the observations. The appeal of the Bayesian perspective lies in its strong prediction guarantee [8]: at test time, can achieve efficient in-context adaptation to any task with positive prior weight :
This provides a principled explanation of in-context learning as amortized Bayesian prediction: training amortizes learning across tasks by encoding an adaptive inference procedure into the forward computation of a neural network. In particular, consider the class of all computable environments1 endowed with the universal prior , where denotes the Kolmogorov complexity of . The resulting single Bayesian predictor becomes Solomonoff universal induction [Sol78, Sol86], which achieves cumulative expected squared prediction error simultaneously in every computable environment , independently of the prediction horizon. Building on this connection, [3] proposed learning increasingly general predictors by meta-training neural networks on synthetic data generated by computationally bounded environments, represented by universal Turing machines with bounded program lengths and runtimes. Progressively relaxing these bounds while increasing model capacity and training data offers a principled approach toward universal prediction, provided that the corresponding prediction objectives can be optimized sufficiently well. The paper [4] placed this framework within the broader context of algorithmic compression: minimizing sequential log loss can be understood as learning to compress observations by approximating and amortizing Bayesian prediction over their underlying generative mechanisms. Under a universal prior, the resulting predictor favors simpler computational explanations while averaging over all computable hypotheses according to both their simplicity and their consistency with the observations.
Our Main Contributions. The classical theory [Sol78, Sol86, Hut05] focuses on the exact predictor that attains the minimum population log loss. In practice, however, limited model capacity and imperfect optimization—even when training data are abundant—mean that learned models generally only approximate the ideal Bayesian predictor. As emphasized by [4], extending the Bayesian view of meta-learning beyond exact realizability and perfect optimization remains an important open direction. Motivated by this line of work, we take a first step toward understanding the fundamental gap between ideal Bayesian prediction and practical learned predictors by addressing the following central question:
To what extent do the guarantees of the ideal Bayesian predictor continue to hold when practical learning architectures can only approximate it?
In this work, we develop a theory of Bayesian and universal sequential prediction under approximation, characterizing performance and robustness across a broad range of regimes: Environment-Wise Worst-Case Robustness. We show in Theorem 1 that every approximate predictor with excess population log loss at most , equivalently , retains a finite bound on its cumulative expected squared prediction error. Specifically, for any environment , the error of the ideal Bayesian predictor is , whereas that of is bounded by . We further show in Theorem 3 that the approximation-dependent term is tight. In particular, for a universal mixture, an error of order is attained for infinitely many computable environments . We also extend the analysis to two other fundamental measures of prediction error: cumulative KL divergence and cumulative squared Hellinger distance.
Typical Average-Case Robustness: In sharp contrast, we prove in Theorem 4 that, when performance is averaged over environments according to the prior —the canonical average-case measure—the excess prediction error due to approximation is only . We show that an analogous average-case bound holds for cumulative KL divergence but not for cumulative squared Hellinger distance.
Distribution-Shift Robustness: Furthermore, we establish a distribution-shift theorem (Theorem 5) for dominated shifts from a training prior to a test prior , where for every environment and some . The resulting error bound is , thereby interpolating between our average-case and worst-case guarantees.
Finite-Data Robustness: Finally, in § 5, we study predictors learned from finitely many training sequences. We show exponential upper-tail bounds for population KL error, yet its expectation can remain infinite when unseen events receive zero probability. Nevertheless, by using a weaker alternative measure of approximation, we extend our robustness analysis to this setting. In particular, we bound the expected average excess prediction error by the representation and optimization errors plus a finite-sample term that decays exponentially with the number of training sequences.
Taken together, these results provide a broad characterization of robustness to approximation in sequential prediction. They explain the sharp contrast between strong aggregate benchmark performance and poor worst-case performance on rare environments. This sharp distinction between worst-case and average-case behavior helps explain why approximation may have a much milder effect in practice. It may also shed light on the strong empirical performance of learned universal predictors [3, 4] and, more broadly, of LLMs across a wide range of tasks and benchmarks. Our distribution-shift theorem has several implications beyond basic transfer principle for sequential prediction. It provides a quantitative rationale for targeted sampling and synthetic data generation designed to protect important but underrepresented families of tasks. In particular, it provides a theoretical rationale for expecting that augmenting real-world datasets with Universal-Turing-Machine-generated (UTM-generated) data may improve prediction performance on practically important task families, as suggested by [3].
Notations and Preliminaries
We assume familiarity with the basic notions in probability theory and algorithmic information theory, including conditional probability and Turing machines. See, for example, [12] for background. Throughout this paper, denotes the base-2 logarithm, while denotes the natural logarithm. Entropy and KL divergence are measured in bits.
Sequence measures and mixtures. Let be a finite alphabet with . Suppose is a probability measures on (representing a sequence generative model). We use to denote the probability that a sequence sampled from begins with , and to denote the induced distribution on . We denote . Let be a countable class of measures. Let be weights satisfying . Then, define the mixture . When is a selected measure from , we write . Note that for every finite string , .
The universal mixture. Solomonoff [Sol78, Sol86] introduced a universal prior for sequence prediction. Unless stated otherwise, we fix a universal monotone Turing machine throughout the paper. We say a measure is estimable if there exists a monotone Turing machine that takes an infinite sequence of independent, uniformly random bits and generates a sequence according to . Let denote the length of the shortest program for on this machine.
Let denote the class consisting of all estimable measures. Set
The programs are described in a prefix-free fashion, so Kraft’s inequality gives , and . The fair-coin measure belongs to , so for every finite string . We also use for the usual prefix-free Kolmogorov complexity of a finite string or integer .
KL divergence. For two measures and and , define
We use the convention that and for . The finite-horizon divergences are nondecreasing in , so the supremum is also their limit. The chain rule gives that:
Cumulative prediction error. As in [Sol78], we use cumulative squared prediction error as our primary criterion for evaluating the prediction performance of :
To examine the scope of our guarantees across different error criteria, we also consider cumulative squared Hellinger error:
and cumulative KL error:
The following inequalities relate these error criteria.
Lemma 1 (Theorem 3.2.3 in [7]). Let and be probability measures on . For every ,
The same inequalities hold for the corresponding infinite-horizon errors , , and .
Robustness under Approximation: Task-wise Guarantee
In this section, we establish a tight task-wise prediction guarantee for a trained model that approximates the Bayesian predictor with KL gap . The KL divergence provides the most natural measure of approximation quality because the population log-loss of at horizon admits the following decomposition, where denotes the Shannon entropy:
Moreover, minimizes the population loss over all probability measures on , attaining the minimum value . Thus, .
Environment-Wise Guarantee: Upper Bound
We first extend the classical analysis of [8] by showing that, under a finite KL-divergence, the cumulative squared prediction error remains finite for every environment in .
Theorem 1 (Environment-wise upper bound). If , then for every ,
The proof is given in Appendix § 8, which bounds the cumulative KL error for every by
Theorem 1 then follows from Lemma 1 that and . We thus also prove that, up to constant factors, this upper bound extends the cumulative squared Hellinger distance .
Specializing to the universal mixture and using the relation , we obtain the following environment-wise prediction-error bound in terms of the Kolmogorov complexity .
Theorem 2 (Upper bound for universal prediction). If , then for every ,
where the implicit constant depends only on the fixed universal Turing machine.
Environment-Wise Guarantee: Tight Worst-Case Lower Bound
Our next theorem shows that the approximation-dependent term in Theorem 1 is unavoidable and tight up to a constant factor. Remarkably, for the universal mixture, this tightness holds for infinitely many environments in . Throughout this subsection, we restrict attention to the binary alphabet .
Theorem 3 (Tight lower bound for universal prediction). For every , there are infinitely many estimable measures such that
where the implicit constant depends only on the fixed universal Turing machine.
Thus, even an arbitrarily small KL-approximation gap can coexist with a cumulative squared prediction error that grows exponentially with the Kolmogorov complexity of the environment. We now describe the high-level idea behind the proof. The complete proof is given in Appendix § 10 and formalizes the following rare-prefix construction.
Rare-prefix construction. The lower bound exploits prefixes that are unlikely under the Bayesian mixture , but occur with probability one under a particular environment. For , let denote the deterministic environment that outputs . Suppose and that a finite prefix of has mixture probability . We construct by replacing the next conditional predictions along after the prefix with the uniform distribution, while leaving all other predictions unchanged.
Each modified history has probability at most under , and replacing its next-symbol distribution with the uniform distribution incurs at most one bit of conditional KL divergence. Hence, the total KL cost is at most . Under , however, every modified history is visited with probability one, and each modified prediction contributes exactly to the squared prediction error. Taking yields the following lemma.
Lemma 2 (Lower bound from a rare prefix). Let , and let be a finite prefix of and . For every , there exists a probability measure with such that
When , it yields an lower bound whenever is sufficiently large. The proof of Theorem 3 identifies infinitely many deterministic environments having prefixes that satisfy this condition. Since , the resulting approximation penalty is .
Finally, we can show that Lemma 2 also extends to cumulative KL and squared Hellinger errors. Then by Lemma 1, we have, for every , and . Thus, our construction can establish similar lower bounds as in Theorem 3 for cumulative KL and squared Hellinger errors.
Robustness under Approximation: Average-Case Analysis
In this section, we show that the environment-wise prediction guarantee under approximation can be sharply improved when performance is averaged across environments. We then establish a distribution-shift theorem for moving from the training prior to a dominated test prior . The result interpolates between our average-case and environment-wise guarantees. Before stating the theorems and discussing their implications, we explain the motivation for developing an average-case theory of robustness under approximation. § 3.2 shows that the approximation-dependent term in Theorem 1 is unavoidable in the worst case. For an environment with a small prior weight , this term can overwhelm the classical prediction guarantee for the ideal Bayesian predictor.
Such pessimistic worst-case guarantees, however, stand in contrast to the strong empirical performance of learned universal predictors [3, 4] and, more broadly, to the outstanding performance of LLMs across a wide range of tasks and benchmarks. This contrast raises the question of why approximation has a milder effect in practice. Empirical evaluations typically aggregate performance across many tasks or environments, whereas an environment-wise guarantee treats each environment individually. Accordingly, we first study prediction performance averaged over the same prior that defines the training mixture . We show that, in sharp contrast to the worst-case setting, approximation has a substantially milder effect under this average-case criterion. Our distribution-shift theorem can then be used to extend this guarantee to other test distributions.
Sharp Contrast between Typical and Task-Wise Prediction Guarantees
Theorem 4 (Approximation robustness: average-case upper bound). If , then
This theorem is a consequence of the following exact average-case decomposition:
Proposition 1 (Exact average-case decomposition). For every predictor and every ,
Theorem 4 matches the exact prediction-error baseline of the ideal mixture : the average cost of replacing with is . Together with Theorem 3, this result shows that approximation can preserve aggregate performance while substantially degrading performance on individual tasks.
Remark 1. Note that Theorem 4 cannot be obtained simply by averaging the bound in Theorem 1. Instead, the proof exploits the mixture structure to identify the average excess error exactly. To see this, consider the case when is infinite, the following average diverges:
A fundamental differentiation: The role of approximation in Theorem 4 also highlights an interesting distinction between exact Bayesian prediction and its approximate counterpart. The classical theory applies to cumulative squared prediction error, KL error, and Hellinger error. As we show in § 3.1, the task-wise guarantees under approximation extend to all three error criteria. By contrast, the average-case decomposition in Proposition 1 holds only for cumulative squared prediction error and cumulative KL error. For KL error, the chain rule immediately yields the analogous decomposition.
Hence, if , then
For cumulative squared Hellinger error, neither the analogous decomposition nor a uniform additive excess-error bound relative to holds in general. The mixture predictor need not minimize average squared Hellinger error; a counterexample is given in Appendix § 12.
Remark 2. Prior work has shown that the average gains of knowledge distillation can coexist with degraded performance on underrepresented subgroups and tail classes [11 WNZ+23]. By viewing as the teacher model and as the student model, our results provide a complementary explanation in Bayesian sequential prediction: a small population KL distillation loss controls the student’s average excess prediction error, but can coexist with substantial degradation relative to the teacher in rare environments.
Approximation Robustness under Distribution Shift
The preceding analysis evaluates performance under the training prior . We now consider a different test prior . Our next theorem shows that the average-case guarantee remains robust when the test prior is dominated by the training prior, in the sense that for every environment and some . This result has several implications beyond establishing a basic transfer principle for sequential prediction and interpolating between the worst-case and average-case analyses of robustness under approximation. One potentially significant implication is that the theory provides basic support for the argument that augmenting real-world datasets with Universal-Turing-Machine-generated (UTM-generated) data can improve prediction performance on practically important families of tasks, as suggested by [3].
Theorem 5 (Approximation robustness under distribution shift). Let be a probability distribution on satisfying for every and some . Define the corresponding mixture
If , then, for every ,
In particular, the average excess error relative to is .
Cumulative KL error satisfies a distribution-shift bound of the same order. Indeed, the KL decomposition and the domination , together with the KL transfer inequality (Lemma 3) used in the proof of Theorem 1, give
Implication I: Rare task families and targeted sampling. Theorem 5 interpolates between average-case and environment-wise guarantees. Taking and recovers the average-case upper bound in Theorem 4. At the other extreme, if is concentrated on a single environment , we may take . Then , and
Thus, the theorem also recovers the order of the environment-wise bound in Theorem 1.
More generally, let be a family of environments with prior mass , and consider evaluation under . Since , we may take and obtain
For this conditional test distribution, the guarantee depends on the total prior mass of the task family.
This suggests a way to improve coverage of an important but underrepresented family. Let be a probability distribution supported on that family, and define
where is the fraction of training mass allocated to . Since , any predictor with satisfies
The approximation guarantee for this family therefore depends on its assigned training mass , independently of its rarity under the original prior .
Within this model, the bound provides a quantitative rationale for increasing the training probability assigned to important but underrepresented task families, for example through targeted sampling or synthetic data generation. In particular, it supports augmenting UTM-generated training data with real-world datasets to improve prediction performance on practically important task families, as suggested by [3].
Implication II: Fine-tuning with replay. Replaying pretraining data during adaptation can mitigate forgetting [1, 9]. Theorem 5 can be used to give a population-level analysis of this strategy. Interpret as the target-task distribution and as the original pretraining distribution. During fine-tuning, draw a fraction of sequences from and a fraction from . The resulting population training distribution is .
Suppose the fine-tuned predictor approximately minimizes the population log-loss, in that
Both task distributions are dominated by the fine-tuning prior: , . Applying Theorem 5 to each distribution yields the simultaneous guarantees
The first bound controls prediction error on the target distribution, while the second controls error on the original distribution. Since minimizes average squared prediction error under , the second bound also limits the possible increase in average error relative to the predictor before fine-tuning.
For fixed approximation accuracy , increasing tightens the target-distribution bound while weakening the original-distribution bound. The mixing parameter therefore controls the trade-off between the performance guarantees on the target and original task distributions. Finally, we remark that these guarantees assume that the fine-tuned predictor achieves the stated population approximation accuracy .
Implication III: Choice of reference universal Turing machine. Let and be two universal Turing machines. By the invariance theorem, there is a constant such that
for every estimable .
Consider the universal priors , , where . Then we have . If training uses the mixture induced by and , then test under the prior induced by satisfies
Thus, changing the reference machine preserves the form of the average-case guarantee, with constants depending only on the pair of machines, independently of the environment and prediction horizon. These constants may nevertheless be large because their dependence on the compiler overhead is exponential.
Robustness of Learning and Prediction with Finite Data
We now examine whether our prediction guarantees remain robust when the predictor is learned from finitely many samples. Fix a prediction horizon , and let be learned from independent training sequences by minimizing the empirical log-loss over the hypothesis class up to an additive optimization error . In addition to this optimization error, the model class incurs a representation error . We consider .
We show an exponential upper-tail bound on the population KL error: for every , there exist constants such that2, for all ,
Nevertheless, this high-probability guarantee need not imply a finite expected KL error. A learned predictor may assign zero probability to an event that has positive population probability but is absent from the training data. Precisely, it is possible that
Despite this difficulty, we show that our squared prediction error guarantees remain meaningful in expectation. It follows from two observations. First, if we characterize the approximation quality with a weaker condition instead of , then there exist constants and such that for any ,
which means that the finite-sample contribution to the approximation error decays exponentially with the training sample size. Moreover, one can check that all of our theorems extend from the KL approximation condition to the weaker condition , with corresponding changes to the constants. In particular, Theorem 4 and (6) together imply: for any ,
Conclusion
We characterize how approximation affects Bayesian prediction and Solomonoff induction. Our results show that approximation can preserve average performance while substantially degrading performance on individual tasks. The analysis also extends to dominated shifts from the training prior to the test prior and is robust in the finite-sample setting.
An important direction for future work is to develop mitigation strategies that provide stronger guarantees across all tasks simultaneously. We explore initial approaches in Appendices § 15 and § 16.
Another open direction is length generalization. Our finite-sample analysis considers training and test sequences of the same length; extending these guarantees to longer test sequences remains a challenge. One possibility is to combine pretrained predictors with continual learning, using minimum description length principle [5 PH05] to guide adaptation as new observations become available.
Acknowledgments
We thank Xinyu Mao and Guangxu Yang for early discussions, and Marcus Hutter for his comments. We used AI tools for polishing the language, searching the literature, and developing and drafting all proofs. We have checked and revised all proofs and take full responsibility for the whole content of this work.
Proof of Theorem 1
Lemma 3 (KL transfer under domination). Let be probability distributions on the same finite space . Suppose that for some . Then
Proof. The claim is immediate when , because then . We may therefore assume . The conclusion is also immediate if .
Write for a probability distribution , and define
on the support of , with elsewhere. Then and . Introduce an auxiliary binary variable . Under , let and
Thus . Define using the same conditional law of given , but with , and put
Finiteness of implies . Let denote the conditional law of under given , and define analogously. One can check that
The conditional kernel from to is the same under the two joint distributions. Hence
Applying the chain rule in the other direction gives
where
is binary KL divergence. Moreover,
and therefore
Together with (8), this yields
We next use the inequality
To verify it, first observe that
If , the second term is nonnegative, and (10) follows. If , the desired inequality is equivalent to
For fixed , let
This function is strictly convex on , and its stationary point is
Writing , direct substitution gives
Now and
while
The numerator of after taking a common denominator is affine and strictly decreasing, positive at and negative at . Thus first increases and then decreases. Since and as , it is initially positive and then crosses zero exactly once. Hence first increases and then decreases to . It follows that , proving (10).
Let . Substituting (10) into (9), and using , gives
Proof of Theorem 1. Fix , and apply Lemma 3 to
Since ,
Applying Lemma 1 and taking the supremum over gives the conclusion. □
Proof of Lemma 2
Let and set . Note that
For , write
We construct a measure on as follows: set
and let agree with at every other history of positive -probability. At histories of zero -probability, choose an arbitrary conditional distribution.
For any distribution on ,
where is the binary entropy. Hence, for every , the chain rule gives
Taking the supremum over yields
The modified conditionals occur only after the prefix . Therefore, the first predictions along are unchanged, and
At each history , the next-symbol distribution under is a point mass, whereas . Thus
Consequently,
This finishes the proof.
Proof of Theorem 3
We remark Lemma 2 already gives a nearly exponential lower bound in the universal setting. Specifically, for every , since , there is a string such that . Let . We have
The first inequality is by combining (i) and (ii) Lemma 2 therefore yields, for all sufficiently large ,
This matches the upper bound in Theorem 1 up to a polynomial factor. We next show that this factor can be removed.
We will make use of the following coding theorem.
Lemma 4 (Corollary 4.3.1 in [12]). If is a lower semi-computable semi-measure on with , then there is a constant such that .
Proof of Theorem 3. For , define
Note that the set is prefix-free. Since , a counting argument gives a string satisfying . Here
is the prefix Kolmogorov complexity of the string . Denote the deterministic measure outputting . Note that is estimable, as it can hard-code the self-delimiting program generating and then output zeros forever. In addition, we have .
Let be the Solomonoff prior
which is semicomputable. Since is prefix-free, . By letting , Lemma 4 therefore gives
Furthermore, a standard fact in algorithmic information theory (see e.g. Section 1.2.5 in [8]) says that . Thus,
In particular,
Fix . Applying Lemma 2 with and gives: there exists a universal constant , such that
for infinitely many . □
Proof of Proposition 1
For a history with , let denote the posterior weight of . For any symbol ,
The variance decomposition gives
Multiplying by and using yields
Summing over all histories , all symbols , and all gives
A counterexample for squared Hellinger error
Let and , with prior weights and , respectively. Write , so that . After the first symbol , every with predicts the remaining sequence perfectly. Thus its averaged cumulative squared Hellinger error is
This is minimized at
Hence is not a minimizer, ruling out the analogous decomposition into its average error and a nonnegative approximation term. Moreover, let for sufficiently small. Since and differ only in their first-step predictions,
However,
Thus, no finite constant can make hold for all sufficiently small .
Proof of Theorem 5
By Proposition 1, we have
and
In the following, we will show that
which concludes the proof.
The assumption implies that, for every finite history ,
In particular, . Hence, by Lemma 1, we have
Next, define
Using and Lemma 1, we obtain
Expanding the square, we obtain
where
By the Cauchy–Schwarz inequality,
It follows that
Together with (13) and (14), this yields (12).
Proofs for Section 5
Fix a horizon . Write , , and . Let denote the predictor achieving . Write for the empirical log-loss, so that .
Proof of (4). Let be the empirical distribution of the training sequences. For , define
Write and
Hoeffding’s inequality and a union bound give
Conditioned on , the nonnegativity of log-loss and approximate empirical optimality imply
Thus,
For any , choose small enough that the second term is at most . The claim follows with and .
Example establishing (5). Take , , and consisting of and . On the event that all training observations are zero, which has probability , exact empirical log-loss minimization selects . Since , the expected KL error is infinite for every finite . Padding deterministic zeros extends the example to any fixed .
Proof of (6). Recall , let . By and , we have
Theorems 4, 5, and 1 under squared-error approximation. Suppose . By Proposition 1, we have
In the proof of Theorem 5, by almost the same argument except adjusting (14) to , we will get
Letting concentrate on a single environment , we recover Theorem 1, up to constants:
Theorem 3 under squared-error approximation. By Lemma 1,
Thus Theorem 3, applied with budget , implies the same infinite-horizon lower bound under squared-error approximation, up to constants.
Mitigation With Category Information
The worst-case bounds of order in Theorem 1 arise when approximation is controlled only through a single aggregate KL bound . When available, training-time task category information enables separate control of approximation errors within each task category, potentially improving category-wise prediction performance. Notably, the learned predictor does not require category tags at test time.
Let be a finite set of categories, with defining the partition . Write
Thus . Recall that, by Theorem 1, an aggregate approximation guarantee gives
Direct control of the category-wise KL error avoids this inverse-category-mass amplification. The following guarantee follows immediately from Lemma 1 and Proposition 1.
Proposition 2 (Category-wise approximation). If , then
In particular, if the tag identifies the generating task and , then
To control the category-wise approximation errors , one may minimize the following worst-category excess log-loss, or a suitable surrogate:
Higher moments of sequence log-loss
One may instead minimize the following sample-only training objective:
Fix a horizon . Let be the minimizer of among all probability distributions on . Suppose the predictor satisfies
Lemma 5. For every ,
Here, the implicit constant in may depend on and , but not on any other parameters.
Proof. Let be the support of . Since , the distribution is also supported on . The minimizer assigns all its mass to , since otherwise, renormalizing its restriction to would strictly decrease the objective. If , then the result follows immediately from
on the unique point . We henceforth assume . In this case, for every .
Write . Consider the Lagrangian
Since the minimizer is an interior point, the first-order condition gives
Thus, setting , we obtain
Since , , and , we also have . Summing (16) over gives
Since is concave, Jensen’s inequality gives
We next show that dominates . If , then (16) gives
If , then , and hence
The two cases imply
We now use the approximate optimality of . Define . Since , it has that on , and thus
For and , we have
Indeed, when with , subtracting the right-hand side from the left-hand side gives a function satisfying and
When with , the corresponding function satisfies and
Applying (18) with
we obtain
By (16),
It follows from the approximation guarantee that
and
Therefore, by Holder’s inequality,
By applying Lemma 1, we obtain the lemma. □
References
- [1][BGB+25] Louis Béthune, David Grangier, Dan Busbridge, Eleonora Gualdoni, Marco Cuturi, and Pierre Ablin. Scaling laws for forgetting during finetuning with pretraining data injection. In Aarti Singh, Maryam Fazel, Daniel Hsu, Simon Lacoste-Julien, Felix Berkenkamp, Tegan Maharaj, Kiri Wagstaff, and Jerry Zhu, editors, Forty-second International Conference on Machine Learning, ICML 2025, Vancouver, BC, Canada, July 13-19, 2025, volume 267 of Proceedings of Machine Learning Research. PMLR / OpenReview.net, 2025.
- [2]DeepSeek-AI. Deepseek-v3.2: Pushing the frontier of open large language models, 2025.arxiv.org/abs/2512.02556
- [3]Jordi Grau-Moya, Tim Genewein, Marcus Hutter, Laurent Orseau, Grégoire Delétang, Elliot Catt, Anian Ruoss, Li Kevin Wenliang, Christopher Mattern, Matthew Aitchison, and Joel Veness. Learning universal predictors. In International Conference on Machine Learning, 2024.
- [4]Tim Genewein, Jordi Grau-Moya, Li Kevin Wenliang, Laurent Orseau, and Marcus Hutter. Algorithmic compression via pretrained neural networks. Entropy, 28(6):596, 2026.DOI
- [5]Peter D Grünwald. The minimum description length principle. The MIT press, 2007.DOI
- [7]Marcus Hutter, David Quarel, and Elliot Catt. An introduction to universal artificial intelligence. CRC Press, 2024.DOI
- [8]Marcus Hutter. Universal artificial intelligence: Sequential decisions based on algorithmic probability. Springer Science & Business Media, 2005.
- [9]Adam Ibrahim, Benjamin Thérien, Kshitij Gupta, Mats L. Richter, Quentin Gregory Anthony, Eugene Belilovsky, Timothée Lesort, and Irina Rish. Simple and scalable strategies to continually pre-train large language models. Trans. Mach. Learn. Res., 2024, 2024.arxiv.org/abs/2403.08763
- [11]Michal Lukasik, Srinadh Bhojanapalli, Aditya Krishna Menon, and Sanjiv Kumar. Teacher’s pet: understanding and mitigating biases in distillation. Trans. Mach. Learn. Res., 2022, 2022.arxiv.org/abs/2106.10494
- [12]Ming Li and Paul Vitányi. An Introduction to Kolmogorov Complexity and Its Applications. Texts in Computer Science. Springer, New York, NY, 3 edition, 2008.
- [13]Long Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida, Carroll Wainwright, Pamela Mishkin, Chong Zhang, Sandhini Agarwal, Katarina Slama, Alex Ray, et al. Training language models to follow instructions with human feedback. Advances in neural information processing systems, 35:27730–27744, 2022.arxiv.org/abs/2203.02155