Statistics Theory
Maximum effective dimension and information gain
David Janz, Arya Akhavan and Alexandre B. Tsybakov
We establish general upper bounds on the maximum effective dimension and information gain of kernel Gram
matrices over arbitrary designs in terms of approximation properties of the kernel or, equivalently, of the corresponding reproducing kernel
Hilbert space. We show that across three regularity regimes, covering as special cases the Matérn and squared exponential kernels, these bounds cannot be improved beyond constant factors.
Read the full abstract ▾
Machine Learning (Statistics)
Sequential operator learning under dependent data
Rafael Oliveira
Learning operators from sequentially collected data arises in adaptive experimental design,
Bayesian optimization, and dynamical-system modelling, where observations may be dependent, and future inputs or sensing operators may depend on preceding data. We derive time-uniform self-normalized concentration bounds for
stochastic processes in
Hilbert spaces with vector-valued noise. We use these bounds to obtain regression-error guarantees for linear operators, including targets outside the
Hilbert estimation space, and for nonlinear parametric operators trained with strongly convex losses and regularizers. Our results allow possibly infinite-dimensional inputs and outputs without independence or mixing assumptions, providing a major step towards convergence guarantees for adaptive operator learning and learning from
stochastic dynamical data.
Read the full abstract ▾
Statistical Methodology
Groupwise Predictor Envelope Models for Multivariate Linear Regression
Sota Osumi, Akira Okazaki and Shuichi Kawano
Envelope methods improve estimation efficiency in multivariate analysis by isolating low-dimensional structures that contain all the information material to the parameter of interest. In multivariate linear regression with random predictors, predictor envelope models achieve this goal by removing variation in the predictors that is immaterial to the regression. In many applications, observations are naturally divided into several groups, such as treatment groups, regions, or demographic strata, and the regression relationship may differ between groups. Motivated by this setting, we propose a groupwise predictor envelope model for multivariate linear regression. The proposed model assumes that the group-specific regression coefficient
matrices are represented through a common predictor envelope subspace while allowing group-specific regression effects and group-specific error covariance
matrices. We derive an objective function for estimating the common predictor envelope, obtain the corresponding regression
estimators, and establish asymptotic normality together with an explicit asymptotic variance formula. Moreover, we show that the proposed
estimator is asymptotically more efficient than the
estimator obtained by fitting predictor envelope models separately to each group. This theoretical advantage over the existing work is also demonstrated through simulation studies.
Read the full abstract ▾
Statistical Methodology
Wild Bootstrap and Efron's Bootstrap for Debiased Cox Regression
Lena Schemet and Sarah Friedrich-Welz
Cox regression with Lasso penalization is widely used for variable selection in time-to-event data, but reliable coefficient inference after selection remains difficult. We investigate bootstrap inference for the debiased Cox
estimator after Cox Lasso selection. Two score-based procedures are considered: a wild bootstrap using independent multipliers and an Efron bootstrap using centered multinomial resampling weights. Both procedures keep the original Cox Lasso fit and debiasing
matrix fixed and apply bootstrap weights to the score contributions that determine the first-order distribution of the debiased
estimator. We establish asymptotic validity for both bootstrap schemes and study their finite-sample behavior in extensive simulations. The bootstrap intervals improve coverage accuracy over first-order debiased Wald intervals in many small- and moderate-sample settings, with the size of the improvement depending on the simulation setting and tuning rule. A SUPPORT2 application illustrates the procedures in practice.
Read the full abstract ▾
Machine Learning (Statistics)
A Heterogeneous Mixture of Experts Framework for Interpretable Machine Learning
Soham Chatterjee, Rwitobroto Dey and Smarajit Bose
Mixture-of-Experts (MoE) models provide a flexible framework for partitioning complex prediction problems into simpler local learning tasks through an input-dependent gating mechanism. Existing interpretable MoE approaches, such as Mixture of Decision Trees (MoDT), achieve transparency by employing homogeneous decision-tree experts, but this restricts the model to a single inductive bias across all regions of the feature space. We extend the MoDT framework by introducing heterogeneous expert families comprising decision trees, linear support vector machines, and quadratic discriminant analysis under a common probabilistic gating mechanism. To ensure coherent likelihood-based inference, non-probabilistic experts are calibrated to produce conditional class probabilities, allowing parameter estimation within the generalized Expectation-Maximization framework of MoDT. We further establish theoretical monotone ascent guarantees for the proposed heterogeneous gating updates, providing a justification for the optimization procedure. Experiments on a diverse collection of synthetic and real-world
benchmark datasets demonstrate that the proposed framework adaptively specializes experts according to local data geometry, yielding interpretable expert assignments while achieving predictive performance competitive with homogeneous MoDT and Random Forests. The proposed approach combines interpretability, adaptive inductive bias selection, and probabilistic coherence within a unified mixture-of-experts framework.
Read the full abstract ▾
Statistics Theory
Testing for Stable Intervals in Non-Stationary Time Series
Florian Heinrichs
Many time series are not stable over their full observation horizon, but may contain scientifically meaningful periods during which a signal remains stable up to a prescribed tolerance. We formulate this as an existence test for stable intervals in a non-stationary regression model with dependent, locally stationary errors. For a signal $d$ derived from the mean function, including deviations from reference levels and derivatives, stability over duration $\delta$ is encoded by $d_\infty=\inf_{t\in[0,1-\delta]}\sup_{s\in[t,t+\delta]}|d(s)|$. The hypothesis $d_\infty\ge\Delta$ states that no interval of length $\delta$ remains within the tolerance $\Delta$, while rejection provides evidence for the existence of a relevant stable period. We estimate $d$ by local linear regression and construct plug-in tests for $d_\infty$. The asymptotic distribution is determined only by near-extremal windows at which the minimax functional is attained. We formalize this localization through extremal sets and derive Gaussian and extreme value approximations for kernel
estimators over possibly shrinking index sets with time-varying long-run variance. The resulting tests are consistent and have asymptotic level control. The theory also extends supremum-based relevant-change tests to time-varying long-run variance and derivative-based hypotheses. Simulations and applications to physiological and engineering time series illustrate the method.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Robust Data-Collection Policy Learning for Low-Variance Online Policy Evaluation
Claire Chen, Shuze Daniel Liu, Licheng Luo et al. (+3 more)
In reinforcement learning policy evaluation, classic on-policy methods often suffer from high variance when estimating policy performance. To mitigate this issue, behavior policy search has been proposed to learn data-collecting policies tailored to reduce online evaluation variance. However, these approaches do not account for uncertainties in the transition functions. In practice, simulator transitions often differ from the real world due to modeling errors or approximation limitations. As a result, behavior policies trained in simulation may still yield high variance when deployed in real environments, leading to costly reliance on real-world evaluation samples. In this work, we propose a double-loop gradient-based
algorithm for learning behavior policies that are both efficient and robust to transition uncertainty. Theoretically, we derive novel transition-variance gradient expressions and establish global convergence guarantees for the
algorithm. Numerically, we demonstrate that our method is less sensitive to transition perturbations than existing approaches, providing supportive evidence for its practical utility.
Read the full abstract ▾
All authors Claire Chen, Shuze Daniel Liu, Licheng Luo, Rohan Chandra, Nan Jiang, Shangtong Zhang
Machine Learning (Statistics)
qshap: Fast Shapley Decomposition of $R^2$ for Gradient-Boosted Trees
Zhongli Jiang, Min Zhang and Dabao Zhang
Numerous methods have been developed to quantify feature attributions in individual predictions for tree ensembles. However, many applications require global measures of feature contributions to overall model performance. Although local attribution scores can be aggregated to characterize feature importance, such summaries do not directly decompose measures of predictive performance, such as $R^2$. This article introduces qshap, available in both R and Python, which provides Shapley decomposition of $R^2$ values for gradient-boosted decision trees (GBDTs) to quantify feature-specific contributions to model performance. By decomposing the quadratic loss of individual observations, qshap provides flexible tools to explore the importance of individual features and observations. qshap currently supports widely used GBDT implementations, including xgboost, lightgbm, and catboost, through a unified tree representation and efficient C++ backends. Its modular design can accommodate other GBDT implementations built from binary decision trees. In addition, we introduce a specialized backend for oblivious trees that exploits their symmetric structure to substantially accelerate computation.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Knowing When to Ask for Help: Bayesian Self-Escalation in Hierarchical LLM Agents
Nadeem Shaikh
Current LLM
agent systems decide delegation before
reasoning begins (a router picks a model) or after a response is complete (a verifier scores it and may retry). We study a third regime: an
agent that recognises, during its own
reasoning, that it is unlikely to succeed and transfers control to a stronger model. We formulate intra-generation delegation as a
Bayesian optimal-stopping problem over a learned competence posterior – an online estimate of the
agent's eventual task success whose sufficient statistics are learned from labelled trajectories, not read off raw
entropy. We derive the myopic escalation threshold in closed form, characterise the optimal policy via dynamic programming, and prove that the optimal policy is a time-varying threshold with no shape assumption on the raw signal. We further prove exponential separation of the oracle belief at the Chernoff-information rate of the signal, a regret bound governed by the calibration of the posterior, and a finite-sample guarantee: with n labelled calibration trajectories the deployed plug-in policy's regret decays as 1/sqrt(n). A controlled simulation study confirms each prediction of the theory, including the predicted 1/sqrt(n) rate. We additionally report a real-model validation on a Qwen2.5-Coder 1.5B->7B code cascade (MBPP, 257 tasks), confirming two of three pre-registered predictions: the escalation frontier dominates post-hoc routing at equal cost, and the cumulative competence belief's discrimination rises over generation.
Read the full abstract ▾
Statistical Methodology
Orthogonal double residual learning for optimal individualized treatment rules
Jiaqi Tong and Fan Li
Individualized treatment rules (ITRs) map baseline characteristics to treatment recommendations, with the optimal ITR maximizing expected reward or policy welfare. Indirect methods may require restrictive modeling assumptions, whereas direct methods can be sensitive to nuisance estimation error and limited overlap. We propose orthogonal double residual learning (ODRL), a two-stage, cross-fitted framework that directly targets the optimal ITR through cost-sensitive classification using the product of treatment and outcome residuals. To our knowledge, ODRL is the first direct method with a universally Neyman orthogonal objective requiring neither restrictive modeling assumptions nor inverse propensity score weighting. Thus, nuisance estimation errors affect regret through a second-order product, and ODRL remains robust under limited overlap. The Fisher consistent objective accommodates general decision rule sieves. We establish nonasymptotic high probability value function regret bounds relative to the Bayes classifier for VC classes, including linear rules and decision trees, and calibrated regret bounds for surrogate relaxations using support vector machines and deep ReLU neural networks. We further show that generic surrogate relaxations need not preserve orthogonality, whereas bounded score hinge learning does. Simulations demonstrate strong performance across complex and linear decision boundaries, limited overlap, and working model misspecification. Applications to the Right Heart Catheterization study and the Oxford Net Zero experiment illustrate interpretable treatment or policy recommendations. The odrlITR R package implements ODRL.
Read the full abstract ▾
Statistical Methodology
CUPED on Steroids: Multivariate Covariate Adjustment for Switchback Experiments
Sergei Pankratev and Palash Arora
Controlled-experiment Using Pre-Experiment Data (CUPED) reduces the variance of the treatment effect
estimator in online experiments by adjusting the in-experiment outcome metric using its lagged pre-experiment value. This method can be strengthened by enriching its
covariate set while keeping it automatable and guarding against overfitting and
covariate leakage. In switchback and related clustered experiment designs, this enrichment may be especially fruitful since randomization-unit-level variation in the outcome amplifies the variance of the treatment effect
estimator, so
covariates that explain it yield disproportionately large efficiency gains. We develop an extended CUPED framework for switchback and related clustered designs that introduces sizeable improvement in variance reduction efficiency relative to conventional CUPED while remaining lightweight and free of assumptions beyond those already implicit in CUPED. The framework combines multiple historical lags with cyclic hour-of-day encodings and cluster- and hour-level fixed-effect dummies in the pre-period prediction model—
covariates that target randomization-unit-level outcome variation—and uses cross-fitting so that realized variance reduction is not inflated by in-sample overfitting. In simulation, we analyze how each component affects validity and efficiency of this method and show that the full framework yields large gains in statistical power. We further validate the approach on publicly available trip records from the New York City Taxi and Limousine Commission. Finally, we derive a closed-form ceiling on the variance reduction attainable by any
covariate adjustment and connect our empirical findings to the composition-dependent limits this ceiling imposes.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
ChorusTIC: Training-Free Multivariate Time Series Classification via Chorus In-Context Learning
Juntao Fang, Shifeng Xie, Ruichu Cai et al. (+6 more)
Time series classification underpins applications in healthcare, sensing, and industrial monitoring. Although time series foundation models support forecasting and transferable representation learning, classification still typically requires fitting a task-specific classifier on each target
dataset, while individual channels of multivariate inputs are often encoded independently. We introduce ChorusTIC, a classification-native foundation model for in-context classification across heterogeneous channel configurations without target-task parameter updates. ChorusTIC combines episode-consistent Random Subchannel Slot Concatenation with a shared dual-axis encoder to model temporal and cross-channel interactions and map variable channel configurations into a fixed-width representation independent of the original channel count. It then calibrates feature axes using context-derived distributions and predicts query labels through leakage-protected in-context learning. We pretrain ChorusTIC solely on synthetic labeled episodes comprising context and query sets that share a task background, with classes distinguished by sparse temporal or cross-channel rules. Evaluations on the complete UEA-30 and UCR-128 archives show strong full-context and low-label performance without target-specific classifier fitting.
Read the full abstract ▾
All authors Juntao Fang, Shifeng Xie, Ruichu Cai, Shengji Zheng, Zijian Li, Keli Zhang, Lujia Pan, Themis Palpanas, Zhifeng Hao
Machine Learning (Statistics) · Cross-listed from cs.LG
Revenge of Monosemanticity: Neuron Specialization as a New Form of Feature Learning in MLPs
Amirhesam Abedsoltan, Enric Boix-Adsera, Fivos Kalogiannis et al. (+1 more)
Understanding how neural networks learn and organize features is central to understanding their behavior. Much existing theory of feature learning has focused on the emergence of a global low-dimensional representation. We show that this picture is incomplete. In regression problems with clustered data, we demonstrate that multilayer perceptrons (MLPs) naturally develop monosemantic specialized
neurons: individual
neurons become strongly aligned with a specific predictive feature relevant to a particular region of the input space. Rather than learning a single global low-dimensional representation, MLPs learn a collection of local low-dimensional representations. We show that this ability to specialize gives MLPs a provable data-efficiency advantage over feature-learning methods based on a global low-dimensional representation.
Read the full abstract ▾
All authors Amirhesam Abedsoltan, Enric Boix-Adsera, Fivos Kalogiannis, Mikhail Belkin
Statistics Theory · Cross-listed from math.PR
A Unified Exact Factorial-Moment Theory for Multi-set Allocation Occupancy (MAO) in Finite Populations
Xing-gang Mao and Xiao-yan Xue
Let $A_1,\ldots,A_T$ be independent uniformly selected subsets of a finite population of size $n$, with prescribed cardinalities $m_1,\ldots,m_T$. For each population element, define its occupancy level as the number of selected subsets containing it. Let $x_t$ and $x_{\geq t}$ denote the numbers of elements with occupancy exactly $t$ and at least $t$, respectively. The 2025 work introduced a general multi-set allocation occupancy (MAO) representation for higher-order occupancy moments. The present paper establishes its joint-probabilistic interpretation and provides a rigorous unified derivation for arbitrary joint occupancy categories and moment orders. Specifically, for arbitrary $B_1,\ldots,B_\ell\subseteq\{0,1,\ldots,T\}$, we prove the exact representation $F_\ell(B_1,\ldots,B_\ell)=G_T(B_1,\ldots,B_\ell)/(n)_\ell^{T-1}$, where $G_T(B_1,\ldots,B_\ell)$ is the corresponding generalized MAO transversal sum. This identity unifies the joint factorial moments of arbitrary occupancy categories within a single exact finite-population framework. In particular, writing $B_{\geq t}=\{t,t+1,\ldots,T\}$, the factorial moments of exact- and threshold-occupancy counts are obtained as the specializations $\mathbb{E}[(x_t)_\ell]=F_\ell(\{t\},\ldots,\{t\})$ and $\mathbb{E}[(x_{\geq t})_\ell]=F_\ell(B_{\geq t},\ldots,B_{\geq t})$. Mixed factorial moments, raw moments, variances, and covariances follow from the same representation through standard transformations. The formulas are verified by exhaustive enumeration over feasible parameter ranges and by Monte Carlo simulation. The resulting theory provides a rigorous and unified finite-population foundation for exact and threshold multi-set occupancy statistics.
Read the full abstract ▾
Statistical Methodology
Causal Effects of Modified Treatment Policies under Positivity Violations: A Partial Identification Approach
Taehyeon Koo, Elizabeth A. Stuart, Kara E. Rudolph et al. (+1 more)
Modified treatment policies (MTPs) are interventions based on each individual's natural treatment value. We study mean outcomes under MTPs for continuous treatments, including exposure mixtures. Positivity is the standard sufficient condition for identifying these mean outcomes without extrapolation: policy-generated values remain supported given
covariates. With multivariate treatments or continuous
covariates, treatment–
covariate combinations can be sparse or unsupported. Retaining the policy, our partial-identification framework decomposes its mean outcome into a point-identified contribution inside a positivity region and one outside. We bound the latter by imposing Lipschitz continuity on conditional mean potential outcomes rather than a parametric extrapolation model. The restriction compares each outside mean with the mean at an anchor inside the region. Metric projection minimizes width among one-anchor intervals but concentrates anchors on a lower-dimensional boundary, making the endpoints not pathwise differentiable. Our novel interior-displaced projection moves anchors inward, restoring pathwise differentiability. With a known region, we derive influence functions, characterize when they are efficient, and obtain asymptotically normal
estimators and confidence intervals. In simulations, our intervals attain at least nominal coverage where those assuming positivity undercover. In a pesticide-mixture application, protective associations suggested by methods assuming positivity are not robust to modest outcome variation beyond the estimated region.
Read the full abstract ▾
All authors Taehyeon Koo, Elizabeth A. Stuart, Kara E. Rudolph, Caleb H. Miles
Statistics Theory
Learning from Uncertainty-dependent Missing Labels for Semi-supervised Classification
You-Gan Wang, Jinran Wu and Geoffrey J. McLachlan
Missing labels are usually regarded as a source of information loss in classification. We study a semi-supervised setting in which the probability of label missingness depends on the observed features through posterior classification uncertainty. In this setting, the missingness indicator is not only a record of an unobserved label, but also an observable signal generated by a mechanism linked to the classifier. We develop a likelihood-based information theory for such uncertainty-dependent missing labels. Under correct specification, we derive a Fisher-information decomposition that separates a partial-labeling component from a nonnegative mechanism-
curvature term. Under joint misspecification of the label model and the missingness mechanism, we obtain the corresponding Godambe–Eicker–Huber–White sensitivity and sandwich-covariance partitions. We also clarify the relevant complete-data
benchmark: favorable missingness can increase information relative to ordinary fully labeled or budget-matched non-informative labeling baselines, but cannot exceed the information in the augmented experiment in which labels and mechanism indicators are both observed. For plug-in classifiers, we connect the information decomposition to margin-based excess-risk bounds. In regular two-component mixture settings this yields the parametric \(n^{-1}\) excess-risk rate, with constants determined by the nuisance-adjusted information in discriminant directions. Gaussian-mixture calculations and a medical diagnosis example illustrate how uncertainty-dependent labeling mechanisms can improve estimation and classification under a fixed labeling budget.
Read the full abstract ▾
Machine Learning (Statistics)
Generalization, memorization, and overfitting for diffusion models trained in the lazy high-dimensional regime
Hugo Latourelle-Vigeant, Sinho Chewi, Aram-Alexandre Pooladian et al. (+2 more)
Modern score-based generative models have achieved remarkable empirical success in high-dimensional tasks such as image, audio, and
video synthesis. These models reduce distribution learning to a sequence of regression problems that, if solved exactly on finite data, would ultimately reproduce the training samples. Their ability to generalize must therefore arise from the implicit or explicit regularization during training. In this work, we develop a generative counterpart to the theory of benign overfitting and algorithmic regularization for overparameterized neural networks in the supervised lazy-training regime. We study denoising score matching in a vector-valued reproducing kernel
Hilbert space with an inner-product kernel. In the proportional high-dimensional regime $n\asymp d$, we derive exact risk trajectories under gradient flow training. These trajectories exhibit three phases governed by qualitatively distinct
estimators: a spectral
estimator that generalizes, a pure-noise score with localized peaks that interpolate the training objective, and an empirical Bayes
estimator that memorizes the data. We then analyze how these
estimators combine along the reverse-time SDE and characterize the distribution of the resulting samples. The analysis reveals familiar mechanisms from supervised learning, including kernel linearization and self-induced regularization from the nonlinear part of the kernel, but also reveals a distinct phenomenology specific to generative modeling.
Read the full abstract ▾
All authors Hugo Latourelle-Vigeant, Sinho Chewi, Aram-Alexandre Pooladian, John Sous, Theodor Misiakiewicz
Statistical Methodology · Cross-listed from econ.EM
Repairing Locally Misspecified GMM: An Empirical Bayes Approach
Patrick Kline
Econometric models offer parsimonious but inexact approximations to data-generating processes. This paper studies the generalized method of moments (GMM) when exchangeable specification errors of order $n^{-1/2}$ contaminate the moment conditions. I develop
estimators for the mean and variance of these specification errors, establishing their consistency in an asymptotic framework where the number of overidentifying restrictions grows with the sample size. These hyperparameter estimates are used to develop a feasible bias-corrected
estimator of target parameters. I also propose an empirical Bayes
estimator that weakly improves precision by subtracting a best linear predictor of the first-order estimation error from the bias-corrected
estimator. Using a combinatorial central limit
theorem, I establish asymptotic normality of both
estimators and provide variance
estimators that enable misspecification-aware frequentist inference. Simulation exercises indicate the procedures can meaningfully improve on standard two-stage least squares estimation when exclusion violations are present. Revisiting the influential study of Angrist and Krueger (1991), I consider an instrument set where exchangeable excludability violations are plausible. Repairing the two-stage least squares estimates of the returns to schooling moves them in the direction of ordinary least squares and reduces sensitivity to the specification of controls.
Read the full abstract ▾