Machine Learning (Statistics) · Cross-listed from cs.LG
Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Experts
Yan Dai, Negin Golrezaei and Patrick Jaillet
Modern recommendation systems increasingly rely on dynamically routing diverse queries to multiple
embedding models. Despite its practical significance, this problem remains poorly understood under realistic conditions like
adversarial queries, bandit feedback, and limited observability of models. We formalize
embedding model routing as an
adversarial contextual linear bandit with low-rank experts, where contexts are queries, actions are items, and experts are the
embedding models working on low-rank latent representation spaces. We first establish that standard regret notions suffer from structural misspecification or statistical intractability, and we identify a log-quadratic policy class that is expressive enough to capture query-dependent model routing, yet structured enough to allow efficient online learning. Focusing on this log-quadratic policy optimization problem under bandit feedback – which is of independent interest – we propose a policy gradient
algorithm called Hypentropy Policy Gradient (HPG). It provably adapts to the unknown low-rank structure under incomplete information and attains $\widetilde{\mathcal O}(s\sqrt{M T})$ linearized policy regret – where $s, M$, and $T$ are the intrinsic rank of the experts, the number of models, and the number of rounds – thus avoiding a curse of dimensionality. We provide computationally efficient and parameter-free implementation of HPG.
Read the full abstract ▾
Statistical Methodology
Flexible Method Comparison with the Probability of Agreement
Nathaniel T. Stevens
The comparison of methods of measurement is a common problem in clinical practice; as novel methods are developed, establishing their agreement with existing methods is crucial. The probability of agreement (PoA) has previously been proposed as an intuitive and informative means of assessing agreement between two methods of measurement. It straightforwardly quantifies the likelihood that two measurements by different methods on the same subject are clinically indistinguishable. In this paper, we overhaul and extend the PoA methodology by developing an inference framework that relaxes several restrictive assumptions made in previous implementations, ultimately increasing its utility in a wider range of applications. We illustrate this more flexible methodology in an example that compares methods of measuring total Prostatic Specific Antigen (tPSA). And we thoroughly investigate its performance via simulation. This work dramatically increases the flexibility, availability, and hence impact of the PoA approach for method comparison.
Read the full abstract ▾
Machine Learning (Statistics)
Audited Conformal Prediction for Classification under Unknown Distribution Shift
Yanfei Zhou, Rizal Fathony, Nam H. Nguyen et al. (+1 more)
We consider the problem of uncertainty quantification for a pretrained classification model deployed under unknown distribution shift. We propose Audited Conformal Prediction (ACP), a method that leverages a small labeled
dataset from the target population to train an auxiliary audit model identifying inputs where the legacy model is likely to fail. By integrating the audit model's outputs into the conformal prediction framework, ACP produces prediction sets that guarantee marginal coverage while achieving substantially higher conditional coverage in practice than existing approaches. We develop and analyze two complementary integration strategies – one targeting marginal coverage with improved conditional performance, the other providing explicit group-conditional coverage guarantees – and establish theoretical guarantees for both. Experiments on synthetic and real-world
datasets validate the method and illustrate trade-offs between prediction set size and conditional coverage.
Read the full abstract ▾
All authors Yanfei Zhou, Rizal Fathony, Nam H. Nguyen, Matteo Sesia
Statistical Methodology
Heterogeneous behavioral mechanisms in epidemiological models
Jessica Pavani, Rob Deardon and Alexandra M. Schmidt
Traditional epidemic models frequently assume behavioral homogeneity. The susceptible-infected-recovered model provides a robust foundation for characterizing disease transmission, but it does so without accounting for how people actually respond to risk. In contrast, behavioral change models incorporate mechanisms that capture how individuals adjust their actions during an outbreak, recognizing that rising infection risk typically motivates protective behaviors. Yet both approaches share a key limitation: they overlook the inherent heterogeneity of a population. In reality, communities are a complex mixture of risk tolerances and behavioral tendencies. Ignoring this inherent heterogeneity can obscure important differences in how individuals perceive and respond to disease threats. This paper introduces a novel
Bayesian mixture model designed to address this limitation by partitioning the population into two distinct behavioral patterns: risk-neutral individuals, who maintain baseline contact rates, and risk-averse individuals, who modulate their behavior in response to epidemic severity. By integrating these disparate dynamics into a unified transmission framework, the proposed model explicitly accounts for varying population behaviors often overlooked by aggregate approaches. Through simulation studies and empirical data applications, we demonstrate that this approach significantly outperforms traditional models in parameter recovery, epidemic trajectory estimation, and
forecasting precision. The findings suggest that failing to account for behavioral diversity leads to biased peak estimates and artificially stretched epidemic curves. Consequently, this research provides a more nuanced computational toolkit for predicting outbreak trajectories in socially fragmented environments, ensuring that public health intervention strategies are informed by a foundation of behavioral realism.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.AI
Relational Structural Causal Models
Adiba Ejaz and Elias Bareinboim
An artificial intelligence must have a model of its environment that is causal, supporting
reasoning about interventions and counterfactuals, and also combinatorial, supporting generalization to unseen combinations of objects. In this work, we formally study when and how such a model can be learned. We develop relational structural causal models, extending structural causal models (Pearl 2009) to settings where objects and their relations vary. First, we show how answers to not only causal but also observational queries about unseen combinations of objects can not be identified without further assumptions. To enable such identification–including in the presence of unobserved confounding–we define relational causal graphs and derive symbolic identification criteria. Finally, we propose relational neural causal models, a provably correct approach that outperforms non-relational baselines on simulated traffic scenes with varying cars, signals, and pedestrians.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from hep-ph
Event Generation with Parallel Langevin Sampling and Learned Stein Diagnostics
Rob Verheyen
Efficient event generation is a major computational challenge for precision
collider phenomenology, especially for high-multiplicity final states where
matrix-element evaluations are expensive and rejection-sampling efficiencies are low. We study an alternative approach based on many parallel underdamped Langevin chains, retaining one terminal state from each chain to obtain unweighted events while avoiding within-chain autocorrelation. A learned Stein discrepancy is used as a convergence diagnostic, providing a data-driven estimate of the relaxation time. We apply the method to tree-level $u\bar u\to Z+n g$ event generation and find that relaxation requires only a modest number of exact-target Langevin steps, with mild growth over the multiplicities studied. Finally, we show that simple neural-network surrogate initialization can substantially reduce the required number of exact
matrix-element and gradient evaluations.
Read the full abstract ▾
Statistical Methodology
Causal Sufficient Dimension Reduction for Multiple Continuous Exposures with an Application to Environmental Mixtures
Thomas W. Hsiao, Howard H. Chang and Razieh Nabi
Estimating causal effects with multivariate continuous exposures is challenging because causal exposure-response surfaces can be high-dimensional, complicating estimation and interpretation of joint exposure effects. Such settings arise in environmental epidemiology, where interest centers on the health effects of chemical and pollutant mixtures. We develop causal sufficient dimension reduction (CSDR), a semiparametric framework for representing causal exposure-response surfaces through low-dimensional exposure summaries. We formalize the reduction target as the causal central mean subspace and propose a modular two-stage
estimator that decouples nuisance-function estimation from subspace estimation, simplifying implementation relative to existing marginal structural model-based approaches. The reduced exposure preserves the information needed to characterize joint causal effects while enabling efficient downstream estimation. We establish a convergence rate for causal subspace recovery accounting for first-stage nuisance estimation error, show that the structural dimension can be estimated consistently, and introduce a subspace importance score that quantifies the contribution of each exposure to the reduction. In simulations, CSDR yielded more accurate estimation and uncertainty quantification of the exposure-response surface than methods using noncausal dimension reduction or the original exposure. We apply CSDR to study the effect of maternal exposure to PFAS chemical mixtures on infant birth weight in the Atlanta African American Maternal-Child Cohort.
Read the full abstract ▾
Statistical Methodology
Bartlett adjustment for Gaussian random effects meta-analysis
Haben Michael
Meta-analyses are often based on too few studies to justify the asymptotic methods underlying the statistical procedures applied to them. We consider higher-order asymptotics as a remedy. We derive the Bartlett correction for the idealized Gaussian case, correcting the formula currently appearing in the literature.
Read the full abstract ▾
Statistics Theory · Cross-listed from math.PR
Lehner's operator norm formulas, semidefinite programming, and spiked matrix models
Dmitriy Kunisky
Lehner (1999) derived elegant formulas for the operator norm $\|\mathfrak{X}\|$ of operators of the form $\mathfrak{X} = \mathbf{A}_0 \otimes \mathfrak{1} + \sum_{i = 1}^n \mathbf{A}_i \otimes \mathfrak{m}_i$, also easily generalized to the spectral edge $\lambda_{\max}(\mathfrak{X})$, in terms of nonlinear optimization problems over positive definite
matrices. Here the $\mathbf{A}_i$ are finite-dimensional Hermitian
matrices, the $\mathfrak{m}_i$ are either free semicircular or free Rademacher families of operators, and $\mathfrak{1}$ is the identity operator. We first show that both of Lehner's nonlinear optimizations can be rewritten as linear semidefinite programs (SDPs), even in the Rademacher case where Lehner's optimization is not itself convex. We give the primal and dual forms of these SDPs, derive the complementary slackness relations and consequences thereof, and propose that the SDPs are more stable and accurate than the iterative numerical scheme proposed in Lehner's original work. We then apply the SDPs from the semicircular case to spiked
matrix models, studied recently via Lehner's formula by Bandeira, Cipolloni, Schröder, and van Handel (2024). We give a new proof of the Baik–Ben Arous–Péché (BBP) transition they establish in models with isotropic (but possibly correlated) Gaussian noise by constructing feasible variables for the associated primal and dual SDPs. Combining our construction with a sensitivity interpretation of optimal dual variables, we study the fluctuations of leading eigenvectors of such models. We
conjecture and give numerical evidence that these fluctuations are Gaussian but anisotropic and non-universal, and that their covariance may be computed in terms of the optimizer of the dual of Lehner's formula, which in turn is approximately the leading eigenmatrix of a completely positive operator associated to the covariance of the noise model.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Optimal Hidden-Target Learning for Online Inventory Optimization on General Convex Sets
Anthony Pineci and Yunzong Xu
Online inventory optimization (OIO) is online convex optimization with physical
memory: inventory carryover makes the feasible action set depend on the past. A natural principle, used in
stochastic inventory learning and recently in OIO under a single linear capacity constraint, is to maintain a hidden target chosen by an online learner and implement its projection onto the currently feasible order-up-to set. We prove that this simple principle is optimal for OIO on arbitrary bounded convex capacity sets. With online gradient descent as the base learner, the method improves the best known regret guarantee for OIO on general convex sets from inverse to inverse-square-root dependence on the common-demand probability, and we prove a matching lower bound. The same principle gives the first polylogarithmic regret guarantee for strongly convex losses and the first dynamic regret guarantee adapting to Euclidean path variation on general convex capacity sets. The analysis introduces a norm
alignment principle: the right state variable is the distance from the hidden target to the feasible set, measured in the same norm as the projection. Under norm
alignment, this distance evolves pathwise as a scalar queue, with target movement as arrival and common demand as service. This reduction to one-dimensional queue control resolves the state dependence and extends the guarantees to general convex capacity sets, beyond the reach of prior productwise approaches. Experiments on synthetic and real-world inventory data corroborate the theory.
Read the full abstract ▾
Statistical Methodology
Testing Preferential Sampling
Isabel Natario and Andreia Monteiro
Geostatistics aims to infer a spatially continuous phenomenon from observations collected at a finite number of locations, frequently measured with error. Whenever there is
stochastic dependence between the spatial and sampling processes, preferential sampling occurs. Ignoring this problem drives to incorrect and biased estimates and, therefore, recognizing it is quite important, but not always simple to execute and understand. In this work, a test for assessing preferential sampling, simple and easy to implement, is presented, overcoming the previous concerns. It is based on the dependence between the number of sampled points and the values of the corresponding measures. The performance of the proposed test id assessed through a large simulation study, which consideres different levels of preferentiability, relation with a
covariate, different sample sizes and different test procedure conditions. The results are quite encouraging, with high levels of correct preferential sampling detections, further confirmed by the test application to already known real data sets of lead concentrations in moss samples and red and blue shrimp capture data.
Read the full abstract ▾
Statistical Computation · Cross-listed from cs.LG
A Statistical and Machine Learning Framework for Operational Threshold Detection and Deployable Dispatch Controller Development in Hydrogen Multi-Energy Systems
Shadi Heenatigala and Hasanika Samarasinghe
This study presents a statistical and machine learning framework for characterizing a hydrogen-based multi-energy system (H-MES) using one year of high-resolution operational data. Statistical analysis revealed a binary operation driven by renewable surplus, with
solar irradiance explaining 45.7% of rank-based variance in hydrogen production, a large effect by conventional standards. Only high-irradiance periods triggered meaningful electrolyzer engagement, while electricity demand exerted a weaker inverse suppression effect ($\epsilon^2 = 0.126$). Multiple regression confirmed electrolyzer power as the dominant linear predictor, with a synergistic
solar-wind interaction. Notably, Random Forest analysis ranked wind output first in predictive importance despite its weak bivariate correlation (r = 0.167), revealing non-linear dynamics invisible to parametric methods. A sequence model exploited strong 24-hour autocorrelation (r = 0.845) for operational
forecasting, while a reinforcement learning
agent optimized hydrogen revenue dispatch. The core contribution is demonstrating that statistical and machine learning approaches are complementary for H-MES modeling and control.
Read the full abstract ▾
Machine Learning (Statistics)
Cluster LOCO: Feature Importance For Interpreting Clusters
Claire M. He and Genevera I. Allen
Clustering is widely used for exploratory analysis and scientific discovery, driving insights from
market segmentation to biological data analysis, but its outputs can be difficult to interpret, audit, and reproduce as modern
datasets become increasingly large and complex. Reliable use of clustering requires understanding which features drive the discovered structure, yet feature-level explanations for clustering remain scarce compared with methods in supervised learning. Furthermore, existing clustering feature importance scores are often tied to specific
algorithms and data assumptions. To address these challenges, we propose Cluster LOCO (Leave-One-
Covariate-Out), a family of model-agnostic feature importance scores for clustering. Cluster LOCO is built on feature occlusion and clustering generalizability, defined as whether cluster labels learned on one subset of the data can be accurately predicted on held-out samples. For any chosen clustering
algorithm, Cluster LOCO quantifies a feature's importance by measuring how much its removal degrades generalizability. We first introduce Cluster LOCO-Split, which relies on data splitting, and then extend it to Cluster LOCO-MP, a minipatch ensemble-based version designed for large-scale data. Across synthetic simulations and an application to
cell-type discovery in single-
cell transcriptomics, we show that Cluster LOCO more reliably recovers informative features than existing clustering feature importance methods.
Read the full abstract ▾
Statistical Methodology
Typical Healthcare Pathways as a Basis for Admixture Modeling of Patient Trajectories
Maryam Farhadizadeh, Carola S. Heinzel, August Sigle et al. (+5 more)
Background: Understanding whether patients follow similar or distinct patterns of care is important for characterizing clinical practice, identifying patient subgroups, and supporting quality improvement. However, routine healthcare trajectories are difficult to compare directly because patients may differ in their diagnostic workup, treatment sequencing, timing of clinical events, and documentation practices. Despite this variation, trajectories often contain recurring patterns at the cohort level. Methods: To address this challenge, we present a framework that explicitly separates cohort-level typical pathway identification from patient-level inference. At the cohort level, we derive an interpretable representation of care processes using a rule-based
algorithm to identify typical healthcare pathways, resulting in a compact pathway graph. These pathways are then modeled as Markov chains and used as structured components in an admixture model, allowing each patient to be represented as a probabilistic mixture of typical pathways rather than being assigned to a single pathway component. The resulting admixture weights provide a compact representation of patient trajectories for subgroup characterization. We further assess the stability of the identified pathways and inferred admixture representations across multiple train-test splits. Results: Across train-test splits, the framework demonstrated consistent pathway structures and patient-level mixture patterns. Applied to routine care data from prostate cancer patients undergoing radical prostatectomy, the framework identified interpretable care patterns and supported the identification of patient subgroups with similar clinical event patterns. Conclusion: Overall, the proposed framework provides an interpretable and stable approach for summarizing treatment pathways and characterizing patient subgroups in real-world practice.
Read the full abstract ▾
All authors Maryam Farhadizadeh, Carola S. Heinzel, August Sigle, Harald Binder, Frederik Wenz, Jan Hasenauer, Peter Pfaffelhuber, Nadine Binder
Machine Learning (Statistics) · Cross-listed from math.OC
Free Heavy-Tailed Lunch for Muon: A Theoretical Justification of Empirical Success
Florian Hübler, Thomas Pethick and Suvrit Sra
Non-Euclidean optimisation methods with
matrix-valued updates, such as
Muon and Scion, have recently shown strong empirical performance for training
Transformer models, yet their theoretical advantages over Euclidean methods remain poorly understood. We address this gap in the heavy-tailed non-convex regime, where
stochastic gradients have bounded $p$-th central moments, $p \in (1,2]$. We show that certain non-Euclidean methods achieve optimal sample complexity under stronger stationarity measures, while Euclidean methods incur additional dimension-dependent costs. As a consequence, for $m \times n$
matrices,
Muon finds an $\varepsilon$-stationary point in
nuclear norm within $\mathcal{O}\left(\min\{m, n\} \frac{\Delta_1 L}{\varepsilon^2} \left(\frac \sigma \varepsilon \right)^{\frac p {p-1}}\right)$ samples, absorbing heavy-tailed noise without extra dimension dependence, unlike Euclidean methods. We further prove this sample complexity, including its dimension dependence, is optimal for all first-order methods under
nuclear-norm stationarity. Experiments on large
language models support our theory. Surprisingly, our results suggest that other Schatten geometries beyond the spectral geometry of
Muon can perform competitively in certain settings.
Read the full abstract ▾
Statistical Methodology
On the design distribution for predictive Bayesian regression
Wanyue Sun and Edwin Fong
The predictive approach to
Bayesian inference accesses the posterior distribution via a sequence of one-step-ahead predictives, enabling inference via predictive resampling without Markov chain Monte Carlo. In the random-design regression setting, an explicit specification of the predictive design distribution is required, yet the impact of this choice has received little formal
attention. We study the role of this predictive design distribution in parametric martingale posteriors for regression, and identify predictive notions of identifiability and design invariance that are essential for valid inference, particularly in the high-dimensional regression setting. Building on these foundations, we introduce a novel class of parametric martingale posteriors for regression that satisfies a weak form of these desiderata, and naturally accommodates the high-dimensional setting through regularization. We then illustrate our method through a simulation.
Read the full abstract ▾
Statistical Methodology · Cross-listed from physics.soc-ph
Modeling inhomogeneous spatial point configurations with applications to replicated patterns in real-life waiting crowds
Lars Sickert Karam, Rui M. Castro, Maarten Schoukens et al. (+1 more)
In this article, we connect statistical inference for spatial point processes with the analysis of waiting pedestrian crowds through two interconnected contributions. First, on the methodological side we develop an inference procedure for semiparametric spatial point process models leveraging replicated spatial patterns, i.e., multiple approximately independent realizations from the same process. Second, we show that spatial point processes provide a suitable modeling framework for waiting pedestrians, capturing two key aspects: spatial inhomogeneity driven by location attractiveness and repulsive interactions between pedestrians. These two components are central to the inference problem itself, since spatial point process modeling hinges on disentangling background intensity from interaction. Although replicated spatial patterns are rare in point process literature, they are available here through a unique real-life pedestrian
dataset, thereby directly linking the methodological development to the physical application. We use the proposed methods to fit and evaluate determinantal and Gibbs point processes in a simulation study and a real-world case study. Despite persistent challenges in decoupling the influences of inhomogeneity from interaction, these models are able to reproduce key empirical features of waiting pedestrians.
Read the full abstract ▾
All authors Lars Sickert Karam, Rui M. Castro, Maarten Schoukens, Alessandro Corbetta
Statistics Theory
Nonparametric inference on Fokker-Plank and McKean-Vlasov models
Adriana Laurindo Monteiro and Roberto Imbuzeiro Oliveira
We propose a kernel-based
estimator of the velocity field governing the transport and
diffusion of $d$-dimensional interacting particles. Assuming the initial positions are i.i.d. with law $\mu_0$, we establish consistency of the
estimator with an explicit mean-squared error rate of order $h^2 + N^(-2/(d+2))$, where $h$ denotes the time-discretization step and N the number of particles. The analysis covers two distinct settings: the Fokker-Planck equation, where we recover the underlying potential function, and the McKean-Vlasov equation, where we deconvolve the interaction kernel driving the mean-field dynamics.
Read the full abstract ▾
Machine Learning (Statistics)
Beyond the Training Distribution: Evaluating Predictions Under Distribution Shift and Selection Bias
Annie Ulichney and Amanda Coston
Understanding how a prediction model will perform in a new environment before deployment is essential to preventing harm when
algorithms inform decision-making. Two common sources of model performance degradation are (i)
covariate shift, where the target
covariate distribution differs from the source, and (ii) selective labels, where the observability of outcomes depends on historical decisions. We study pre-deployment model evaluation under the joint presence of
covariate shift and labeling of outcomes selectively based on observed features. In particular, we present a double machine learning procedure for estimating the target risk of an arbitrary black-box prediction model under a general loss function. We show identification of this estimand under standard assumptions and derive a bias-corrected
estimator based on the influence function of the target risk. Finally, we evaluate our
estimator through experiments using the eICU electronic health records database, showing that it tracks the true target risk more accurately than methods that address either selective labels or
covariate shift alone, as well as baselines that combine standard plug-in approaches.
Read the full abstract ▾
Statistics Theory
Universality for Products of Random Matrices with i.i.d. Entries and the Fuss–Catalan Number
Yanjin Xiang, Kun Chen and Zhihua Zhang
Let \((w_{ij})_{i,j\ge1}\) be a single infinite array of independent identically distributed real- or complex-valued entries of mean zero, variance \(\sigma^2\), and finite fourth moment. Set \(W_n=(w_{ij})_{1\le i,j\le n}\) and \(X_n=n^{-1/2}W_n\). For every fixed \(k\ge1\), we identify the almost sure limiting operator norm of several fixed products built from this family. Define the \(k\)-th freeness coefficient by \[ \gamma_k:=\sqrt{\frac{(k+1)^{k+1}}{k^k}}. \] Then we prove \[ \|X_n^k\|\to\sigma^k\gamma_k \qquad \text{almost surely}. \] The same limit holds for products sampled with replacement from any fixed finite pool of independent copies of \(X_n\); in particular, it holds for the product of \(k\) independent copies. Thus, the freeness coefficient captures the non-commuting characteristic between large random
matrices %powers and independent or fixed-pool sampled products under the finite fourth moment assumption. The improvement of the classical Bai–Yin-type power estimate from the scale \(\sigma^k(k{+}1)\) to \(\sigma^k \sqrt{k{+}1}\) is a direct corollary of our result. The main technical challenge is to prove the upper bound using a high-moment expansion of %the upper bound is proved by a high-moment expansion of \(\E\Tr((X_n^kX_n^{*k})^m)\). The leading zero-defect trace words are tree-like and are counted by the Fuss–Catalan number \[ F_{k,m}= \frac1{km+1}\binom{(k+1)m}{m}. \] The combinatorial tool helps to devise a defect-sensitive global enumeration: if \(L=km\) and \[ r=(L+1-v)+(L-q), \] then the number of admissible word classes with defect \(r\) is at most \(F_{k,m}(Cm)^{Dr}\). This
polynomial-in-\(m\) loss, with degree proportional to the defect, is summable in the logarithmic moment range.
Read the full abstract ▾
Statistical Methodology
Joint Nuclear and $\ell_1$ Regularization for Logistic Matrix Regression with Applications to Brain Imaging
Damian Brzyski, Aaron Cohen, Zijian Wang et al. (+3 more)
We introduce a new convex optimization framework for logistic scalar-on-
matrix regression which incorporates
nuclear and $\ell_1$ norm penalties to enforce simultaneously low-rank and sparse structures in the estimated coefficient
matrix. The proposed method enables interpretable modeling of high-dimensional
matrix-valued predictors in the presence of binary responses. We derive a custom
algorithm based on the Alternating Direction Method of Multipliers (ADMM) to efficiently solve the resulting convex optimization problem and establish the theoretical properties of the obtained solution. Numerical experiments clearly demonstrate the effectiveness of our method in recovering meaningful predictive patterns. Finally, we apply our method to the
brain imaging data to identify structures in functional
brain connectivity
matrices that are characteristic of subjects with a family history of alcohol use disorders (AUDs).
Read the full abstract ▾
All authors Damian Brzyski, Aaron Cohen, Zijian Wang, Mario Dzemidzic, David A. Kareken, Jaroslaw Harezlak
Statistics Applications
Stable Multivariate Functional Time Series Prediction for Major Geomagnetic Indices
Yian Yu, Shasha Zou, Tuija Pulkkinen et al. (+1 more)
High-resolution scientific data, such as geomagnetic index streams, often exhibit complex temporal dependencies that can be modeled through functional data analysis. Conventional functional time series (FTS) methods typically partition continuous processes into non-overlapping segments, which artificially fragments temporal continuity and can limit estimation efficiency and stability. This is particularly evident in geomagnetic time series prediction due to their noisy, sudden, and large-scale changes. This study presents a robust multivariate FTS
forecasting framework for multi-dimensional time series with inter-series correlations and the existence of exogenous predictors. We introduce an overlapping rolling-window scheme that preserves temporal coherence and mitigates boundary information loss, thereby enriching the effective sample size for a more efficient and stable estimation. We integrate functional principal component analysis for dimension reduction with a vector autoregressive model with exogenous inputs to capture latent dynamics across correlated series. We also construct computationally efficient conformal prediction intervals for uncertainty quantification. The framework is motivated by and applied to the simultaneous
forecasting of five critical geomagnetic indices, Kp, Dst, SYM-H, SME, and SMR, using
solar wind parameters as predictors. Empirical results show that this approach outperforms state-of-the-art machine learning baselines, extends forecast horizons to 6-24 hours, and provides calibrated uncertainty bounds.
Read the full abstract ▾
All authors Yian Yu, Shasha Zou, Tuija Pulkkinen, Yang Chen
Machine Learning (Statistics) · Cross-listed from cs.LG
Federated Learning for Feature Generalization with Convex Constraints
Dongwon Kim, Donghee Kim, Sung Kuk Shyn et al. (+1 more)
Federated learning (FL) often struggles with generalization due to heterogeneous client data. Local models are prone to overfitting their local data distributions, and even transferable features can be distorted during aggregation. To address these challenges, we propose FedCONST, an approach that adaptively modulates update magnitudes based on the parameter strength of the global model. This prevents over-emphasizing well-learned parameters while reinforcing underdeveloped ones. Specifically, FedCONST employs linear convex constraints to ensure training stability and preserve locally learned generalization capabilities during aggregation. A Gradient Signal to Noise Ratio (GSNR) analysis further validates the effectiveness of FedCONST in enhancing feature transferability and robustness. As a result, FedCONST effectively aligns local and global objectives, mitigating overfitting and promoting stronger generalization across diverse FL environments, achieving state-of-the-art performance.
Read the full abstract ▾
All authors Dongwon Kim, Donghee Kim, Sung Kuk Shyn, Kwangsu Kim
Statistics Applications
A Deep Zero-Inflated Model of North Atlantic Right Whale Presence To Support Blue Economy Management
Jiaxiang Ji, Laura Nazzaro and Ahmed Aziz Ezzat
Effective modeling of endangered marine mammal species, such as the North Atlantic Right Whale, is critical for balancing marine conservation with the growing blue economy. Passive acoustic monitoring data collected by autonomous
underwater vehicles provide new opportunities for localized marine species detection and oceanographic sensing, but introduce complex statistical challenges such as zero inflation, imperfect detection, and intricate dependence structures. In response, we propose the Deep Zero-Inflated Bernoulli (DeepZIB) model, a deep statistical method which jointly models latent species presence and conditional detection probabilities while learning complex habitat relationships from heterogeneous
covariate information. We establish theoretical results on the model's structural properties and conduct simulation experiments to demonstrate its ability to recover underlying parameters and latent presence fields. Application to real-world passive acoustic monitoring data on the North Atlantic Right Whale along the U.S. East Coast demonstrates improved model adequacy and predictive performance in capturing the species' dynamic and spatially varying habitat. We demonstrate DeepZIB's ability to generate high-resolution, spatially and temporally varying presence maps, providing valuable insights for targeted and risk-aware management of blue economy industries, ranging from offshore and marine energy, to fisheries management and maritime transport
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cond-mat.dis-nn
Local Coverage Governs Memorization in Diffusion Models
Claudia Merger and Sebastian Goldt
Memorization in
diffusion models is often treated as a global property of the model or
dataset. In practice, however, a single
diffusion model can simultaneously generate both memorized and novel samples. Which training samples are most likely to be memorized? In this work, we show that memorization is governed by local data coverage. Leveraging the connection between
diffusion models and kernel density estimation (KDE), we derive a theoretical criterion that predicts whether a point is memorized based on the density of training data in its neighborhood and the size of the training
dataset. In the high-dimensional limit, this leads to a sharp, local transition: regions of low coverage are dominated by isolated training samples, which are memorized, while dense regions support interpolation and generalization. We validate these predictions empirically, showing that memorization increases with local sparsity and that
diffusion models exhibit a coexistence of memorized and novel samples within the same model. Extending this framework to multi-class settings, we further show that classes with higher intra-class sparsity (and thus lower local coverage) are more strongly memorized. Our results provide a local view of memorization in
diffusion models, explaining when and where memorization occurs in terms of data geometry.
Read the full abstract ▾
Statistical Methodology
Predictive Concordance for Parameter Optimisation and Mixture Synthesis
Tobias Adrian, Domenico Giannone, Matteo Luciani et al. (+1 more)
We discuss probabilistic measures of concordance between two probability distributions based on the expected misclassification rate (EMR). The focus is on comparing a given reference distribution with other distributions in a parametrised class, and optimising concordance by identifying parameter values maximising EMR or a regularised variant. EMR is a practical and decision-theoretically meaningful measure, and its optimisation has direct interpretation as a
Bayesian decision analysis with a bounded utility function. We explore theoretical properties of EMR, discuss relationships with other measures including Küllback-Leibler divergence, and recognise that its optimisation has a synthetic
Bayesian emulation interpretation that aids understanding and specification of regularisation penalties. A main area of methodology is in mixture synthesis where the parametrised family is a discrete mixture of given distributions. A detailed example comes from scenario
forecasting in macroeconomic policy settings, a key applied area motivating the new methodology. Theoretical developments underlie efficient numerical optimisation and analysis is easily implemented using direct Monte Carlo simulation.
Read the full abstract ▾
All authors Tobias Adrian, Domenico Giannone, Matteo Luciani, Mike West
Statistics Theory
Recovery thresholds for hidden weighted sparse graphs
Zhe Hou and Jingcheng Liu
Recovering structural information from noisy high-dimensional data is a fundamental task in statistical inference. We investigate the recovery thresholds for a graph hidden in a randomly weighted complete graph. Specifically, an unknown graph $H^* \in H_n$ is chosen uniformly at random, and hidden in a complete graph of $n$ vertices as follows: the weight of an edge $e \in H$ is distributed independently according to $P_n$; otherwise the weight is distributed independently according to $Q_n$. The goal is to recover almost all of $H$ from these edge weights. Assuming a local Lipschitzness of the Rényi divergence between distributions $P_n$ and $Q_n$, and a mild density condition for the graphs $H_n$, we give a unified characterization of the information-theoretic limit for recovering almost all of $H$ (also known as almost exact recovery). Our characterization connects the KL divergence between $P_n$ and $Q_n$ to the logarithm of the first moment threshold of $H$ in the Erdős-Rényi random graph model $G(n,p)$. Our lower bound also extends to the task of partial recovery, in which only a constant $\lambda$-fraction of $H$ needs to be recovered. Last but not least, for certain Bernoulli and Exponential regimes, and for Gaussian distributions, we are able to show an All-or-Nothing (AoN) threshold phenomenon at the exponential scale.
Read the full abstract ▾
Machine Learning (Statistics)
Nonlocal Bayesian Modeling of Continuous Spatio-Temporal Dynamics
Jaeyeong Lee and Heeyoung Kim
Real-world spatio-temporal
forecasting must handle irregular time points, spatially sparse observations, and the need for uncertainty quantification. This setting is often further compounded by nonlocal interactions (long-range spatial coupling). Modeling continuous-space, continuous-time nonlocal dynamics naturally leads to infinite-dimensional integro-differential equations (IDEs), making principled
Bayesian inference intractable. We propose the NonLocal
Bayesian Spatio-Temporal model (NLBST), a hierarchical
Bayesian framework for continuous spatio-temporal fields that learns explicit nonlocal coupling while retaining tractable inference. NLBST represents the latent field via a coordinate-based spatial basis expansion and models the coefficient process with a continuous-time ODE whose learnable linear operator corresponds to a Galerkin reduction of a nonlocal IDE; a Neural ODE residual captures additional nonlinear dynamics. A linear-Gaussian observation model enables Kalman-style sequential updates under missing and irregular observations, while the spatial basis representation enables inductive prediction at unmeasured locations without retraining. Global parameters are learned via variational inference, and uncertainty is handled through a
Bayesian hierarchy. Experiments on synthetic and real-world
datasets demonstrate strong
forecasting and spatial generalization with well-calibrated uncertainty, yielding substantial gains over baselines in strongly nonlocal and partially observed regimes.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from math.OC
Operator Calculus for Population-Based Optimization: Modular Convergence and Finite-Population Guarantees
Pekka Malo, Lauri Viitasaari, Patrik Nummi et al. (+3 more)
Population-based optimizers combine update rules such as mutation, selection, and recombination. When one rule changes, it is often unclear which convergence guarantees survive or how the new combination should be assessed. We develop an operator calculus: an operator is a population-update rule, and the calculus specifies how separately checked effects can be combined. Under explicit regularity and small-step conditions, the leading changes caused by the updates add, yielding reusable building blocks for convergence analysis. The framework distinguishes finding and retaining a good solution, reducing the population's mean objective, and concentrating candidates near an optimizer, and identifies the extra approximation conditions needed for finite evaluation-budget guarantees. Applications include distribution adaptation, recombinative
evolution, and consensus dynamics, with verified nonconvex cases. Controlled experiments on a common nonconvex problem collection show how component effects change with population geometry and the performance measure: a rule can worsen the mean objective yet produce better candidates.
Read the full abstract ▾
All authors Pekka Malo, Lauri Viitasaari, Patrik Nummi, Antti Suominen, Ankur Sinha, Olli Tahvonen
Machine Learning (Statistics)
Gradient boosting for extremes: sampling theory and application to insurance
Stéphane Lhaut and Olivier Lopez
We develop a statistical learning theory for gradient boosting applied to the estimation of
covariate-dependent Generalized Pareto (GP) distributions in the context of Peaks-over-Threshold modeling. After an orthogonal reparametrization of the GP likelihood that diagonalizes its Fisher information
matrix, we cast the estimation problem within the Empirical Risk Minimization (ERM) framework and derive non-asymptotic error bounds for the boosting
estimator. Our analysis accounts for three distinct sources of error in the process: statistical fluctuations, the approximation bias inherent to the asymptotic nature of the GP model-controlled under second-order regular variation-and the approximation error associated with the finite number of boosting iterates, making explicit the resulting bias-variance trade-off. We illustrate the practical benefits of the reparametrization through simulations, showing that it significantly reduces gradient correlation during training and improves convergence stability. The methodology is applied to a medical malpractice insurance
dataset from the Texas Department of Insurance, comprising over 18 000 closed claims. The gradient boosting approach yields a good fit for the tail of settlement cost distributions and reveals that the number of days to settlement is the dominant predictor of tail heaviness, consistent with earlier findings in the reserving literature.
Read the full abstract ▾