Machine Learning (Statistics) · Cross-listed from cs.LG
Quantifying Ranking Uncertainty in LLM Benchmarks
Bitya Neuhof and Yuval Benjamini
Pretrained models are typically ranked on multi-task leaderboards to assess their effectiveness across diverse tasks. Rank confidence intervals were recently introduced as a method to quantify the uncertainty in these rankings by aggregating pairwise hypothesis tests. In this work, we analyze the sources of uncertainty in the knowledge evaluation
benchmark MMLU and show how hypothesis tests can be modified to account for their effects. We demonstrate that ranking variability across MMLU subjects is substantial and should be considered when comparing LLMs or identifying the top-performing models.
Read the full abstract ▾
Statistical Computation · Cross-listed from math.NA
Stochastic Analysis of Fade Duration Using Wiener Chaos Expansion and Malliavin Calculus: Optimal Importance Sampling via Adaptive SGD
Francisco Delgado-Vences
Characterizing fade duration in wireless channels is fundamental for designing robust communication systems. Classical approaches – Rice's level-crossing theory and Monte Carlo simulation – lack precision for tail events and are computationally prohibitive for rare-event probability estimation. This paper introduces a rigorous framework combining Wiener Chaos Expansion (WCE), Malliavin Calculus, and importance sampling with adaptive weights to analyze fade duration $Z(T)$ distributions. Main contributions include: (i) high-accuracy moment estimation and CCDF characterization via WCE minimizing Monte Carlo variance; (ii) Markovian projection reducing infinite-dimensional dynamics to tractable systems ($\dim \leq 3$) for Rayleigh, Rician, and Nakagami models under stated assumptions; (iii) asymptotically optimal importance sampling weights derived from Malliavin sensitivities, achieving 839 to 2516x variance reductions; (iv) a theoretically grounded and provably efficient adaptive SGD
algorithm with Robbins-Monro step size schedule for parameter estimation. Numerical experiments validate our approach with relative errors below 0.5%, enabling gradient-based optimization of fade duration statistics even for regimes where $P \sim 10^{-15}$, without requiring $\mathcal{O}(1/P)$ samples, by evaluating sensitivities through analytical Malliavin weights.
Read the full abstract ▾
Statistical Methodology
Multi-Source Transfer Learning of Sparse Single-Index Models
Ye Tian
Transfer learning leverages knowledge from related source domains to improve learning in a target domain. Recent theoretical advances cover a broad range of regression settings within (generalized) linear models. Despite their diversity, these methods share two common constraints: they assume a known link function or linear structure and require direct access to raw source data. To move beyond these constraints, we propose a source-data-free transfer learning framework based on the single-index model (SIM). Instead of requiring raw source data, our method transfers only summary statistics derived from a generalized Stein's lemma in a one-time communication. This design preserves
privacy and avoids side effects caused by dissimilarities of unknown nonlinear link functions across domains. To capture flexible, unknown nonlinearity, we employ a multilayer perceptron guided by the pre-estimated index from the transferred statistics, which significantly mitigates overfitting. Extensive experiments on synthetic data and a real-world application demonstrate consistent improvements over existing (generalized) linear model-based approaches. The proposed framework thus offers a practical,
privacy-preserving, and nonlinear-adaptive solution for transfer learning.
Read the full abstract ▾
Statistical Methodology
Beyond Local Independence: High-Dimensional Latent Class Graphical Models with Shared Block Structure
Seunghyun Lee and Yuqi Gu
Latent class models are central tools for multivariate categorical data from heterogeneous populations, but their standard local-independence assumption is often unrealistic in modern high-dimensional applications. We propose a high-dimensional latent class graphical model for ordinal responses with block-structured local dependence. The model retains the interpretability and parsimony of classical latent class analysis by imposing a shared block partition of variables, while allowing class-specific graphical dependence within each block. We develop a scalable three-step
estimator that first recovers latent classes by spectral clustering of a flattened response
matrix, then estimates class-specific latent covariance
matrices and aggregates them to recover the shared block partition, and finally estimates sparse within-block precision
matrices. We establish finite-sample error bounds for clustering, covariance estimation, block recovery, and precision-
matrix estimation, yielding end-to-end consistency of all model components under high-dimensional scaling. Simulations demonstrate accurate recovery of latent classes, the shared block partition, and class-specific dependence graphs with scalable computation. Applications to American National Election Studies survey data and HapMap3 genotype data show that the method uncovers interpretable local dependence structures while accounting for latent heterogeneity.
Read the full abstract ▾
Machine Learning (Statistics)
Bidirectional Autoregressive Latent Diffusion for Forward and Inverse Magnetohydrodynamics
Alexander Scheinker
This work presents a new bidirectional autoregressive latent
diffusion approach for predicting the
evolution of multiple fields (mass density, pressure, velocity, and
magnetic field components) for magnetohydrodynamics. We show that this bidirectional flow can be used as a self-supervised consistency metric for uncertainty and error estimation, which enables the model to estimate test-time uncertainty and error without access to ground truth, by comparing how closely flowing forwards and backwards in time returns to the same predicted fields. We also demonstrate this methods's potential to serve as a non-invasive
plasma diagnostic, and show how adaptive feedback can be used to make the model more robust based on sparse diagnostics or limited views/measurements.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
How AI settled the complexity of the oldest SGD algorithm
Michał Dereziński and Xiaoyu Dong
In 1937, Stefan Kaczmarz proposed a simple
algorithm for solving systems of linear equations. This
algorithm turned out to be the earliest known example of
stochastic gradient descent, a ubiquitous computing paradigm that drives the training of modern AI models such as ChatGPT and Gemini. Now, those AI models have joined forces to discover the worst-case complexity of the Kaczmarz
algorithm. This paper tells the story of how it happened.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Optimizer Memory Makes Shuffle Order a First-Order Source of Fine-Tuning Noise
John Sweeney
Shuffle order can be a larger source of fine-tuning noise than a memoryless analysis predicts: fixed-clock optimizer
memory makes local equal-multiset contrasts first order in the learning rate rather than second order, and the resulting order channel can be large enough for a single seed to flip a close A/B comparison. We isolate this mechanism and derive a fit-free way to size the noise it produces. For a memoryless optimizer, reordering an equal multiset has no first-order endpoint term; the leading local contrast is the $O(\eta^2)$ gradient bracket. Fixed-clock optimizers such as AdamW are different. Their moment buffers, preconditioner state, and de-biasing counters advance with the step index rather than with the learning-rate-scaled time $\tau=\eta k$, so the same gradient can receive a position-dependent endpoint weight. For any fixed finite measurement window, a lifted-state expansion gives an $O(\eta)$ equal-multiset contrast whenever the first-order replay coefficient is nonzero, while regular and clock-matched controls remain $O(\eta^2)$; a bare fixed-$\beta$ momentum buffer is already enough. A bitwise-deterministic replay from one warmed optimizer state isolates the mechanism, giving order-variance slopes 1.83 for AdamW, 2.00 for fixed-$\beta$ momentum, and 4.00 for SGD; matching the
memory clock to $\tau$ restores the regular exponent. For AdamW with a frozen preconditioner, the same impulse-weight kernel gives a closed-form asymptotic order-variance floor after the local potentials are measured, with no fitted coefficients. The result is local to the measurement window (independent reshuffling can average the channel across windows), but it yields order-noise error bars, positional attribution weights, and a seed-budget criterion for fine-tuning comparisons.
Read the full abstract ▾
Statistical Methodology
Modelling and detecting mild and gross anomalies in circular data via double-contaminated models
Antonio Punzo, Andriëtte Bekker, Arno Otto et al. (+2 more)
In this paper, we propose a model-based framework to robustify inference for circular data in the presence of anomalous observations, distinguishing between mild and gross anomalies. Starting from a unimodal and symmetric reference model on $[0,2\pi)$, parametrized by a mean direction and concentration, we construct a family of finite mixtures: a gross-anomaly model obtained by adding a circular uniform component; a mild-anomaly (contaminated) model obtained by mixing the reference distribution with a less concentrated version sharing the same mean direction; and a general three-component specification combining both models, the double-contaminated model. Posterior component probabilities provide an automatic classification of observations without ad hoc thresholds, while the mixing weights yield interpretable measures of anomaly prevalence and dispersion inflation. For illustration, we consider two classical circular reference distributions, the wrapped normal and von Mises. The methodology is evaluated through an extensive simulation study and three real-data applications involving animal movement directions and wind directions. The results indicate that jointly modelling mild and gross departures improves model fit and yields an informative decomposition of the directional data, demonstrating that mixture-based robustness is valuable not only for anomaly detection but also for the interpretation and the identification of latent structure in directional data.
Read the full abstract ▾
All authors Antonio Punzo, Andriëtte Bekker, Arno Otto, Priyanka Nagar, Cristina Tortora
Machine Learning (Statistics) · Cross-listed from cs.LG
Not All Objectives Are Born Equal: Priority-Constrained Descent for Hierarchical Multi-Objective Optimization
Dara Varam and Mohamed I. Alhajri
Deep learning problems rarely involve objectives that are equal in importance. A primary objective defines the goal, whilst secondary objectives, such as sparsity, compression, or robustness constrain the solution. While existing multi-objective methods have proven effective in practice, they have a clear
symmetry problem and neglect the inherent objective hierarchy built into these objective spaces. We introduce Priority-Constrained Descent (PCD), a gradient-based optimization framework designed to explicitly exploit hierarchical objective structures. PCD preserves the direction of primary descent whilst allowing for the minimal distortion necessary to guarantee progress on secondary objectives, controlled by a single $\tau \in [0, 1]$ that dictates the strength of the distortion. The resulting formulation is invariant to objective scaling and admits exact closed-form solutions for problems with two and three objectives. We evaluate PCD within structured network compression settings, unstructured sparsity and low-rankness, and across a variety of synthetic experiments, showing Pareto dominance and better per-objective performance with secondary progress guarantees over existing methods, further exhibiting the interpretable trade-off that $\tau$ provides.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
A Mathematical Optimization Approach for Expert-Informed Bayesian Best Subset Selection
Nolan Alexander and Henning Mortveit
A central challenge in statistical modeling is identifying the subset of features that belong in the true regression model. The classical best subset selection problem, recently made tractable via mixed-integer optimization (MIO), finds the globally optimal sparse solution. It does not, however, make use of any information beyond the observed data. In many applied settings, domain experts can meaningfully rank or score the relevance of candidate predictors, yet no existing framework integrates such probabilistic expert assessments directly into the best-subsets objective. This paper presents Expert-Implied
Bayesian Best Subsets (EBBS), a method that incorporates domain-expert probability estimates of feature relevance into the MIO best-subsets problem through a maximum a posteriori (MAP) framework. Expert views from multiple respondents are aggregated into a single prior probability per feature using the Poisson binomial distribution for marginal probability estimates, the pairwise win rate for pairwise comparisons, or the normalized mean rank for ordinal rankings. This probability enters the objective function as a log-odds penalty term that smoothly encourages or discourages the selection of each feature consistent with the expert consensus. This paper provides analytic derivations of the MAP formulation and characterizes its theoretical properties. The proposed model reduces to Best Subsets when experts all have no views. Empirical results on synthetic and real
datasets are forthcoming.
Read the full abstract ▾
Statistical Methodology
Scalable Bayesian Spatial Mixture Modelling for Remote Sensing Image Segmentation
Bao Khanh Nguyen, Iain Cameron, Cecilia Balocchi et al. (+1 more)
Accurate and scalable land cover classification is essential for global conservation monitoring and policy-making. While remote sensing images provide a cost-effective alternative to ground surveys, current methods often lack principled uncertainty quantification and require substantial labelled data, limiting their usability and reliability in new regions with distribution shifts. We propose a
Bayesian spatial mixture modelling approach for image
segmentation, extending the classical Potts model by allowing for a generalised spatial dependence structure and incorporating informative priors estimated from pre-existing labelled data. Our framework, called POTTERS (Potts Model for Enhanced Remote Sensing), enables robust uncertainty quantification, accounts for class interactions, and can detect new clusters in the target region of interest. Crucially, our model does not require labelled data from the target region; instead, it incorporates prior information about the labels from pre-existing externally labelled images. To ensure scalability to large remote sensing images, we develop an efficient variational inference
algorithm for posterior approximation. We demonstrate the benefits of our approach in simulation studies and apply it to land cover classification in a case study in Scotland, leveraging publicly available remote sensing data from England.
Read the full abstract ▾
All authors Bao Khanh Nguyen, Iain Cameron, Cecilia Balocchi, Torben Sell
Machine Learning (Statistics)
Self-Organized Conformal Prediction: Reducing Regional Coverage Gaps with Unsupervised Group Discovery
Louis Berthier, Ahmed Shokry, Maxime Moreaud et al. (+2 more)
Conformal prediction guarantees marginal coverage, but a pooled calibration
quantile can hide systematic undercoverage across heterogeneous regions of the feature space. We introduce Self-Organized Conformal Prediction (SOCP), a calibration scheme that discovers input-space groups with an unsupervised Self-Organizing Map (SOM) trained without calibration data. At prediction time, the query's best-matching unit (BMU) retrieves a calibration buffer from one
cell, a fixed grid neighborhood, or a prototype enlargement (Regimes 1 to 3). When fixed neighborhoods are too sparse, Regime 3 adds
cells by prototype distance, using a global budget selected from training
cell occupancies and the planned calibration size before any calibration score is observed. The predictor and nonconformity score remain unchanged.
Cell-only
retrieval has exact
cell-conditional validity, and each fixed union of
cells has exact retrieved-set validity. Interpreting a neighborhood threshold at its central
cell incurs an explicit Kolmogorov-Smirnov (KS) bias term. Across ten regression and classification
benchmarks, SOCP lowers the mean weighted coverage gap (WCovGap) relative to pooled split conformal prediction (SCP) on all ten
datasets. The mean relative change is -14.8%, with a mean increase of 3.0% in output size. Under fixed-neighborhood
retrieval, SO calibration lowers mean WCovGap across ten seeds in 43 of 50 score and
dataset comparisons, while SO-SCP lowers it for every
dataset at all three tested external partition granularities. These results provide local group calibration without supervised partitions or predictor retraining, while the diagnostics make the cost and limits of locality explicit. The code is available at https://github.com/LouisTier/SOCP.
Read the full abstract ▾
All authors Louis Berthier, Ahmed Shokry, Maxime Moreaud, Guillaume Ramelet, Aymeric Dieuleveut
Statistics Applications
Bayesian Copula Directional Dependence is Cross-Network Robust for Gene-Regulatory Pair Direction: A Benchmark Study on DREAM5
Xiaoying Wei and Clara Grazian
Inferring the direction of a gene-regulatory relationship is harder than inferring whether a relationship exists, and most direction-inference methods are validated mainly on a single in silico
benchmark. We ask which method remains reliable as the data move from a synthetic network to real organisms and as sample size decreases. We embed a copula-based measure of directional dependence (CDD) in a
Bayesian framework that returns, for each candidate pair, a posterior distribution over a directional contrast, a 95% credible interval, a posterior sign-support score, and a principled no-call. We
benchmark this
estimator against eight direction-inference methods, including two
Bayesian DAG-posterior baselines, on the three core DREAM5 networks (in silico, S. aureus, and E. coli), with S. cerevisiae used as an out-of-domain eukaryotic stress test. Across the three core networks,
Bayesian CDD is the only method whose called accuracy is always above 60%, whose coverage is always above 88%, and whose direction AUROC is always above 0.6; every competing method falls to chance or below on at least one network. CDD ranks first on both real-organism networks, remains stable on the smallest-sample network where bootstrap-interval methods collapse, and is the only
Bayesian method that is simultaneously above chance and high-coverage under a 95% posterior gate. We position CDD as a post-screening, uncertainty-aware direction-refinement tool for candidate regulatory pairs.
Read the full abstract ▾
Statistics Applications
Critique of "Use of roster charts in the investigation and prosecution of nurses ..." by John O' Quigley
Richard D. Gill
The paper "Use of roster charts in the investigation and prosecution of nurses suspected of inflicting deliberate harm on patients" by Prof. John O'Quigley explores an interesting hypothesis concerning statistical information hidden in the part of the infamous Lucy Letby roster chart pertaining to the 37 other nurses. Unfortunately, we have to point out some serious errors in his statistical analyses. The data actually contains information which strongly disproves his main modelling assumption. We do, however, strongly agree with him that from a forensic statistical point of view, the roster chart is fake evidence which should not have been shown to jurors.
Read the full abstract ▾
Statistics Applications
Semantic insurance pricing with large language models
Christopher Blier-Wong and Derek Kusmenko
Classical actuarial pricing models, such as the generalized linear model, are valued for transparency and ease of governance, but they use interactions among risk factors only when these are supplied through explicit feature engineering. We study whether
embeddings from a pre-trained large
language model, computed from a natural-
language description of each policyholder, can replace hand-crafted features as inputs to a standard actuarial pricing model, taking Poisson claim-frequency regression as the main example. The
language model is used only to construct deterministic
embedding covariates; pricing is performed by a standard generalized linear model. Using French motor third-party liability data, the
embedding-based model outperforms the generalized linear model, especially when data are scarce, whereas at larger sample sizes the comparison is model- and dimension-dependent. Insurance-specific fine-tuning further improves the
embeddings, and a
prompt-sensitivity diagnostic shows that the pipeline reacts to any appended out-of-template field, making controlled
prompts a governance requirement.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Sample Complexity of Scientific Discovery: PAC Learnability of Compositional Function Trees
Şuayp Talha Kocabay, Talha Rüzgar Akkuş and Kerem Yalç{\i}n
Scientific discovery via symbolic regression is often viewed as statistically and computationally intractable because the hypothesis space of expressions grows combinatorially with depth. This paper revisits the statistical side through the lens of PAC learning, focusing on compositional function trees built from a finite vocabulary of smooth operators (e.g., $\{+,\times,\sin,\exp\}$ and affine maps). We prove that the relevant generalization quantity, Rademacher complexity, hence the excess risk, does not necessarily blow up exponentially with the number of distinct symbolic structures, but is controlled by (i) the depth $d$ and (ii) the Lipschitz constants of the base operators along the composed computation graph. Concretely, under mild Lipschitz conditions on operators and bounded affine leaves, a finite-union bound over a vocabulary of size $K=|\mathcal{H}_{\mathrm{base}}|$ together with Maurer-type vector contraction yields $\mathfrak{R}_n(\mathcal{H}_{\mathrm{comp}}^{d}) \leq (Kb\sqrt{2}L)^{d-1}\mathfrak{R}_n(\mathcal{H}_{\mathrm{comp}}^{1})$ with arity bound $b$; corresponding high-probability risk bounds scale as $\mathcal{O}(L^{d}/\sqrt{n})$ when $K,b=O(1)$ and $\mathfrak{R}_n(\mathcal{H}_{\mathrm{comp}}^{1})=O(n^{-1/2})$. We complement the theory with a modular codebase that trains differentiable operator trees (not MLPs) on synthetic "physics-like" targets of controlled depth and shows that the empirical generalization gap correlates positively with the predicted complexity term $(\widehat{L}^{d})/\sqrt{n}$.
Read the full abstract ▾
Machine Learning (Statistics)
Gradient boosting with vector-valued leafs
David Cortes
Gradient boosting in the form of decision tree ensembles has successfully been applied to a variety of problems using simple objective functions based on log-likelihoods of a single variable. The concept extends naturally to objective functions operating on vectors - for example, multinomial logistic log-likelihood for multi-class classification, where observations have a score for each class - but popular frameworks approach these functions by either updating one value of the input vectors at a time, or by using a diagonal upper bound on the second derivative. This work extends the usual gradient boosting framework to functions of vector inputs and sketches a simple
algorithm that can be used efficiently with histogram-based decision trees.
Read the full abstract ▾
Machine Learning (Statistics)
Generalization Analysis of Transformers in Distribution Regression
Peilin Liu and Ding-Xuan Zhou
In recent years, models based on the
Transformer architecture have seen widespread applications and have become one of the core tools in the field of deep learning. Numerous successful techniques, such as parameter-efficient fine-tuning and efficient scaling, have been proposed surrounding their applications to further enhance performance. However, the success of these strategies has always lacked the support of rigorous mathematical theory. To study the underlying mechanisms behind
Transformers and related techniques, we first propose a
Transformer learning framework motivated by distribution regression, with distributions being inputs, connect a two-stage sampling process with natural
language processing, and present a mathematical formulation of the
attention mechanism called
attention operator. We demonstrate that by the
attention operator,
Transformers can compress distributions into function representations without loss of information. Moreover, with the advantages of our novel
attention operator,
Transformers exhibit a stronger capability to learn functionals with more complex structures than convolutional neural networks and fully connected networks. Finally, we obtain a generalization bound within the distribution regression framework. Through the aforementioned theoretical results, we further discuss some successful techniques emerging with large
language models (LLMs), such as
prompt tuning, parameter-efficient fine-tuning, and efficient scaling. We also provide theoretical insights behind these techniques within our novel analysis framework.
Read the full abstract ▾
Statistical Methodology · Cross-listed from cs.LG
When Prices Double in a Week: Forecasting of Agricultural Volatility in Import-Isolated Markets
Ranuga Weerasekara, Heshan Nethmina, Manuja Ranathunga et al. (+6 more)
Vegetable prices in Sri Lanka are highly volatile because the
market is largely import-isolated, so supply disruptions quickly drive prices up. This study develops a machine learning framework to forecast such
volatility by incorporating supply-chain-aware features and explicitly modelling the country's two cultivation seasons, Maha (October-April) and Yala (May-September). An integrated
dataset was constructed by combining retail and farmer-gate prices with origin-aligned weather variables, diesel costs, and exchange rates across 12 vegetable varieties and 14
market centres from 2013 to 2019. A gradient-boosted ensemble model (XGBoost and LightGBM) was trained and optimised using Optuna, and unified and season-specific configurations were compared. Results show that season-specific models improve within-season fit, with the Yala-specific model achieving the highest R2 of 0.9420 (95% CI [0.690, 1.000]), while the unified model delivers the best overall predictive accuracy of 90.84% (95% CI [88.34%, 91.52%]) and an R2 of 0.9281 (95% CI [0.760, 1.000]). Notably, the unified model maintains 85.96% accuracy on a completely unseen 2024 hyperinflationary period without retraining, successfully tracking major price surges. These findings suggest that agricultural price movements in import-constrained
markets are meaningfully predictable when models capture supply-chain dynamics, offering practical value for early warning and decision making by farmers, traders, and policymakers. Existing studies on Sri Lankan vegetable prices are confined to Autoregressive Integrated Moving Average (ARIMA) and Generalized Autoregressive Conditional Heteroskedasticity (GARCH) applied to single
markets, with no supply-chain features, seasonal
segmentation, or cross-regime validation.
Read the full abstract ▾
All authors Ranuga Weerasekara, Heshan Nethmina, Manuja Ranathunga, Vinma Wettasinghe, Dinithi Navodya, Subavarshana Arumugam, Nirasha Munasinghe, Nisansa de Silva, Sandareka Wickramanayake
Statistical Computation
Using Variational Inference to Improve the Efficiency of MCMC Algorithms
Pingping Yin and Xiyun Jiao
Bayesian statistics makes inference based on Bayes'
theorem, but the posterior distribution of unknown parameters is typically analytically intractable. To estimate the posterior, two widely used numerical approximation methods are Markov Chain Monte Carlo (MCMC) and variational inference (VI). MCMC methods produce asymptotically exact samples but are computationally intensive, while VI methods are faster and more scalable but may lack accuracy. This paper proposes combining MCMC and VI to construct
algorithms that leverage the strengths of both. The first proposed
algorithm uses Gaussian variational inference (GVI) with various covariance structures to derive a linear transformation
matrix for
Hamiltonian Monte Carlo (HMC). This method improves the efficiency of HMC, particularly in high-dimensional and complex target distributions. The second
algorithm combines a VI-based generative model, the variational auto-encoder (VAE), with the Metropolis-Hastings (MH) sampler. The resulting VAE-MH sampler is efficient and effectively traverses the parameter space, outperforming standard MCMC methods in identifying all modes of multi-modal distributions.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Bayesian Best-Arm Identification with Abstention: A Polynomial-to-Exponential Phase Transition
Yuqi Huang, Yunlong Hou and Vincent Y. F. Tan
We study the
Bayesian fixed-budget best-arm identification problem in which a learner can abstain from making a terminal recommendation. Subject to an abstention budget $\alpha$, we analyze the probability of undetected error–the risk of recommending a suboptimal arm without abstaining. Our central finding is that abstention induces a phase transition: without abstention, the error probability decays polynomially in the sampling budget $T$; in contrast, introducing any small positive abstention budget shifts this to an exponential decay. For Gaussian priors and rewards, in the regime $T\to\infty$ followed by $\alpha\downarrow0$, we establish exact matching information-theoretic lower bounds and algorithmic upper bounds on the optimal error exponent, which takes the form $\exp(-\frac{\alpha^{2}T}{8\kappa_{\nu}^{2}})$. The hardness parameter $\kappa_{\nu}$ represents the prior density of the top-two gap at zero, highlighting that nearly tied instances drive the fundamental error. We introduce an adaptive
algorithm, PGWS, that successfully achieves this optimal exponent by expending its abstention budget on statistically ambiguous instances. We further demonstrate that this
polynomial-to-exponential improvement is exclusively a
Bayesian phenomenon–in the frequentist setting, abstention only affects lower-order exponent terms. We also extend our results beyond the Gaussian model.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Dead-Direction Conditioners: Gauge-Equivariant Preconditioning for Deep Networks
Tejas Pradeep Shirodkar
A deep network's loss is invariant to continuous
symmetries of its parameters: the logit shift, the ReLU rescaling, the LayerNorm scale, the per-head
attention rotation. Adam's per-coordinate preconditioner drifts along each
symmetry orbit, which pulls the trajectory off the
symmetry quotient where the optimization lives and blurs the singular-learning rate the quotient makes readable. We build DDC, a Dead-Direction Conditioner that lifts a base optimizer into a $G$-equivariant one: it conditions the optimizer's state in the orbit decomposition of a $G$-invariant metric, so the trajectory stays a preconditioned gradient flow on the quotient $\bar\Theta = \Theta/G$. The construction carries four architectural gauges (cross-
entropy shift, ReLU and SwiGLU rescaling, LayerNorm and RMSNorm scale, and a per-head $O(d_{\rm head})$
attention rotation matched to RoPE), proves exactly equivariant on an Adam base, and composes with a
Muon base through a gauge-equivariant orthogonaliser. Respecting the
symmetry changes both the minimum the optimizer reaches and what it leaves measurable there. On a
language model trained past the point of fit, DDCAdam resists the over-training collapse AdamW falls into, holding a validation-train loss gap of 0.67 against 5.88, and reads the dead-direction rate in 32 of 65 layer-by-observable
cells where AdamW reads it in 7. A
vision transformer trained from scratch reaches lower validation loss (1.71 against 2.12) while compressing spare feed-forward capacity a matched AdamW leaves intact. On a
Muon base, where the rotation gauge composes exactly, DDCMuon groks ten of eleven seeds at depth 24 that a plain
Muon never reaches. Built into the optimizer, a network's gauge
symmetry sharpens the minimum it finds and turns that minimum's geometry into something the trajectory can measure.
Read the full abstract ▾