Machine Learning (Statistics) · Cross-listed from cs.LG
ShamAN-Q: Shampoo Augmented NanoQuant for Sub-1-bit LLM Weights
Jonathan Mei, Sang Hyub Kim, Oliver Knitter et al. (+2 more)
We introduce ShamAN-Q, a sub-1-bit post-training quantization method that extends NanoQuant by replacing each its diagonal reconstruction geometry with a tractable dense
curvature metric, using a general paradigm popularized by the Shampoo optimizer. For each linear weight, ShamAN-Q fits a Kronecker product to the empirical Fisher information
matrix of a small calibration set by Kullback–Leibler minimization, forming a Mahalanobis reconstruction loss from the result. The continuous ADMM updates from NanoQuant become solutions to Sylvester equations, while its discrete projection and deployment format remain unchanged. Because the
curvature is local to a given set of weights, ShamAN-Q re-measures the input
curvature statistic for each layer immediately before layer factorization, periodically refreshing all statistics on the partially quantized model. ShamAN-Q also redistributes the uniform rank from NanoQuant across layers at the same total number of bits. On Qwen3-Base, ShamAN-Q lowers WikiText-2 perplexity at $\approx$1 bpw from 27.56 to 22.96 (0.6B), 19.21 to 16.72 (1.7B), and 14.29 to 13.80 (4B) while matching or improving zero-shot accuracy on the Eleuther LM Evaluation Harness. On 0.6B, ShamAN-Q at $\approx$0.8 bpw matches the published perplexity of NanoQuant at $\approx$1.0 bpw.
Read the full abstract ▾
All authors Jonathan Mei, Sang Hyub Kim, Oliver Knitter, Chi Chen, Martin Roetteler
Statistics Theory · Cross-listed from cs.LG
Grokking through the Lens of Minimum-Norm Interpolation
Gil Kur, Ileana Rugina, Clémentine Carla Juliette Dominé et al. (+1 more)
Grokking shows that fitting the training data and learning the underlying signal can occur at very different stages. However, existing theories offer limited quantitative insight into how this delayed generalization depends on inductive bias and signal structure. Our work addresses the gap by developing a statistical theory that characterizes how regularization geometry and signal sparsity govern generalization near interpolation. In particular, we focus on the prototypical setting of high-dimensional regression and identify regimes in which sparsity-promoting regularization makes exact interpolation much more accurate than approximate fitting. In strongly overparameterized noiseless problems, we prove a zero–one generalization law and construct a family of convex norms whose interpolators transition from the trivial risk of the all-zero predictor to exact recovery, while keeping the training error equal to $0$. Furthermore, when feature dimension and sample size are proportional, we provide a precise characterization of training and generalization errors along $\ell_r$-regularization paths. This in turn allows us to quantify the generalization gain that remains near interpolation: we show that this gain increases as the norm becomes more sparsity-promoting and as the target becomes sparser, with a sharp drop in generalization reached for noiseless data and $\ell_1$ regularization. Experiments on diagonal linear networks and
transformers trained on modular arithmetic demonstrate the generality of our theoretical predictions. Finally, beyond grokking, our work reveals a statistical instability in minimum-norm interpolation: small perturbations in the regularization strength can lead to drastically different generalization, while preserving small training error.
Read the full abstract ▾
All authors Gil Kur, Ileana Rugina, Clémentine Carla Juliette Dominé, Marco Mondelli
Statistics Applications · Cross-listed from cs.LG
An Input-Frugal Deep Learning Framework for Weather-Driven National Crop-Yield Forecasting: A Case Study of Brazilian Soybean
Fernando Dupin da Cunha Mello, Prashant Kumar and Erick G. Sperandio Nascimento
Reliable, timely crop-yield forecasts are essential for
market stability and risk management, yet many approaches rely on costly or hard-to-scale inputs. We present a frugal, transferable, and architecture-agnostic deep learning framework that uses routine weather as the only time-varying input plus two lightweight static context inputs (crop year and an agro-environmental label) to capture long-run change and regional heterogeneity, while supporting multiple sequence encoders under identical data requirements. Using a 20-season Brazilian soybean case study (2001/02-2020/21) with leave-one-year-out cross-validation, we
benchmark MLP, CNN, LSTM, CNN-LSTM, a
Transformer encoder and the Mamba state-space model against linear ridge regression and a five-year moving-average "farmer" baseline. All deep learning variants outperform ridge, and all sequential encoders surpass the non-sequential MLP. The
Transformer achieves the best national accuracy (RMSE 149 kg ha^-1; rRMSE 5.3%; R^2 = 0.784), reducing error by 47.6% relative to the farmer baseline. In-season forecasts improve monotonically from early- to late-season issuance, reaching approximately 50% lower error than the baseline at the latest forecast point. Ablations indicate that the agro-environmental label and spatial instance expansion (multiple grid-node weather sequences per municipality-year) contribute positively without increasing input complexity. SHAP diagnostics suggest crop year explains most of the long-run trajectory, whereas within-season weather and agro-environmental context primarily drive interannual deviations, with moisture/cloud and thermal-demand variables dominating. Overall, the framework is straightforward to deploy across other crops and geographic regions and is naturally compatible with operational weather forecasts for routine monitoring.
Read the full abstract ▾
Statistics Theory
Local polynomial density ratio estimation
Hajo Holzmann and Alexander Meister
We propose a novel local-
polynomial estimator of the ratio $r=f/g$ of two $d$-dimensional densities $f$ and $g$, from which independent samples are available. The
estimator is shown to achieve pointwise minimax optimal rates over Hölder classes of arbitrary smoothness index without additional logarithmic factors, and with smoothness being only assumed of $r$ but not of $f$ nor $g$. Our analysis remains valid for points on the boundary of the support of the distribution associated to $g$ under a mild
geometric assumption on the (unknown) boundary. We also derive a concentration inequality for the
estimator, which can be useful in applications to classification, and give a rate in the supremum norm, where the supremum is also taken over boundary support points. The smoothness class over which the rate is obtained is sufficiently large that the individual densities $f$ and $g$ cannot be consistently estimated uniformly over this class. Direct
estimators of the partial derivatives of $r$ together with rates of convergence are also provided. We also obtain asymptotic normality of the
estimators under quite general assumptions and again including boundary points, with a consistent variance
estimator allowing for data-driven studentization. Moreover, we show how our
estimator can be used to estimate the Kullback-Leibler information, in a construction which additionally uses debiasing. We provide the parametric rate together with asymptotic normality under sufficient smoothness of $r$ relative to the dimension $d$.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from quant-ph
Advantage of Sample Complexity in Quantum PAC Learning Requires Inverse Access to State-Preparation Unitaries
Natsuto Isogai, Satoshi Yoshida and Mio Murao
Whether
quantum computation can reduce the amount of data sampled from an unknown probability distribution required to learn a prediction rule is a fundamental question in
quantum machine learning.
Quantum PAC learning studies this question using
quantum data as a
quantum state whose squared amplitudes encode the unknown distribution from which classical learning data are sampled. With only copies of such
quantum data, the optimal worst-case sample complexity asymptotically matches that of classical PAC learning. In contrast, access to both a state-preparation
unitary for this state and its inverse can improve the query-complexity dependence on the accuracy parameter in realizable learning. However, it has remained unclear whether forward-only access allows such an improvement. In this work, taking the worst case over compatible state-preparation
unitaries and their finite ambient dimensions, we show that the optimal forward-only query complexities of realizable and agnostic learning are, respectively, $\Theta((d+\log(1/\delta))/\varepsilon)$ and $\Theta((d+\log(1/\delta))/\varepsilon^2)$, where $d$ is the VC dimension of the concept class, $\varepsilon$ the accuracy parameter, and $\delta$ the failure probability. These bounds match the optimal sample complexities with classical data or
quantum data copies. To prove them, we establish a reduction using $q$ copies of the prepared state to approximate the Haar-averaged output of any $q$-query forward-only
algorithm. These results show that forward-only access cannot provide an asymptotic query-complexity advantage over learning from classical data or
quantum data copies in this worst-case setting, and establish the essential role of inverse access in the known realizable-setting improvement. Our reduction also provides a new framework for analyzing limitations of forward state-preparation access via state-copy lower bounds.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from astro-ph.IM
PTED: A multi-dimensional two-sample test for scientific inference and generative machine learning
Connor Stone
Two-sample tests are widely applicable in inference and generative modelling, yet users frequently fall back on heuristics and visual inspection due to lack of an accessible test that operates in multiple dimensions. I present Permutation Test using the Energy Distance (PTED), a Python implementation of a powerful two-sample test described in Székely and Rizzo [2004]. The test statistic is the energy distance, a metric on probability distributions that admits a natural sample
estimator built from pairwise distances. A permutation test is run on energy distances to produce an exact two-sample test. Because the statistic depends on the data only through pairwise distances, the test applies in high dimensions, on learned feature representations, at large or small or imbalanced sample sizes, and to any data type on which a distance can be defined. An approximation to the formula allows PTED to scale linearly with both number of dimensions and samples while retaining most discriminative power. I highlight a number of instructive sensitivity tests on PTED alongside other multi-dimensional two-sample tests for comparison. PTED is the only method that is sensitive to all tests, though conventional Maximum Mean Discrepancy is identical for all but an over-fitting test.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Learning to Plan from Random Exploration
Deqian Kong, Guangyan Sun, Sheng Cheng et al. (+6 more)
Random exploration reveals how an environment can be traversed before a goal is specified. Can this experience support long-range planning without policy-improvement training? Our random-walk analysis explains what temporal relations contain: short horizons reveal geodesic geometry in the
diffusion limit, while longer horizons reveal connectivity between regions before mixing removes these distinctions. We learn these relations with a conditional energy-based model that estimates temporal log-density ratios through horizon-conditioned
embeddings. The model is trained on observation pairs by noise-contrastive estimation, without action or reward labels. The planner queries these learned relations at different horizons as it moves toward the goal. At test time, a separate local dynamics model predicts candidate action outcomes, and the temporal model evaluates their progress toward the goal by selecting or aggregating estimated improvements across horizons. The
agent executes one action and replans with both models fixed. Experiments demonstrate long-range maze planning from random exploration using states and images. Learned score fields,
embedding probes, and planned routes exhibit properties of a multiscale cognitive map. We further demonstrate egocentric navigation from random exploration and manipulation planning from suboptimal data.
Read the full abstract ▾
All authors Deqian Kong, Guangyan Sun, Sheng Cheng, Sirui Xie, Bo Pang, Jianwen Xie, Tony Geng, Caiwen Ding, Ying Nian Wu
Machine Learning (Statistics)
Lower Bounds for Linear-Oracle Online Learning
Mohit Sinha
Can a constant number of linear minimizations per round improve on the $T^{3/4}$ regret rate of online Frank-Wolfe on general convex sets? Weibel et al. conjectured that fixed-coefficient methods cannot. We prove their
conjecture and extend the lower bound to every deterministic learner in an oracle-only model. The learner receives an initial feasible point and a diameter bound, and must remain feasible on every domain consistent with its oracle replies. For $T$ rounds, at most $b$ calls between decisions, diameter bound $D$, and gradient norm bound $L$, we construct an instance in dimension $d=2b(T-1)+1$ with regret at least $2^{-1/4}LDb^{-1/4}T^{3/4}$. The adversary fixes the domain, initial point, deterministic tie rule and linear losses before play. The vertices form a path on which every point available before a decision has zero current loss, while the final vertex has negative loss on every round. For constant $b$, the result matches the known upper rate for dimension-independent guarantees. For one-call fixed schedules with a nonzero coefficient on the newest gradient, a second construction gives regret at least $3LDT^{3/4}/4$ with unique minimizers at every issued query. Exact-arithmetic certificates for the tuned schedule of Weibel et al. closely match their finite-horizon numerical worst cases, with unique oracle replies.
Read the full abstract ▾
Machine Learning (Statistics)
Acceleration of Diffusion Language Model through Discrete Average Generator
Yidong Ouyang, Zhengyan Wan, Themis Haris et al. (+8 more)
Discrete
diffusion models and flow matching have emerged as powerful frameworks for generative modeling over discrete state spaces, yet efficient few-step generation remains a fundamental challenge. In this work, we introduce the Discrete Average Generator, a principled extension of MeanFlow to Continuous-Time Markov Chains (CTMCs). Analogously to how MeanFlow defines an average velocity field over a time interval in continuous spaces, we define an average generator as the normalized increment of the transition kernel over a time interval. We show that this average generator satisfies a self-consistency identity, which provides the foundation for our training objective. We further develop training strategies that align with the standard training paradigm of
diffusion language models while keeping the resulting objective tractable. When projected onto per-coordinate marginals, the self-consistency identity admits a closed-form expression, enabling efficient training and inference. In Potts model simulations, our objective reduces the total variation distance of the $K$-step sampler by up to 67%. On OpenWebText, our method achieves the lowest generative perplexity among the evaluated methods for 8 to 64 sampling steps while enabling a $16\times$ acceleration, and achieves comparable performance to existing methods on ImageNet.
Read the full abstract ▾
All authors Yidong Ouyang, Zhengyan Wan, Themis Haris, Tian Tan, Liqian Peng, Henry Li, Ziqian Lin, Jianhang Chen, Maryam Karimzadehgan, Alec Go, George Michailidis
Statistical Methodology
Inference for Standard Trial Estimands after Data-Driven Subgroup Discovery
Larry F. Leon and Keaven M. Anderson
Data-driven subgroup identification is increasingly used in clinical trials, yet the effect reported for a discovered subgroup is typically the standard analysis: a Cox or generalized-linear-model treatment coefficient fitted within that subgroup. Selection by an effect-driven rule — rewarding large estimated effects subject to screening and size criteria — inflates the reported coefficient: a winner's curse invalidating naive intervals whether or not the identifier's internal estimand was de-biased by cross-fitting. We develop a procedure-agnostic framework for post-selection inference that separates discovery from reporting: forest search, difference-in-natural-parameters
estimators, and causal forests generate interpretable candidate subgroups; selection is aligned with the coefficient to be reported; and the target is the standard-analysis effect of the selected subgroup, the candidate family held fixed — a conditional, data-adaptive estimand. Resampling the search then reduces to jointly perturbing all candidate effects, yielding refit-free multiplier resampling with an infinitesimal-jackknife interval whose coverage we characterize under standard regularity and an explicit condition on the limiting competition. It matches the full bootstrap on a fixed family; on model-generated families it remains valid for the same conditional target, which the bootstrap reproduces only under further conditions. Two trial applications and matched simulations show mitigated selection bias and improved coverage.
Read the full abstract ▾
Statistical Methodology
Small-study effects on the hierarchical summary ROC curve: latent accuracy and threshold trends in meta-analysis of diagnostic test accuracy
Yuta Nonomiya and Kazuki Nishida
Diagnostic meta-analyses commonly assess small-study effects using the Deeks test, which tests for a study-size trend in the log diagnostic odds ratio. Under the hierarchical summary receiver operating characteristic (HSROC) model, this quantity reflects both latent accuracy and threshold unless the curve is symmetric. A threshold trend can therefore generate a log diagnostic odds ratio trend without a latent accuracy trend, or conceal an existing one. We develop a likelihood-ratio test of the latent accuracy trend under a common-shape binomial HSROC model, leaving the threshold trend unrestricted. Its null hypothesis states that the summary curve is unchanged across study sizes. A log diagnostic odds ratio test using the same binomial fit provides a comparison with a different null hypothesis. In simulations, the proposed test remained near its nominal level under its null and had higher power than Deeks with a latent accuracy trend alone. In a colorectal cancer review of faecal immunochemical tests, it detected a negative latent accuracy trend obscured on the log diagnostic odds ratio scale by an opposing threshold contribution. All three tests detected size trends for impedance plethysmography in deep-vein thrombosis, where the fitted curve was nearly symmetric. The proposed test distinguishes changes in the summary curve from changes in operating position.
Read the full abstract ▾
Statistics Applications · Cross-listed from cs.LG
Kinematic signatures of impairment: Detecting alcohol intoxication in e-scooter riders using sensor data and machine learning
Rahul Rajendra Pai, Marco Dozza, Alexander Rasch et al. (+2 more)
Alcohol intoxication is a leading contributor to fatal and severe-injured e-scooterist crashes. Current countermeasures, such as temporal restrictions or pre-ride cognitive screening, cannot continuously assess an e-scooterist's physical motor control or impairment in real time. We conducted a controlled experiment in which 25 participants rode an instrumented e-scooter through a test track while sober and at two targeted blood alcohol concentration levels (0.05% and 0.08%). The e-scooter was instrumented with a six-axis inertial measurement unit (IMU), and throttle and brake lever position sensors, all sampled at 100 Hz. Two complementary signal features were computed: normalised permutation
entropy, which quantifies temporal complexity, and standard deviation, which quantifies signal amplitude. Repeated measures correlation identified seven kinematic features (all IMU and throttle signals) whose
entropy decreased (p < 0.001) while standard deviation increased (p < 0.01) with increasing intoxication, indicating that intoxicated riders shift from continuous, low-amplitude micro-corrections to fewer, high-amplitude reactive corrections. An
entropy based multi-class logistic regression classifier, evaluated through leave-one-participant-out cross-validation, achieved 85% overall accuracy and a weighted one-vs-rest area under the receiver operating characteristic curve (AuROC) of 0.94, with a sober-vs-high AuROC of 1.00. Steering rate and lateral acceleration were the most important predictive features, indicating that alcohol induces a distinct collapse in lateral
equilibrium during riding. Ultimately, these results demonstrate that onboard kinematic sensing combined with
entropy-based signal analysis can reliably distinguish sober from intoxicated e-scooter riding, providing a foundation for automatic intoxication detection systems that preserve mobility for sober riders.
Read the full abstract ▾
All authors Rahul Rajendra Pai, Marco Dozza, Alexander Rasch, Ali Mohammadi, Marco Capuccini
Statistics Theory · Cross-listed from math.PR
Learning the exogenous rate but not the distance to criticality in nearly unstable heavy-tailed Hawkes processes
Mauricio Herrera-Marín
In the nearly unstable heavy-tailed regime of Jaisson and Rosenbaum, where the intensity of a linear Hawkes process converges to a rough square-root Volterra process, we ask what a record of length $T$ reveals about the relative distance to criticality $1-\rho$ and the exogenous rate $\mu$. For the first, the information is governed by $\mathcal I_T=T\mu(1-\rho)$, not by the number of events; $\mathcal I_T$ stays bounded although the event count diverges like $T^{2\gamma}$, so no
estimator of the relative distance to criticality is uniformly locally consistent. For the second, the Fisher information diverges because the rough limit has an atom at zero: the asymptotic information geometry is singular in the exogenous direction and, unlike the light-tailed case, has no Feller-type threshold. We prove that the joint maximum likelihood
estimator of $(\mu,\rho)$, whose likelihood is jointly concave, recovers $\mu$ consistently at the random rate $\{Q_T/\log\log Q_T\}^{1/2}$ given by the observed information $Q_T$, while its estimate of the relative margin is only tight: the exogenous rate is learned although the endogenous parameter it is coupled to is not. For empty-start records $Q_T$ diverges because the rough limit started at zero spends positive time at zero; on windows observed long after the start, still with the branching ratio unknown, learnability holds with probability tending to one as the window grows, through the ergodicity of the rough limit and the atom of its stationary law. We also prove finite-dimensional convergence of the stationary intensity to the stationary rough Volterra process.
Read the full abstract ▾
Statistics Theory
Minimax and adaptive inference for branching ratios under long memory
Mauricio Herrera-Marín
The branching ratio $\rho$ of a self-exciting point process determines the branching margin $1-\rho$ to the critical boundary. We study inference on $\rho$ from a finite record for stationary Hawkes processes with completely monotone kernels whose slow relaxation mass obeys an envelope near zero rate. Slow excitation can then be exchanged for exogenous immigration; a spectral bound on the Kullback–Leibler rate and a coupling argument for the unobserved past give a minimax lower bound of order $T^{-\gamma/(2\gamma+1)}$ for a power-law envelope with tail exponent $\gamma$. A block-dispersion
estimator whose block length grows as $T^{1/(2\gamma+1)}$ attains this rate. A quantitative central limit
theorem, obtained from the cluster representation and a second-order Poincaré inequality on Poisson space, yields honest confidence intervals of nearly minimax expected length. For unknown $\gamma$, a Lepski-type rule adapts in probability at a logarithmic cost; free adaptation between
memory classes is impossible, and honest intervals cannot adapt to lighter
memory. Simulations are consistent with the rates.
Read the full abstract ▾
Machine Learning (Statistics)
Robust LassoNet: Enhancing Feature Selection in Neural Networks via Robust Loss Functions
Daniela De Canditiis, Italia De Feis and Paola Stolfi
Feature selection in neural networks remains a challenging problem, particularly in the presence of noisy or contaminated data. LassoNet is a recent approach that addresses this issue by combining neural networks with hierarchical sparsity constraints, enabling simultaneous prediction and variable selection. However, its standard formulation relies on the mean squared error (MSE) loss, which is known to be highly sensitive to outliers. In this paper, we present Robust LassoNet, an extension of LassoNet that incorporates robust loss functions, such as Huber, Cauchy, Tukey's bisquare, and Nonnegative Garrote, to mitigate the effect of extreme observations. The proposed approach preserves the original optimization framework while improving stability under data contamination. Through experiments on synthetic and real
datasets, we show that robust LassoNet significantly improves both predictive performance and feature selection accuracy in the presence of outliers or heavy tailed noise, while maintaining comparable performance in clean settings. These results highlight the importance of robustness in neural network based feature selection and suggest practical guidelines for choosing appropriate loss functions.
Read the full abstract ▾
Statistical Methodology
Proximal causal inference through cross-proxy balancing
Grace V. Ringlein, Trang Q. Nguyen, Elizabeth A. Stuart et al. (+1 more)
Proximal causal inference identifies causal effects in the presence of unmeasured confounding by drawing on two sets of proxy variables. Identification typically utilizes bridge functions, defined as solutions to integral equations. However, the mechanism by which fitted bridge functions correct for confounding bias remains opaque, offering little to interpret, inspect or stress-test. We show that the defining equation of a treatment bridge function is already a balance condition, with a cross-proxy form: the weights are functions of the treatment confounding proxies and
covariates, and they balance the outcome confounding proxies and
covariates (i.e., reweighting the distribution in a particular treatment arm to match the distribution across treatment arms). Several existing identification paths via a treatment bridge function can then be interpreted as providing conditions under which balance on the outcome proxies implies balance on the unobserved confounders, which we call balance propagation. Leveraging this framing, we show that estimation of the treatment bridge function is a type of balancing weight estimation. Finally, we show that an outcome-weighted
estimator form can also be obtained for a large class of proximal
estimators, including those that utilize an outcome bridge function. Using this framing, we provide conditions under which common
estimators are numerically equivalent.
Read the full abstract ▾
All authors Grace V. Ringlein, Trang Q. Nguyen, Elizabeth A. Stuart, Harsh Parikh
Statistics Theory
Goodness-of-fit for distributions on metric spaces
Diego Serrano, Eduardo García-Portugués and Ingrid Van Keilegom
We propose a general goodness-of-fit framework for distributions on separable metric spaces. Under suitable identifiability conditions, probability distributions are characterized by distance profiles, which motivates their use in goodness-of-fit testing, for simple and composite null hypotheses. For composite null hypotheses, parameter estimation is incorporated via a Bahadur-type expansion, and the asymptotic distribution of the empirical process for distance profiles is obtained under the null. We define test statistics based on this empirical process and derive their asymptotic null distributions. We further study the behavior of the proposed tests under fixed and local alternatives, establishing consistency results. Multiplier bootstrap procedures are developed, and their conditional asymptotic validity is established under both simple and composite null hypotheses. The methodology is illustrated with simulation studies and real-data applications for data on the sphere, hyperboloid, and simplex.
Read the full abstract ▾
Statistical Methodology
Global Synchronization for Multi-Source Data Integration under Blockwise Missing Patterns
Runbing Zheng and Dmitriy Kunisky
Multi-source data integration problems over
datasets from different sources covering different but possibly overlapping sets of entities have become increasingly important in many real-world areas, including genomics, single-
cell analysis, and healthcare research. In such problems, one often first learns a low-dimensional representation of the entities within each source and then integrates these representations across sources. As the representations from different sources are only identifiable up to some transformation, how to align them across sources using the sources' overlapping entities becomes a key challenge. Existing methods align the sources in a sequential or tree-structured manner, and are therefore sensitive to the chosen order and exploit only part of the available overlapping information. Motivated by this limitation, we propose Global Synchronized Multiple
Matrix Integration (GSMMI), which formulates this
alignment problem as a global synchronization problem and jointly aligns all sources using all pairwise overlaps at once, thereby making full use of all overlapping information across the sources. We develop an efficient iterative
algorithm for GSMMI that is fast and scalable to the large-scale data arising in these applications. We show both theoretically and empirically that GSMMI improves
alignment accuracy, with clear improvements even under modest overlap structure. Moreover, we develop GSMMI to be broadly applicable across data types, covering symmetric positive semidefinite, symmetric indefinite, and asymmetric or rectangular
matrices, and even settings where sources overlap only in their rows or only in their columns, making it suitable for a wide variety of application scenarios.
Read the full abstract ▾
Machine Learning (Statistics)
ReCIRC: Rectified Conformal Risk Control
Bruno Marcondes e Resende, Helton Graziadei, Thiago Rodrigo Ramos et al. (+1 more)
Many applications of black-box predictive models require controlling task-relevant error rates, such as missed lesion pixels in segmentation or missed labels in multilabel classification. Conformal risk control (CRC; Angelopoulos et al., arXiv:2208.02814) gives distribution-free guarantees for such losses, but it calibrates a single threshold shared by all inputs. Because conditional risk varies with the input, this marginal guarantee often overprotects easy cases and underprotects hard ones. We propose ReCIRC (Rectified Conformal Risk Control), which inverts each input's estimated local risk curve to reparameterize the calibrated threshold as a risk budget $a$ representing a common target conditional risk, and then applies CRC unchanged to the resulting family. ReCIRC retains CRC's finite-sample marginal guarantee regardless of the accuracy of the estimated curves, while accurate curves yield approximate conditional risk control and, under additional conditions, asymptotically exact conditional risk control; they also support a risk-calibration diagnostic. Across three synthetic and five real-data settings spanning segmentation, multilabel and multiclass classification, and regression, ReCIRC attained the lowest average worst-group risk and mean positive group excess in every setting, while maintaining marginal risk close to the target, whereas changes in prediction size were application-dependent.
Read the full abstract ▾
All authors Bruno Marcondes e Resende, Helton Graziadei, Thiago Rodrigo Ramos, Rafael Izbicki
Machine Learning (Statistics) · Cross-listed from cs.LG
Traversing the solution space of neural networks with Hessian Null Space Continuation
Ann Huang, Mitchell Ostrow, Zhouyang Lu et al. (+3 more)
On a single task, deep networks can learn many solutions, depending on their optimizer, training data, architecture, and hyperparameters. Many of these solutions are mode-connected: rather than isolated points in weight space, they are connected by low-loss regions. Yet how their internal computation varies within these regions is unknown. A parallel line of work has identified the degeneracy of neural representations: many networks reach similar training loss with distinct internal structures. However, it is unclear how these solutions are related in weight space. We unify these subfields and show for the first time that many different internal mechanisms exist within a local mode-connected region in weight space. To do so, we introduce Hessian Null Space Continuation (HNC), a scalable method that uses local
curvature to traverse regions of weight space that preserve network function, and can be steered toward solutions with specified properties. In RNNs trained on a
memory task, HNC reaches drastically different representations and dynamics with maintained behavior. In ImageNet-trained
Vision Transformers, HNC finds representations that differ more from the original network than any independently trained model with a different architecture or objective. In reinforcement-learning
agents, HNC uncovers a distinct navigation strategy at comparable return and exposes reward hacking in an AI Safety Gridworld. Finally, HNC measures the local geometry of the solution set, showing how model size and task complexity shape its dimension and functional sensitivity. Our results show that a surprisingly large amount of representational diversity exists near a single trained solution, unseen by standard gradient-based optimization. HNC identifies and quantifies this diversity, opening new possibilities for mechanistic understanding of solution spaces and for model merging, editing, and fine-tuning.
Read the full abstract ▾
All authors Ann Huang, Mitchell Ostrow, Zhouyang Lu, William T. Redman, Leo Kozachkov, Kanaka Rajan
Machine Learning (Statistics)
Latent Inference-Time Guidance of Time Series Foundation Models
Chloé Hashimoto-Cullen, Amaury Durand, Laurent Bozzi et al. (+3 more)
Time Series Foundation Models (TSFMs) currently provide state-of-the-art results in forecasting tasks. They are available out-of-the-box and rely on in-context learning to make their predictions, which makes the quality of their performance highly sensitive to the user-selected lookback,
covariates, horizon and training data distributions. In practise, the quality of the forecasts are variable but complementary, which highlights the need for a principled ensembling approach, rather than selecting the best context. This paper introduces Latent Inference-Time Guidance for TSFMs, which adaptively combines a pool of TSFM forecasts through a time-dependent latent space with independent components. The framework comes equipped with identifiability and reconstruction guarantees, whilst maintaining the off-the-shelf aspect of foundation models. We provide experiments on
datasets at various frequencies and from multiple domains: these show that the approach is competitive with traditional ensembling approaches.
Read the full abstract ▾
All authors Chloé Hashimoto-Cullen, Amaury Durand, Laurent Bozzi, Benjamin Guedj, Yannig Goude, Sylvain Le Corff
Machine Learning (Statistics) · Cross-listed from cs.LG
When do data mixtures improve scaling laws? Insights from high-dimensional regression
Diyuan Wu, Lehan Chen, Theodor Misiakiewicz et al. (+1 more)
Modern machine learning systems are trained on mixtures of data from different domains, and choosing the right mixture can substantially improve downstream performance. Despite an extensive literature on data mixing and reweighting, existing work is largely empirical and it remains unclear when auxiliary data genuinely improves scaling laws rather than merely providing more samples. To gain insight into this question, we study a high-dimensional mixed-data regression model with a shared regression function, heterogeneous covariances and noise levels, and
dataset sizes that may grow at different rates. We establish the minimax risk under an ellipsoidal parameter constraint for the general covariance structure and derive deterministic equivalents for the test error of ridge regression under commutative covariances. We then specialize to a target domain and an auxiliary domain with aligned power-law covariance spectra, where the theory yields explicit scaling laws in terms of spectral decay, target regularity, and the relative growth of the two
datasets. These laws identify regimes in which combining data mixtures provably yields a faster scaling rate than using either
dataset alone. In particular, improving the scaling law requires a specific interplay between spectra and relative sample sizes of the domains. Our numerical experiments on
language models exhibit the same qualitative phenomenon: appropriate data mixtures yield a faster decrease in target-domain test loss than training on either domain alone.
Read the full abstract ▾
All authors Diyuan Wu, Lehan Chen, Theodor Misiakiewicz, Marco Mondelli
Statistical Computation
Stochastic Variational Inference for Vine Copula Distributional Regression
Gianmarco Callegher and Thomas Kneib
Structured additive distributional regression flexibly relates all parameters of a conditional response distribution to
covariates, but multivariate extensions remain challenging when dependence is complex. We propose a multivariate structured additive distributional regression model based on regular vine copulas. Different vine edges may use different pair-copula families, while every pair-copula parameter may vary with
covariates through a structured additive predictor. The model therefore accommodates heterogeneous marginal distributions together with pair-specific asymmetric, tail-dependent, and
covariate-dependent dependence structures. For scalable inference, we develop a tree-wise
stochastic variational inference procedure based on component-specific Gaussian variational approximations. Marginal models are estimated first, followed by pair-copula regressions sequentially along the vine trees. We also adapt sequential vine selection by fitting candidate
covariate-dependent pair-copula regressions and using information criteria based on effective degrees of freedom for both family and tree selection. In simulations, the tree-wise
estimator remains close to an oracle using the true recursive conditional inputs, whereas global refinement yields more concentrated approximations and lower frequentist coverage for upstream components. An application to six-dimensional meteorological data from the Netherlands yields a vine combining Gaussian and non-Gaussian pair copulas, with pronounced nonlinear spatial and temporal variation in dependence.
Read the full abstract ▾
Statistics Theory · Cross-listed from cs.IT
Mutual Information Constrained Chernoff Bottleneck
Dier Tang and Guangyue Han
The classical information bottleneck (IB) measures the relevance of a representation $U$ of $X$ to a target $Y$ by $I(U;Y)$, which does not directly characterize the error of downstream decisions. For a binary hypothesis $Y$ inferred from many separately encoded observations, the optimal error exponent is the Chernoff information between the two conditional distributions of $U$ given $Y$. We study the mutual information constrained Chernoff bottleneck, which seeks an encoder that maximizes this Chernoff information subject to a rate constraint $I(U;X) \leq R$. We show that its optimal value $C(R)$ increases strictly up to $R = H(V)$, where $V$ merges the symbols of $X$ with equal likelihood ratio, remains at the uncompressed exponent beyond, and, unlike the IB curve, need not be concave. We further show that $k+1$ outputs suffice to attain $C(R)$, where $k$ is the cardinality of $V$. We propose an alternating
algorithm that updates the encoder via a generalized Blahut–Arimoto
algorithm and the Chernoff parameter $s$ via a nonlinear equation, and prove that its iterates remain feasible, with nondecreasing and convergent Chernoff information. Numerical experiments confirm the theory, and on real topic-detection data from the 20 Newsgroups corpus, compressing each word to only $17\%$ of its
entropy retains $90\%$ of the error exponent and nearly the accuracy of the uncompressed classifier.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from physics.flu-dyn
A shape-similarity latent space for fluid interfaces: invertible reduced-order modelling of droplet morphology
Ali R. Hashemi, Mohammad R. Hashemi and Pavel B. Ryzhakov
A droplet breaks up in tens of microseconds, and a recording captures perhaps a dozen frames. The states in between cannot be recovered without repeating the experiment, and simulating them is too costly to sweep an operating envelope. Yet they are present in the corpus as a whole: a campaign spanning a device's actuation range produces morphologies resembling those any single recording missed. Exploiting that requires a representation that is low-dimensional, invertible, and faithful to shape rather than to sampling. Proper orthogonal decomposition supplies the first two but measures distance in sampled coordinates;
manifold learning supplies the third but no map back to a shape; elastic shape analysis supplies a shape metric but no reduced coordinates. SHROM composes all three. Interfaces are represented by their square-root velocity functions, a neighbour graph is built over that shape space, and an autoencoder is trained to reconstruct while penalising latents in which graph neighbours are not latent neighbours. The demonstration uses 301,539 inkjet droplet contours and four filmed break-up sequences. The graph term does not improve reconstruction. It determines whether position in the latent carries meaning: clustering the latent of an otherwise identical model recovers the shape-space partition at chance level (ARI = 0.063 +/- 0.059), and at 0.781 +/- 0.049 with the term active. Waveform parameters predict the full contour at R^2 = 0.878 +/- 0.026. Negative results are reported in the same terms. The graph metric proved immaterial across five choices, and interpolation error saturates at the reconstruction limit, so a plain autoencoder leads that task. The main limitation is interpolation across a
topology change: no component of the regularised latent holds both a single-component and a post-break-up shape, whereas an unregularised autoencoder mixes them freely.
Read the full abstract ▾
Machine Learning (Statistics)
Identifiability Guarantees for Drivers and Dynamics of Delayed Physical Systems
Julien Boussard, Antoine Debouchage and Théo Saulus
A wide range of methods have been proposed, including physics-informed neural networks, which are powerful but do not guarantee identifiability of the dynamics, symbolic regression, which requires a set of precomputed operations, and causal discovery, which is more principled but usually relies on strong assumptions that physical systems may violate. In this work, we develop a theory-grounded method and prove that under a set of permissive assumptions, the structural drivers and drift of
stochastic delayed differential equations are identifiable. Our method outperforms others on a
benchmark for driver identifiability, and on a second
benchmark to evaluate physical consistency of the learned dynamics.
Read the full abstract ▾
Statistics Theory
Semiparametric Bernstein-von Mises theorems from Stein's method
Paul Rosa
We introduce a novel proof strategy for semiparametric Bernstein-von Mises
theorems based on Stein's method combined with an information-
geometric framework. Rather than controlling the effect of the prior on the asymptotic marginal posterior distribution of the functional of interest through the stability of an integrated likelihood under a suitable perturbation, we characterise its influence through the prior-weighted divergence of suitable vector fields over the statistical model. Applying Stein's method in this setting instead of the usual Laplace-transform approach yields explicit non-asymptotic upper bounds on the bounded-Lipschitz distance between marginal posterior distributions and the corresponding Gaussian limits predicted by semiparametric efficiency theory. These bounds consist entirely of local scalar differential quantities associated with the functional, the prior and a chosen vector field-typically related to the efficient influence function-evaluated over posterior contraction sets. We apply the theory to quadratic functionals in Gaussian white-noise models and to linear, quadratic and general integral functionals in histogram density models, with both conjugate and non-conjugate priors. For linear and quadratic functionals, the same theory identifies the differential term responsible for posterior bias and allows us to remove it through a natural functional correction, yielding Bernstein-von Mises-type
theorems in regimes where they may fail for the original uncorrected functional.
Read the full abstract ▾
Machine Learning (Statistics)
Post-Anomaly Detection Inference for Deep SVDD
Cao Le Cong Thanh, Dang Quang Vinh and Vo Nguyen Le Duy
Deep Support Vector Data Description (Deep SVDD) has become a prominent framework for unsupervised anomaly detection by learning latent representations that compactly characterize normal data around a center. Despite its empirical success, anomaly decisions produced by Deep SVDD are typically made solely based on anomaly scores without rigorous statistical guarantees, thereby limiting their reliability in safety-critical and high-stakes applications where false positives must be strictly controlled. In this paper, we propose PADI (Post-Anomaly Detection Inference), a novel framework that equips a trained and frozen Deep SVDD detector with statistically valid inference by leveraging the Selective Inference framework. Specifically, PADI performs inference conditional on the event that a test instance is identified as anomalous by Deep SVDD, thereby enabling rigorous statistical assessment of anomaly decisions. Based on this formulation, we derive valid selective p-values that quantify the statistical significance of the detected anomaly. Using these p-values, we theoretically establish control of the false positive rate (FPR) at a user-specified significance level $\alpha$ (e.g., $\alpha=0.05$). Furthermore, we extend the proposed framework to Deep Semi-Supervised Anomaly Detection (Deep SAD), providing a principled approach for statistically reliable inference in semi-supervised anomaly detection settings. Extensive experiments on both synthetic and real-world
benchmark datasets robustly support the theoretical findings. The results demonstrate that PADI consistently achieves proper FPR control while attaining superior true positive rates compared with existing approaches.
Read the full abstract ▾
Statistics Theory · Cross-listed from cs.LG
Search Dimension in Unlabeled Projection Pursuit: A Scaling Law for Subspace Restriction
Rares Dimitrie Grozavescu and Mark Girolami
Projection pursuit searches for a direction along which the data look least Gaussian. When the observation space contains a large Gaussian complement, the empirical objective can be minimized by a direction that carries no signal, with empirical kurtosis as low as at the truth. Sample splitting exposes rather than repairs this failure. Appending coordinates independent of the latent regime degrades the search while leaving Bayes recoverability unchanged. Restricting the search to the column space of a known forward operator removes the failure exactly on the negative-kurtosis branch. Estimating a principal subspace from the data is the alternative. In a controlled two-component model, the leading sufficient scalings differ in the gain with which the operator transmits the discriminant: $\varsigma^{-4}$ for covariance-spike estimation and $\varsigma^{-8}$ for fourth-moment search. At fixed search dimension, the measured threshold ratio collapses onto $n/p^2$ with exponent $0.156$, close to the predicted $1/8$. This is an empirically supported scaling motivated by sufficient bounds, not a proved asymptotically tight law. When the search dimension is varied, the measured exponent is $0.325$, substantially larger than $1/8$, and the tested range does not identify its functional form. The crossing location also depends on calibration and model configuration. Under a downstream excess-error criterion, the scaling largely disappears.
Read the full abstract ▾
Statistical Methodology
Two-sample tests for principal eigenvalues and eigenvectors in high-dimensional elliptical factor models
Xinyue Xu, Mengtao Wen and Long Feng
Changes in principal
eigenvalues and eigendirections provide complementary diagnostics of structural instability in factor models, but covariance-based inference can be unreliable under heavy tails. We develop two-sample tests for equality of these features of trace-normalized shape
matrices under high-dimensional elliptical factor models. Our approach combines Tyler's identity with orthogonalization and cross-fitting to accommodate general location and precision pilots. We derive a spectral limit theory that yields asymptotically valid calibration without upper-tail moment assumptions on the radial variables, allowing the two populations to have different radial distributions and factor ranks. Simulations demonstrate good size control and gains in size-adjusted power over covariance-based
benchmarks under heavy-tailed elliptical distributions. An application to S&P~500 stock returns illustrates how the tests distinguish changes in relative component strength from changes in component orientation.
Read the full abstract ▾