Machine Learning (Statistics)
Sliced Inner Product Gromov-Wasserstein Distances
Xiaoyun Gong, Gabriel Rioux and Ziv Goldfeld
The Gromov-Wasserstein (GW) problem provides a framework for aligning heterogeneous
datasets by matching their intrinsic geometry, but its statistical and computational scaling remains an issue for high-dimensional problems. Slicing techniques offer an appealing route to scalability, but, unlike Wasserstein distances, GW problems do not generally admit closed-form solutions in one-dimension. We resolve this problem for the GW problem with inner product cost (IGW), propose a sliced IGW distance that enjoys a natural rotational invariance property, and comprehensively study its structural and computational properties. Numerical experiments validating our theory are presented, followed by applications to heterogeneous clustering of text data and
language model representation comparison.
Read the full abstract ▾
Statistics Applications
Accounting for variable detection functions in temporal abundance modeling via transfer learning
Kevin M. Collins, Erin M. Schliep, Tyler Wagner et al. (+1 more)
Relative abundance, measured as the number of animals caught per unit of sampling effort (CPUE), is commonly used to monitor fish and wildlife populations, largely because sampling methods are cost-effective to implement. Modeling relative abundance, however, requires the assumption that the detection probability is constant across sampling events. This assumption is likely not valid, as the probability of detection often varies as a function of several factors, including the characteristics of individual animals and environmental conditions at the time of sampling. In contrast, methods to estimate absolute abundance, such as capture-recapture (CR), account for variable detection, but are often infeasible to implement across large spatiotemporal scales. Despite this, CR data are sometimes available for species of interest, albeit at smaller spatiotemporal extents. Leveraging information on detection probabilities from CR data to help inform estimates of widely available CPUE data could strengthen inferences about the status of fish and wildlife populations. We propose an approach to (i) learn the effect of environmental
covariates on detection probabilities from CR data and (ii) transfer these detection functions to CPUE models for improved inference. Shown empirically through a simulation study, this approach improves estimates of abundance and the ability to detect temporal trends. We apply our transfer learning method using CR and CPUE data to recreationally important smallmouth bass (Micropterus dolomieu) fisheries in Pennsylvania, USA rivers.
Read the full abstract ▾
All authors Kevin M. Collins, Erin M. Schliep, Tyler Wagner, Christopher K. Wikle
Statistical Methodology
An Object-Oriented Spatial Statistics Approach for Human Activity Space Estimation
Haoyang Wu, Yen-Chi Chen and Adrian Dobra
Human activity spaces are shaped by individual mobility and the built environment, motivating statistical methods that integrate GPS observations with GIS representations of places and routes. We propose a novel methodology to estimate activity spaces in built environments from GPS data within the Object Oriented Spatial Statistics framework. We characterize daily mobility through the distribution of time across spatial polygons and road segments, aiming to capture entity-specific time-use fractions and level-$\gamma$ activity spaces. We develop a time-weighted
estimator to handle irregularly sampled GPS observations. We derive an error bound that quantifies the effects of measurement error, nearest-entity misclassification, temporal gaps, boundary crossings, and day-to-day variability. We also develop a map-augmented representation of daily activity patterns, a dwell-time-weighted distance for clustering daily trajectories, and polygon- and road-based stability summaries. Simulation studies and a real-data application demonstrate that the proposed framework recovers concentrated stationary anchors, interpretable travel corridors, and distinct stabilization behavior for dwelling and movement components, supporting the benefits of weighting under irregular sampling. KEYWORDS: GPS data, GIS, human mobility, space-time geography.
Read the full abstract ▾
Statistics Theory · Cross-listed from cs.LG
Scaling Limits of Long-Context Transformers
Giuseppe Bruno, Shi Chen, Zhengjiang Lin et al. (+2 more)
We study the long-context limit of softmax self-
attention with a fixed query and a random context of $n$ i.i.d. keys on the sphere, viewing the inverse temperature $\beta_n$ as the scaling parameter that decides whether
attention degenerates into uniform averaging or collapses onto the single closest key. We show that the critical scale at which selectivity emerges is determined by the local exponent of the distance-to-query distribution near zero rather than by global features of the context, and scales like $\beta_n^\ast \asymp n^{2/(d-1)}$ for uniform keys on $\mathbb{S}^{d-1}$. Furthermore, we characterize the limiting laws of the ordered
attention weights and of the
attention output across all regimes of $\beta_n$: a subcritical regime in which the output reduces to a local average around $q$ with explicit deterministic bias and Gaussian fluctuations; a critical regime in which a finite collection of nearest keys retains macroscopic mass without single-key collapse; and a supercritical regime in which all mass concentrates on the closest key. Of notable interest is the subcritical case with identity value
matrix where the
attention map approximately implements a backward heat equation.
Read the full abstract ▾
All authors Giuseppe Bruno, Shi Chen, Zhengjiang Lin, Yury Polyanskiy, Philippe Rigollet
Machine Learning (Statistics)
Sinkhorn Treatment Effects: A Causal Optimal Transport Measure
Medha Agarwal and Alex Luedtke
We introduce the Sinkhorn treatment effect, an entropic optimal transport measure of divergence between counterfactual distributions. Unlike classical quantities such as the average treatment effect, this measure captures differences across entire distributions. We analyze this divergence as a statistical functional and show it can be written as a smooth transformation of counterfactual mean
embeddings with an appropriate kernel. This characterization allows us to establish first-order pathwise differentiability in general, and second-order pathwise differentiability under the null hypothesis of equal counterfactual distributions. Leveraging this smoothness, we construct debiased
estimators and use them to obtain asymptotically valid tests for distributional treatment effects with a fixed entropic regularization parameter. Because the power of the test depends on this unknown parameter, we further propose an aggregated test that combines evidence across a grid of regularization choices. Experiments on simulated and image data demonstrate the practical advantages of our
estimator and testing procedure.
Read the full abstract ▾
Statistical Computation · Cross-listed from math.NA
Randomized quasi-Monte Carlo for walk on spheres
Valerie N. P. Ho and Art B. Owen
We investigate the use of randomized quasi-Monte Carlo (RQMC) in walk on spheres
algorithms to solve boundary value problems for functions with Dirichlet boundary conditions in $\mathbb{R}^d$. For harmonic functions with $d=2$, the integrands of interest are periodic indicator functions over regions $\Theta$ in the
torus $\mathbb{T}^k$. We give conditions for $\partial\Theta$ to have $k-1$ dimensional Minkowski content which allows us to use results of He and Wang (2015). The RQMC estimates involve multiple values of $k$. We see sampling variances decreasing with the number $n$ of sample points at slightly better than Monte Carlo rates. The median variance rate in $4$ RQMC methods over $5$ worked examples, including some with $d=3$ and some with nonzero source functions, was slightly better than $O(n^{-1.1})$. The variance reduction factors ranged from $1.8$ to $10.7$ at $n=2^{17}$. None of the four RQMC methods dominated the others. Changes: cite Liu (2025), describe the
lattices that were used, describe usage of AI
Read the full abstract ▾
Statistics Theory
Asymptotics for likelihood ratio tests of boundary points with singular information and unidentifiable nuisance parameters
Karl Oskar Ekvall, Ola Hössjer, Matteo Bottai et al. (+1 more)
We establish the asymptotic distribution of likelihood ratio tests when nuisance parameters are unidentifiable under the null hypothesis, parameters of interest lie on the boundary, and the information
matrix of the identifiable parameters may be singular. Mixture models, change point models and genetic linkage analysis exhibit all three features simultaneously. Under the null hypothesis the limit is the supremum of a chi-bar-squared process, whose marginal distributions are mixtures of chi-squared distributions with weights depending on the nuisance parameter; under local alternatives it is the supremum of a noncentral chi-bar-squared process. Unlike prior work on singular information, where singularity stems from the parameter of interest and changes the limit's form, here it occurs only at isolated values of the nuisance parameter, and under additional conditions the limit keeps its nonsingular form. Classical boundary results and application-specific results for mixtures, change points and linkage analysis are recovered and extended.
Read the full abstract ▾
All authors Karl Oskar Ekvall, Ola Hössjer, Matteo Bottai, J. M. Patrik Albin
Machine Learning (Statistics) · Cross-listed from cs.LG
Sink vs. diagonal patterns as mechanisms for attention switch and oversmoothing prevention
Peter Súkeník, Cristina López Amado, Christoph H. Lampert et al. (+1 more)
This paper studies the role of sinks and diagonal patterns as
attention switch and anti-oversmoothing mechanisms. We analyze
geometric conditions under which sinks can be represented, showing a necessary
alignment between the
embedding of the sink and all other
embeddings. Next, we refine the current understanding of the role of sinks in oversmoothing prevention: we specify the conditions under which dense
attention provably smooths more than sparse
attention, and empirically verify that such conditions are often satisfied in practice. We further prove an equivalence between sinks and hard
attention switch, in which the output of the
attention is identically 0. Finally, we relax the hard
attention switch by allowing
token self-communication: we provide a quantitative comparison of the costs of representing sinks vs.\ diagonal patterns, showing why sinks are favored in pretrained
transformers. The introduction and analysis of diagonal patterns and the generalization of the
attention switch close the gap between what oversmoothing prevention requires and what sinks provide, while also establishing when and why
attention layers act like MLPs if
token communication is not necessary.
Read the full abstract ▾
All authors Peter Súkeník, Cristina López Amado, Christoph H. Lampert, Marco Mondelli
Machine Learning (Statistics) · Cross-listed from cs.CL
A Semantic-Sampling Framework for Evaluating Calibration in Open-Ended Question Answering
Zhanliang Wang, Jiancong Xiao, Ruochen Jin et al. (+3 more)
Calibration measures whether a model's predicted confidence aligns with its empirical accuracy, and is central to the reliable deployment of large
language models (LLMs) in high-stakes domains such as medicine and law. While much recent work focuses on improving LLM calibration, the equally important question of how to evaluate it in realistic settings remains underdeveloped. Open-ended question answering (QA), the most common deployment setting for modern LLMs, is where existing evaluation methods fall short: logit-based metrics need restricted output formats and internal probabilities; verbalized confidence is self-reported and often overconfident; and sampling-based methods rely on task-specific extraction rules without a clear finite-sample target. We introduce Sem-ECE (Semantic-Sampling Expected Calibration Error), a calibration evaluation framework for open-ended QA that samples answers from the model, groups them into semantic classes, and uses the resulting frequencies as confidence. We study two
estimators within this framework: Sem$_1$-ECE, the same-sample self-consistency score, and Sem$_2$-ECE, a held-out variant that separates answer selection from confidence evaluation. We prove both are asymptotically unbiased, and further show that they agree on easy questions but diverge on hard ones with Sem$_2$ achieving strictly smaller calibration error, so their gap also serves as a diagnostic for question difficulty. Experiments on three open-ended QA
benchmarks across five leading commercial LLMs match our theoretical predictions and show that Sem-ECE outperforms verbalized confidence and existing sampling-based methods, while complementing logit-based evaluation when internal probabilities are unavailable.
Read the full abstract ▾
All authors Zhanliang Wang, Jiancong Xiao, Ruochen Jin, Shu Yang, Bojian Hou, Li Shen
Machine Learning (Statistics)
Active Multiple-Prediction-Powered Inference
Nicholas Brawand, Nima Leclerc, Anhthy Ngo et al. (+4 more)
Post-deployment monitoring of healthcare AI requires statistically valid, label-efficient methods, but gold-standard labels from clinician chart review are expensive. Prediction-powered inference (PPI) and active statistical inference (ASI) reduce label cost by combining a small labeled sample with abundant model predictions, but both are restricted to a single predictor, a poor fit for modern clinical pipelines that have multiple predictors of differing cost and accuracy available at inference time. We propose Active Multiple-Prediction-Powered Inference (AM-PPI), which routes each instance to a cost-appropriate predictor subset, samples gold-standard labels in proportion to the chosen subset's residual uncertainty, and reweights predictions to minimize
estimator variance, all under a single deployment-time budget. AM-PPI generalizes ASI to leverage multiple predictors and extends Multiple-PPI from global per-predictor allocation to per-instance adaptive routing. We derive closed-form Karush-Kuhn-Tucker (KKT) conditions for all three decisions and prove, via biconvexity and strong duality, that the resulting fixed point is a global optimum despite the joint problem being non-jointly-convex. We establish asymptotic normality with valid coverage, minimum-variance unbiasedness within the linear-prediction augmented inverse propensity weighted (AIPW) class, and a closed-form criterion identifying when multiple predictors help. On synthetic data and three healthcare monitoring tasks, AM-PPI produces 10 to 40 percent narrower confidence intervals (CIs) than single-predictor ASI in the budget regime where routing matters, and matches the better baseline elsewhere.
Read the full abstract ▾
All authors Nicholas Brawand, Nima Leclerc, Anhthy Ngo, Matthew Peterson, Sriram Vishwanath, Laith Alhussein, Ben Wellner
Machine Learning (Statistics) · Cross-listed from cs.LG
Queryable LoRA: Instruction-Regularized Routing Over Shared Low-Rank Update Atoms
Omatharv Bharat Vaidya, Connor T. Jerzak, Nhat Ho et al. (+1 more)
We present a data-adaptive method for parameter-efficient fine-tuning of large neural networks. Standard low-rank adaptation methods improve efficiency by restricting each layer update to a fixed low-rank form, but this static parameterization can be too rigid when the appropriate correction depends on the input and on the evolving depth-wise computation of the network. Our approach replaces a purely layer-local adapter with a shared queryable
memory of low-rank update atoms. For each block of layers, the model forms a query from the current low-rank state and a running summary of previous blocks, uses this query to retrieve a content-dependent combination of shared update components via
attention, and applies the resulting routed operator within the low-rank bottleneck. In this way, the method retains the efficiency and scalability of low-rank adaptation while allowing the effective update to vary across inputs and to share reusable structure across layers. The resulting architecture provides a principled middle ground between static LoRA-style updates and fully generated parameter updates: it remains compact and parameter-efficient while supporting dynamic, context-sensitive adaptation. Further, we incorporate instruction-regularization by augmenting routing logits with a
language-induced prior over update atoms, thereby biasing the selection of low-rank transformations toward semantically relevant directions without generating unconstrained parameter updates. Experiments on noisy non-linear regression tasks and LLM fine-tuning suggest that this queryable update-
memory formulation can improve final test performance and training stability compared to standard low-rank adaptation, while using a comparable number of trainable parameters.
Read the full abstract ▾
All authors Omatharv Bharat Vaidya, Connor T. Jerzak, Nhat Ho, Chandrajit Bajaj
Statistical Methodology
Rolling-Origin Conformal Prediction under Local Stationarity and Weak Dependence
Stanisław M. S. Halkiewicz
We propose and analyse rolling-origin conformal prediction for time-series
forecasting. The method calibrates the conformal
quantile against the $m$ most recent pseudo-out-of-sample forecast errors, adapting to serial dependence,
volatility clustering, and distributional drift that invalidate classical conformal guarantees. Under Hölder-$\beta$ local stationarity and $\alpha$-mixing, we establish a four-term coverage-error decomposition and derive the optimal calibration window $m^{\star} \asymp T^{2\beta/(2\beta+1)}$ with coverage-error rate $O(T^{-\beta/(2\beta+1)})$. A Le Cam two-point construction shows this rate is minimax-optimal over the Hölder-$\beta$ model class. The Bahadur representation is proved under both $\alpha$-mixing and the physical-dependence framework of Wu (2005). An oracle inequality formalises Winkler cross-validation as an adaptive window selector; the required uniform concentration condition is established in an appendix. Validation on six real series and 93 M4 competition series confirms the theory: rolling-origin calibration outperforms full-history calibration in 86% of comparisons (median Winkler improvement 12.3%), maintains coverage within $\pm2\%$ of the 90% target at short and medium horizons, and the cross-frequency log-log regression slope $0.614$ ($95\%$ CI $[0.424, 0.805]$) is consistent with the theoretical $2/3$ after controlling for frequency fixed effects.
Read the full abstract ▾
Statistics Theory
On Observation Time for Recovering Latent Hawkes Networks
Jonas Linkerhägner, Michele Bortolasi, Lorenzo Baldassari et al. (+2 more)
Dynamics of interacting systems in engineering, society, and nature often evolve over latent networks that govern which entities can interact. We study the problem of inferring these networks from event-based observations, which arise naturally in finance, seismology, and neuroscience. While there is substantial algorithmic work addressing this important problem, theoretical results are scarce. In this paper we ask the following fundamental question: what is the minimum time that one must observe the dynamics in order to exactly recover the underlying network, as a function of the number $d$ of interacting entities? For a class of stationary Hawkes processes with sparse, weak interactions, we prove that an observation time of order $\log d$ is sufficient and necessary. For the upper bound we construct a two-stage
estimator that uses clipped and binned event data for screening, followed by a least-squares refinement, and apply concentration bounds derived from the Poisson cluster representation. For the lower bound we combine Fano's inequality with Jacod's Girsanov formula for point processes on a suitable subclass of networks.
Read the full abstract ▾
All authors Jonas Linkerhägner, Michele Bortolasi, Lorenzo Baldassari, Maarten V. de Hoop, Ivan Dokmanić
Statistical Methodology
Statistical Design of Pragmatic Trials Using Electronic Health Record Data when Outcome Assessments are Uncontrolled and Irregular
Jennifer F. Bobb, Sungtaek Son, Melissa L. Anderson et al. (+3 more)
Pragmatic trials increasingly define outcomes using real-world data such as electronic health records, where assessments are collected during routine care rather than at fixed timepoints. Consequently, these uncontrolled assessments may be irregular, sparse, and affected by the intervention (intervention-dependent assessments), which can lead to biased treatment effect estimates. We developed a simulation study to inform the statistical approach for trials with uncontrolled assessments, which we applied to the MI-CARE pragmatic trial. Using a pre-trial cohort mimicking eligibility and outcome measurement, we estimated assessment frequency and timing and combined these estimates with assumptions about how the intervention effects might impact assessment. We simulated sparse and intervention-dependent assessments and compared single-measure approaches with longitudinal models using all scores. Under intervention-dependent assessments, we found that naive methods such as using the best score or using a randomly selected score without adjusting for measurement timing produced substantial bias. Models that adjusted flexibly for the follow-up timing estimated time-point specific or time-averaged treatment effects without bias. Simulation results informed the selection of the statistical approach for the MI-CARE trial. Among unbiased methods, the most powerful was a linear mixed model with exponential correlation structure, adjustment for time since baseline, and a time-varying intervention effect to estimate the intervention effect at the end of the intervention window. Future studies can use pre-trial data to conduct a simulation study tailored to the trial's data features to inform the analytic approach. Trials with uncontrolled assessments should consider the potential for intervention-dependent assessments and select an appropriate method to avoid bias.
Read the full abstract ▾
All authors Jennifer F. Bobb, Sungtaek Son, Melissa L. Anderson, Noorie Hyun, Lynn L. DeBar, Katharine A. Bradley
Statistics Applications
Transfer Learning for Dead Fuel Moisture Prediction Using Time-Warping Recurrent Neural Networks
Jonathon Hirschi, Jan Mandel and Adam Kochanski
This paper proposes a time-warping transfer learning method, a technique for temporally rescaling the learned dynamics of a recurrent neural network (RNN) with a Long Short-Term
Memory (LSTM) layer to enable task transfer across fuel moisture classes. Fuel moisture content (FMC) is divided into idealized classes based on characteristic lag time. Large quantities of real-time data are available for 10h fuels from sensors on weather stations, but observations of other fuel classes are sparse in space and time. We use transfer learning to adapt an RNN pretrained on 10h FMC to predict FMC for 1h, 100h, and 1000h fuels. We validate this method using data from a landmark field study conducted in Oklahoma that was used to calibrate the state-of-the-art Nelson fuel moisture model.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Embedding Dimension Lower Bounds for Universality of Deep Sets and Janossy Pooling
Ali Syed, Aditya Nambiar and Jonathan W. Siegel
In many practical applications it is important to build
symmetries into neural network architectures. Consider the important case of permutation
symmetry on point clouds consisting of $n$ points in $d$ dimensions. In this case the network learns a function on a set of $n$ points in $\mathbb{R}^d$, and a natural paradigm for constructing invariant networks is Janossy pooling, which generalizes the popular Deep Sets architecture. We study the universality of this approach, in particular the important question of how large the
embedding dimension must be to guarantee universality of this architecture. Specifically, using a novel technique, we prove new lower bounds on the required size of this
embedding dimension. For Deep Sets, this gives the correct minimal dimension up to a constant factor for all $d > 1$. For $k$-ary Janossy pooling, we prove the first non-trivial lower bound on the required
embedding dimension when $k > 1$.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Convergence Analysis of Newton's Method for Neural Networks in the Overparameterized Limit
Konstantin Riedl, Konstantinos Spiliopoulos and Justin Sirignano
A convergence analysis is developed for the regularized Newton method for training neural networks (NNs) in the overparameterized limit. As the number of hidden units tends to infinity, the NN training dynamics converge in probability to the solution of a deterministic limit equation involving a "Newton neural tangent kernel" (NNTK). Explicit rates characterizing this convergence are provided and, in the infinite-width limit, we prove that the NN converges exponentially fast to the target data (i.e., a global minimizer with zero loss). We show that this convergence is uniform across the frequency spectrum, addressing the spectral bias inherent in gradient descent. The
eigenvalues of the NTK for gradient descent accumulate at zero, leading to slow convergence for target data with high-frequency components. In contrast, the NNTK has uniformly lower bounded
eigenvalues if the regularization parameter is selected appropriately, allowing Newton's method to converge more quickly for data with high-frequency components. Mathematical challenges that need to be addressed include the implicit parameter update of the Newton method with a potentially indefinite Hessian
matrix and the fact that the dimension of this linear system of equations tends to infinity as the NN width grows. This substantially complicates deriving the training dynamics in the overparameterized limit as well as proving the convergence of the finite-width dynamics thereto. Our analysis identifies a scaling formula for selecting the regularization parameter, which we show can vanish at a suitable rate as the NN width becomes larger. In addition, we prove that, for sufficiently large numbers of hidden units, the regularized Hessian remains positive definite during training and the Newton updates for individual NN parameters converge to zero, demonstrating that the model behaves as a linearization around the initialization.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
When Attention Beats Fourier: Multi-Scale Transformers for PDE Solving on Irregular Domains
Brandon Yee, Pairie Koh, Jack Rodriguez et al. (+1 more)
We study the problem of architecture selection for deep learning models trained to solve partial differential equations (PDEs), asking when
transformer-based architectures with learned
attention outperform
Fourier-domain neural operators. We introduce the Multi-Scale
Attention Transformer (\msat{}), a deep learning architecture that encodes spatiotemporal solution histories as
token sequences and trains end-to-end via a composite supervised objective with optional physics-informed regularization terms. We conduct a comprehensive empirical evaluation against nine baselines – including physics-informed neural networks (PINNs), neural operators (FNO, DeepONet, GNOT), and state-space models (Mamba-NO) – across five
benchmark problems from the PINNacle suite, using identical train/test splits and reference data for all methods. \msat{} achieves state-of-the-art generalization on complex geometry problems ($L^2_\mathrm{rel} = 0.0101$ on Heat2D-CG, a $3.7\times$ improvement over FNO) at $34\,\mathrm{s}$ total inference vs.\ $120{,}812\,\mathrm{s}$ for Mamba-NO. Ablation studies over the physics regularization component reveal a precise inductive bias tradeoff: physics priors reduce test error on
diffusion-dominated problems but degrade generalization on chaotic and recirculating-flow regimes, directly characterizing the prior misspecification boundary. Approximation error bounds as a function of domain boundary complexity $\kappa$ provide a theoretical basis for these empirical findings and a principled rule for architecture selection.
Read the full abstract ▾
All authors Brandon Yee, Pairie Koh, Jack Rodriguez, Mihir Tekal
Statistics Applications
Quantifying Exposure Information Uncertainty in Regional Risk Assessment
Chenhao Wu and Henry Burton
Exposure characterization in regional risk assessment aims to assign physical properties to the assets of interest so they can be associated with damage and loss functions. While this process has benefited from the growing availability of public infrastructure inventories, these
datasets often lack the detailed attributes required for high-resolution risk assessment. Missing attributes are commonly inferred using predictive models or engineering-based rulesets. However, these imputations are inherently imperfect and can introduce bias and additional uncertainty in regional risk estimates. This study proposes a methodology to quantify the bias and uncertainty in regional risk assessment that arises from probabilistic exposure characterization. By integrating analytical and simulation-based approaches, the methodology decomposes the total uncertainty into contributions from incomplete exposure information as well as other sources, including hazard and damage characterization. This decomposition clarifies how bias and uncertainty associated with missing exposure information are generated and propagated through the risk assessment pipeline. The methodology is applied to both bridge-specific and regional risk assessments. A high-resolution bridge exposure inventory is developed using a data augmentation framework that combines publicly available information with machine learning and engineering-based imputation methods.
Read the full abstract ▾
Machine Learning (Statistics)
A Note on Non-Negative $L_1$-Approximating Polynomials
Jane H. Lee, Anay Mehrotra and Manolis Zampetakis
$L_1$-Approximating
polynomials, i.e.,
polynomials that approximate indicator functions in $L_1$-norm under certain distributions, are widely used in computational learning theory. We study the existence of non-negative $L_1$-approximating
polynomials with respect to Gaussian distributions. This is a stronger requirement than $L_1$-approximation but weaker than sandwiching
polynomials (which themselves have many applications). These non-negative approximating
polynomials have recently found uses in smoothed learning from positive-only examples. In this short note, we prove that every class of sets with Gaussian surface area (GSA) at most $\Gamma$ under the standard Gaussian admits degree-$k$ non-negative
polynomials that $\eps$-approximate its indicator functions in $L_1$-norm, for $k=\tilde{O}(\Gamma^2/\varepsilon^2)$. Equivalently, finite GSA implies $L_1$-approximation with the stronger pointwise guarantee that the approximating
polynomial has range contained in $[0,\infty)$. Up to a constant-factor, this matches the degree of the best currently known Gaussian $L_1$-approximation degree bound without the non-negativity constraint.
Read the full abstract ▾
Statistical Methodology · Cross-listed from econ.EM
Vibe Econometrics and the Analysis Contract
Lydia Ashton
"Vibe coding" and "vibe analytics" have been framed as a democratization of technical capability. This paper argues that AI-assisted methodology more broadly, or what I call "vibe methodology," also democratizes the failure modes specific to each domain. When AI assists with methods whose validity depends on assumptions that cannot be verified from the output alone (a class I call "vibe inference"), the failure surface is structurally different: the output does not reliably signal invalidity, and when it does, recognizing the signal requires the expertise the workflow bypasses. I focus on "vibe econometrics," the subset of AI-assisted causal analysis where identification can be named faster than it can be audited. The claim of this paper is not that AI invents inferential failures that did not previously exist, but that it changes their incidence, observability, and persuasive force enough to create a practically distinct governance problem. This results in three failure modes: method-data mismatch, where AI bypasses expertise at execution; confidence laundering, where AI amplifies the credibility of formatted output; and invisible forking, which spans both. What is new is not the failure modes but AI's industrialization of their packaging. The barrier between naming a method and executing it has collapsed, and weak foundations, dressed as rigorous analysis, now reach audiences at a scale, speed, and polish that previously required expertise. I propose the Analysis Contract, a pre-commitment framework that adapts the logic of pre-analysis plans and the Causal Roadmap to the AI-assisted setting. The contract imposes three conditions before a causal claim is made: a method-data contract, a data audit, and a pre-commitment statement defining what would count as a disconfirming result. The framework generalizes across domains of vibe inference through domain-specific instantiation.
Read the full abstract ▾
Statistical Methodology
Empirical Bayes Rebiasing
Wanyi Ling, Sida Li, Junming Guan et al. (+1 more)
We study methods for simultaneous analysis of many noisy and biased estimates, each paired with an even noisier estimate of its own bias. The analyst's goal is to construct short calibrated intervals for each parameter. The standard debiasing approach, which subtracts the bias estimate from each biased estimate, inflates variance and yields long intervals. In this paper, we propose an empirical Bayes rebiasing strategy that starts from the fully debiased estimates and learns from data how much bias to reintroduce by estimating the unknown bias distribution. We provide convergence rates for the coverage of our intervals when the bias distribution is estimated using nonparametric maximum likelihood. Furthermore, we demonstrate substantial precision gains in prediction-powered inference, including pairwise LLM win-rate evaluations, as well as for inference of direct genetic effects in family-based GWAS.
Read the full abstract ▾
All authors Wanyi Ling, Sida Li, Junming Guan, Nikolaos Ignatiadis
Machine Learning (Statistics) · Cross-listed from astro-ph.SR
Period spacings and global seismic parameters for K2 red giants using deep learning
Nipun Ghanghas, Siddharth Dhanpal, Shravan Hanasoge et al. (+2 more)
Gravity-mode period spacings (DPi_1) of red
giants probe the
stellar core directly, constraining its structure, mass and evolutionary state. Their measurement requires resolving narrow, densely spaced mixed modes and has so far relied on the four-year baseline of Kepler. Recovering DPi_1 from the much shorter (~80-day) baselines typical of K2 remains largely unexplored at the ensemble scale. We develop an automated machine-learning technique to measure global asteroseismic parameters and
gravity-mode period spacings for red
giants from single-campaign K2 photometry. Two deep residual neural networks take the full-resolution power spectrum of an ~80-day light curve as input, without background fitting or mode identification, and return a probability distribution for each parameter, yielding a point estimate and asymmetric uncertainties. They are trained on ~8 million synthetic spectra and evaluated on held-out synthetics, on Kepler data degraded to K2-like resolution, and on K2 observations. On Kepler data at K2-like resolution, numax and Dnu are recovered for 96% and 91% of
stars with robust dispersions of 3.6% and 1.2%; DPi_1 for 22%, with a dispersion of 0.9%. Applied to 18,704 K2 red
giants, the inferred numax and Dnu agree with catalogue values to 5.8% and 1.9%, the additional scatter arising from known K2 artefacts. Among the 2,059 young red
giants we obtain DPi_1 for 232
stars with a median fractional uncertainty of 1.3%, following the same Dnu-DPi_1 sequence as the Kepler sample although the training set carries no imprint of that relation. We show that numax, Dnu and - for a subset of young red
giants - DPi_1 can be recovered from a single ~80-day K2 campaign within an automated, probabilistic machine-learning framework. This approach can also be extended to short-baseline samples from TESS and, in future, Roman and PLATO.
Read the full abstract ▾
All authors Nipun Ghanghas, Siddharth Dhanpal, Shravan Hanasoge, Praneeth Netrapalli, Karthikeyan Shanmugam
Statistical Methodology
Semi-supervised Method for Risk Prediction with Doubly Censored EHR Data
Jie Zhou, Enhao Wang and Xuan Wang
The rapid expansion of large-scale electronic health record (EHR) data offers unique opportunities to improve the accuracy and efficiency of clinical risk estimation. Yet, because clinical events may occur outside the recording health system, clinical event outcomes are frequently subject to double censoring (both left and right). Besides, gold-standard event times can often only be ascertained through labor-intensive manual chart reviews, yielding labels for only a small subset of patients. Reliance on this limited labeled set alone is limited in efficiency, whereas widely available surrogate outcomes such as the time to first diagnostic code or first disease mention are error-prone and can yield biased estimates if used directly. Semi-supervised learning (SSL) methods provide a principled way to integrate labeled and unlabeled data, and prior work has demonstrated their advantages in settings with binary or right-censored outcomes. However, existing approaches do not accommodate double censoring for risk prediction, which poses additional methodological challenges. To address this gap, we develop a novel SSL framework for risk prediction that combines a small set of gold-standard labels with large-scale surrogate information under double censoring. We establish the theoretical validity of the proposed
estimator. Through extensive simulation studies, we show that our method substantially improves estimation efficiency relative to existing supervised
estimators (based on the labeled data). Finally, we demonstrate its practical value by applying it to study risk factors for type 2 diabetes (T2D) using EHR data from a health system in the US.
Read the full abstract ▾
Machine Learning (Statistics)
Semiparametric Efficient Test for Interpretable Distributional Treatment Effects
Houssam Zenati and Arthur Gretton
Distributional treatment effects can be invisible to means: a treatment may preserve average outcomes while changing tails, modes, dispersion, or rare-event probabilities. Kernel tests can detect discrepancies between interventional outcome laws, but global tests do not reveal where the laws differ. We propose DR-ME, to our knowledge the first semiparametrically efficient finite-location test for interpretable distributional treatment effects. DR-ME evaluates an interventional kernel witness at learned outcome locations, returning causal-discrepancy coordinates rather than only a global rejection. From observational data, we derive orthogonal doubly robust kernel features whose centered oracle form is the canonical gradient of this finite witness. For fixed locations, we characterize the local testing limit: DR-ME is chi-square calibrated under the null, has noncentral chi-square local power, and uses the covariance whitening that optimizes local signal-to-noise for discrepancies visible through the selected coordinates. This efficient local-power geometry yields a principled location-learning criterion, with sample splitting preserving post-selection validity. Experiments show near-nominal type-I error, competitive power against global doubly robust kernel tests, and interpretable learned locations that localize distributional effects in a semi-synthetic medical-imaging study.
Read the full abstract ▾
Statistical Methodology
Randomization Tests for Distributions of Individual Treatment Effects via Combined Rank Statistics
David Kim, Yongchang Su, Jake Bowers et al. (+1 more)
What proportion of treated units actually benefited from an experimental intervention? What is the median or the largest individual treatment effect? This paper develops methods for answering such questions about the distribution of individual causal effects in randomized experiments. Existing approaches require the analyst to select a rank-based test statistic before observing the data. A poor choice can substantially reduce power, while searching over multiple test statistics and adjusting for multiplicity using Bonferroni correction also incurs power loss. We propose inference procedures that adaptively combine multiple rank-based statistics while maintaining finite-sample validity. For stratified experiments, we further develop weighting schemes that effectively aggregate evidence across strata of heterogeneous sizes. The resulting combined test achieves power comparable to, or exceeding, that of the best individual test, without requiring prior knowledge of the optimal statistic. When applied to a randomized experiment evaluating a teacher training program, the combined test suggests that roughly half of treated teachers benefited, whereas a single rank-based test may indicate only a small minority. Thus, the choice of test determined whether the program appears broadly successful or narrowly effective.
Read the full abstract ▾
All authors David Kim, Yongchang Su, Jake Bowers, Xinran Li
Statistical Methodology
BAMIFun: Bayesian Multiple Imputation for Functional Data
Ziren Jiang, Lei Xuan, Eric F. Lock et al. (+1 more)
Missing data are pervasive in modern functional
datasets, where trajectories are often sparsely or irregularly observed. Although Functional Principal Component Analysis (FPCA) is widely used to reconstruct incomplete curves, existing approaches typically employ single imputation, leading to overly optimistic inferences in downstream analyses. To address these challenges, we develop a novel
Bayesian multiple imputation framework for functional data (BAMIFun). For single-level functional data, we impose a
Bayesian low-rank model that incorporates penalized spline representations to enforce smoothness of the functional domain and derive an efficient Gibbs sampler
algorithm for posterior computation. In addition, we demonstrate and validate how to properly account for estimation uncertainties in downstream analysis. Furthermore, we extend the framework to multiway functional data using Functional Tensor Singular Value Decomposition (FTSVD) model, enabling
Bayesian multiple imputation in settings not supported by existing methods. Simulation studies show that BAMIFun achieves substantially improved coverage and more reliable downstream inference compared to existing methods, while maintaining similar imputation accuracy. Case studies using a physical activity
dataset and an infant gut microbiome
dataset further demonstrate the practical advantages of our proposed methods under severe missingness. The code is available at https://github.com/ZirenJiang/BAMIFun.
Read the full abstract ▾
All authors Ziren Jiang, Lei Xuan, Eric F. Lock, Erjia Cui
Statistical Computation · Cross-listed from cs.AI
Abductive Reasoning with Probabilistic Commonsense
Joseph Cotnareanu, Chiara Roverato, Han Zhou et al. (+3 more)
Recent efforts to improve the
reasoning abilities of Large
Language Models (LLMs) have focused on integrating formal logic solvers within neurosymbolic frameworks. A key challenge is that formal solvers lack commonsense world knowledge, preventing them from making
reasoning steps that humans find obvious. Prior methods address this by using LLMs to supply missing commonsense assumptions, but these approaches implicitly assume universal agreement on such commonsense facts. In reality, commonsense beliefs vary across individuals. We propose a probabilistic framework for abductive commonsense
reasoning that explicitly models this variation, aiming to determine whether most people would judge a statement as true or false. We introduce Probabilistic Abductive CommonSense (PACS), a novel
algorithm that uses an LLM and a formal solver to sample proofs as observations of individuals' distinct commonsense beliefs, and aggregates conclusions across these samples. Empirically, PACS outperforms chain-of-thought
reasoning, prior neurosymbolic methods, and search-based approaches across multiple
benchmarks.
Read the full abstract ▾
All authors Joseph Cotnareanu, Chiara Roverato, Han Zhou, Didier Chetelat, Yingxue Zhang, Mark Coates
Machine Learning (Statistics) · Cross-listed from math.OC
Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems
Yiyang Shen, Yutian He, Weiran Wang et al. (+1 more)
We study a class of bilevel optimization problems in which both the upper- and lower-level problems have minimax structures. This setting captures a broad range of emerging applications. Despite the extensive literature on bilevel optimization and minimax optimization separately, existing methods mainly focus on bilevel optimization with lower-level minimization problems, often under strong convexity assumptions, and are not directly applicable to the minimax lower-level setting considered here. To address this gap, we develop penalty-based first-order methods for bilevel minimax optimization without requiring strong convexity of the lower-level problem. In the deterministic setting, we establish that the proposed method finds an $\epsilon$-KKT point with $\tilde{O}(\epsilon^{-4})$ oracle complexity. We further show that bilevel problems with convex constrained lower-level minimization can be reformulated as special cases of our framework via Lagrangian duality, leading to an $\tilde{O}(\epsilon^{-4})$ complexity bound that improves upon the existing $\tilde{O}(\epsilon^{-7})$ result. Finally, we extend our approach to the
stochastic setting, where only
stochastic gradient oracles are available, and prove that the proposed
stochastic method finds a nearly $\epsilon$-KKT point with $\tilde{O}(\epsilon^{-9})$ oracle complexity.
Read the full abstract ▾
All authors Yiyang Shen, Yutian He, Weiran Wang, Qihang Lin
Statistical Methodology
Cellwise and Casewise Robust Multivariate Regression with Inference
Fabio Centofanti, Mia Hubert and Peter J. Rousseeuw
Multivariate linear regression is a fundamental statistical task, but classical
estimators such as ordinary least squares are highly sensitive to outliers. These may occur as casewise outliers that affect entire observations, or as outlying
cells, that are individual contaminated entries in the predictor and/or response
matrix. Moreover, modern
datasets frequently contain missing values and are high-dimensional. To address these challenges we propose the cellwise multivariate regression (cellMR)
estimator, a robust regression method that simultaneously accommodates casewise and cellwise outliers, missing data, and high dimensionality. The approach builds on a cellwise robust covariance
estimator and uses ridge regularization for numerical stability. We further introduce cellBoot, a novel bootstrap-based inference procedure tailored to the cellMR framework. Relying on indirect inference, cellBoot provides asymptotically valid confidence intervals that are robust to casewise and cellwise contamination. We derive influence functions and breakdown bounds of the regression
estimator and prove the asymptotic validity of the cellBoot confidence intervals. Simulations and a real genomics application illustrate the strong finite-sample performance of the proposed methods.
Read the full abstract ▾