stat.MLSep 22, 2026

Statistical Gains from Looped Estimation under Parameter Budgets

Authors: Xinyu TianXiaotong Shen

Abstract

Growing memory demands in artificial intelligence motivate learning with fewer trainable parameters. We ask whether a looped estimator, which repeatedly applies one fitted operator with parameters shared across iterations, can improve statistical accuracy under a common parameter budget. Its conventional untied counterpart uses separate parameters at each iteration. For general likelihood models, we establish an upper bound on squared Hellinger risk for looped sieve maximum likelihood and a minimax lower bound over the tuned untied family. These bounds reveal a parameter--iteration--accuracy tradeoff: repeated computation can improve approximation without adding parameters, while increasing computational cost and fitted-class complexity. For targets of known Hölder smoothness, looped residual feedforward networks and a specified post-layer-normalized Transformer attain the minimax polynomial rate up to logarithmic factors with a fixed number of bounded real parameters. At sufficiently large fixed budgets, looped worst-case risk vanishes as sample size grows, whereas optimal worst-case untied risk remains bounded away from zero. Under specified growing-budget conditions, the loop-to-untied risk ratio also tends to zero. Gaussian and Laplace regression, binary response, and energy-based density estimation illustrate the theory.

Explore similar work

Jun 29, 2026cs.AI

Using Large Language Models as Low-Cost Statistical Estimators for Human-Response Data

Quantitative research across the social and behavioral sciences depends on human subject experiments that are expensive, slow, and subject to sampling bias. Here we show that pretrained large language models induce risk-equivalent estimators of conditional expectations under squared loss, establishing restricted functional risk equivalence: under squared loss, the LLM induces an estimator whose risk matches the Bayes optimal risk for squared-loss prediction of conditional expectations for any inference that depends on the data only through the conditional mean. We formalize the LLM as a misspecified functional estimator T(P^n)T(\hat{P}_n) trained on i.i.d.\ data, decompose the estimation error into representation bias εrepε_{\mathrm{rep}} and optimization error, and prove that under mild regularity conditions the LLM's expected error converges to the irreducible population variance plus the squared representation bias, with the representation bias bounded by the Pinsker inequality. The identifiability error δδ propagates into the effective bias, inflating the asymptotic risk floor. We establish restricted functional risk equivalence via a bidirectional Le Cam deficiency analysis: the forward deficiency vanishes asymptotically while the reverse deficiency is exactly zero. We provide finite-sample concentration bounds and a calibration protocol with explicit decision rules. The result is a precise, provable statement: a well-calibrated LLM achieves the Bayes-optimal risk for conditional-mean-dependent inference, bounded by explicit scope conditions. In practical applications, this means that under satisfied conditions and well-calibrated models, large language models can be used in many prediction and decision-making tasks that originally relied on human experiments, approximating near-optimal statistical inference at lower cost.
Haobo Yang
Date pendingcs.LG

On the Residual Scaling of Looped Transformers: Stability and Transferability

Looped (weight-tied) Transformers apply a shared residual block NN times (hh+εf(h)h \leftarrow h + \varepsilon\,f(h), same ff at each step), increasing effective depth without adding parameters. Prior depth-scaling analyses prescribe ε=1/ ⁣L\varepsilon = 1/\!\sqrt{L} for depth-LL residual networks. We show that this is insufficient for looped architectures: weight sharing makes residual updates correlated across iterations, requiring the stronger scaling ε=1/N\varepsilon = 1/N. For multi-layer blocks (LL unique layers looped NN times), we derive a factored parameterization ε=λ/(N ⁣L)\varepsilon = \lambda/(N\!\sqrt{L}) that separates the two sources of growth: 1/N1/N controls the within-layer loop correlation, and 1/ ⁣L1/\!\sqrt{L} controls the across-layer variance. A key consequence is that the optimal learning rate depends only on the number of unique layers LL, not on the loop count NN, enabling direct hyperparameter transfer from small to large NN without retuning. Experiments on looped Transformers confirm that 1/N1/N scaling improves trainability and yields better loss than 1/ ⁣N1/\!\sqrt{N} scaling across loop counts.
Shaowen Wang, Bingrui Li, Ge Zhang +3
Sep 22, 2026stat.ML

Optimal Tradeoffs Between Network Size and Parameter Magnitude in Neural Approximation and Minimax Regression

The statistical accuracy of neural networks depends on both their approximation power and the complexity of the class fitted from data. While increasing network size is a natural way to improve approximation, parameter magnitude provides another resource whose role must be quantified in both respects. We establish a sharp width--magnitude tradeoff at fixed depth using one elementary bounded 11-Lipschitz Dyadic--Triangular Activation. For the unit ββ-Hölder ball on [0,1]d[0,1]^d with 0<β10<β\leq1, the optimal LpL^p approximation error for 0<p<0<p<\infty is of order [N2log(eNT)]β/d[N^2\log(eNT)]^{-β/d} when the network width satisfies N2d+3N\geq2d+3 and the parameter magnitudes are bounded by T1T\geq1. Matching lower bounds hold for every fixed globally Hölder activation; its Hölder exponent affects the constants but not the rate. Under bounded design densities and independent centered sub-Gaussian noise, approximate least squares over the full clipped class at depth 2323 attains the classical Hölder minimax risk O(M2β2β+d)\mathcal{O}(M^{-\frac{2β}{2β+d}}) without logarithmic loss whenever N2log(eNT)Md2β+dN^2\log(eNT)\asymp M^{\frac{d}{2β+d}}, where MM is the sample size. This yields a continuum of statistically optimal choices, ranging from unit parameter radius to fixed network size. At fixed size, four hidden layers with at most 8d+78d+7 nonzero parameters give a near-optimal radius, while six layers with at most 8d+278d+27 attain the optimal order logT=O(ηd/β)\log T=\mathcal{O}(η^{-d/β}) at approximation error ηη. The same decoding method also yields fixed-size Transformer approximation.
Baicheng Li, Zuowei Shen, Haizhao Yang +1