Machine Learning (Statistics)
Analytical Extraction of Conditional Sobol' Indices via Basis Decomposition of Polynomial Chaos Expansions
Shijie Zhong and Jiangfeng Fu
In uncertainty quantification, evaluating sensitivity measures under specific conditions (i.e., conditional Sobol' indices) is essential for systems with parameterized responses, such as spatial fields or varying operating conditions. Traditional approaches often rely on point-wise modeling, which is computationally expensive and may lack consistency across the parameter space. This paper demonstrates that for a pre-trained global
Polynomial Chaos Expansion (PCE) model, the analytical conditional Sobol' indices are inherently embedded within its basis functions. By leveraging the tensor-product property of PCE bases, we reformulate the global expansion into a set of analytical coefficient fields that depend on the conditioning variables. Based on the preservation of orthogonality under conditional probability measures, we derive closed-form expressions for conditional variances and Sobol' indices. This framework bypasses the need for repetitive modeling or additional sampling, transforming conditional sensitivity analysis into a purely algebraic post-processing step. Numerical
benchmarks indicate that the proposed method ensures physical coherence and offers superior numerical robustness and computational efficiency compared to conventional point-wise approaches.
Read the full abstract ▾
Statistics Applications · Cross-listed from cs.CL
Mind the Unseen Mass: Unmasking LLM Hallucinations via Soft-Hybrid Alphabet Estimation
Hongxing Pan, Yingying Guo, Wenqing Kuang et al. (+1 more)
This paper studies uncertainty quantification for large
language models (LLMs) under black-box access, where only a small number of responses can be sampled for each query. In this setting, estimating the effective semantic alphabet size–that is, the number of distinct meanings expressed in the sampled responses–provides a useful proxy for downstream risk. However, frequency-based
estimators tend to undercount rare semantic modes when the sample size is small, while graph-spectral quantities alone are not designed to estimate semantic occupancy accurately. To address this issue, we propose SHADE (Soft-Hybrid Alphabet Dynamic
Estimator), a simple and interpretable
estimator that combines Generalized Good-Turing coverage with a heat-kernel trace of the normalized
Laplacian constructed from an entailment-weighted graph over sampled responses. The estimated coverage adaptively determines the fusion rule: under high coverage, SHADE uses a convex combination of the two signals, while under low coverage it applies a LogSumExp fusion to emphasize missing or weakly observed semantic modes. A finite-sample correction is then introduced to stabilize the resulting cardinality estimate before converting it into a coverage-adjusted semantic
entropy score. Experiments on pooled semantic alphabet-size estimation against large-sample references and on QA incorrectness detection show that SHADE achieves the strongest improvements in the most sample-limited regime, while the performance gap narrows as the number of samples increases. These results suggest that hybrid semantic occupancy estimation is particularly beneficial when black-box uncertainty quantification must operate under tight sampling budgets.
Read the full abstract ▾
All authors Hongxing Pan, Yingying Guo, Wenqing Kuang, Jiashi Lu
Statistics Applications
And Quiet Does Not Flow the Don: Statistical Analysis of a Quarrel Between Nobel Prize Laureates
Nils Lid Hjort
The Nobel Prize in literature 1965 was awarded Mikhail Sholokhov (1905-1984), for the epic novel Tikhij Don about Cossack life and the birth of a new Soviet society (And Quiet Flows the Don, or The Quiet Don, in different translations). Sholokhov has been compared to Tolstoy and was at least one and two generations ago called `the greatest of our writers' in the Soviet Union. In Russia alone his books have been published in more than a thousand editions, selling in total more than sixty million copies. He was an elected member of the USSR Supreme Soviet, the USSR Academy of Sciences, and of the CPSU Central Committee. But in the autumn of 1974 an article was published in Paris, Stremya `Tihogo Dona' (Zagadki romana (`The Rapids of Quiet Don: the Enigmas of the Novel'), by the author and critic D$^*$. He claimed that Tikhij Don was not at all Sholokhov's work, but that it rather was written by Fiodor Kriukov, a more obscure author who fought against bolshevism and died in 1920. The article was given credibility and prestige by none other than Aleksandr Solzhenitsyn (a Nobel prize winner five years after Sholokhov), who wrote a preface giving full support to D$^*$'s conclusion. Scandals followed, also touching the upper echelons of Soviet society, and Sholokhov's reputation was faltering abroad (see e.g. Doris Lessing's (1997) comments; `vibrations of dislike instantly flowed between us'). Are we in fact faced with one of the most flagrant cases of plagiarism in the history of literature?
Read the full abstract ▾
Statistical Methodology
Transfer Learning for Degree-Corrected Mixed Membership Network Models
Yong He, Kangxiang Qin and Haoran Tang
Statistical analysis of network data has attracted considerable
attention in recent years, due to the rapid advancement of well-trained network models and the accessibility of large public network
datasets. In this article, we propose a transfer learning procedure for boosting estimation accuracy of a target network structure based on the well-known Degree-Corrected Mixed-Membership (DCMM) model in the literature. By leveraging useful information from informative source
datasets, we theoretically prove that the transfer learning procedure greatly improve the estimation accuracy for the target connection probability
matrix. Our theoretical analysis also reveals that the benefits from knowledge transfer in this context attributes to the enlarged
eigenvalue gap of the target connection probability
matrix. Additionally, we propose a random projection step in conjunction with the conventional aggregation procedure to alleviate the heavy computational burden in practice. In the presence of potentially harmful sources, we further provide an iterative truncation
algorithm for selecting useful
datasets and avoiding negative transfer. Numerical results showcase the practical utility of our methods in real-world network
dataset analysis, including journal citation network
dataset and international trade network
dataset.
Read the full abstract ▾
Statistical Methodology
The General Formulation of Loss-Based Priors for Parameter Spaces
Cristiano Villa
Loss-based priors assign probability mass to parameter values according to the inferential loss incurred when they are excluded from the parameter space, and provide a general solution for discrete parameters. Extending this idea to continuous settings is challenging, as the exclusion of a single point induces no loss. We propose a neighbourhood-exclusion framework in which inferential loss is defined by removing a local region around each parameter value. Under standard regularity conditions, this yields a class of prior distributions driven by the local geometry of the Kullback–Leibler divergence. In one dimension, the resulting prior coincides with Jeffreys' prior, while in higher dimensions it leads to a family of priors indexed by the geometry of the exclusion region. The proposed formulation provides a unified extension of loss-based priors and offers a
geometric interpretation of objective prior construction beyond isotropic settings.
Read the full abstract ▾
Machine Learning (Statistics)
Fast estimation of Gaussian mixture components via centering and singular value thresholding
Huan Qing
Estimating the number of components is a fundamental challenge in unsupervised learning, particularly when dealing with high-dimensional data with many components or severely imbalanced component sizes. This paper addresses this challenge for classical Gaussian mixture models. The proposed
estimator is simple: center the data, compute the singular values of the centered
matrix, and count those above a threshold. No iterative fitting, no likelihood calculation, and no prior knowledge of the number of components are required. We prove that, under a mild separation condition on the component centers, the
estimator consistently recovers the true number of components. The result holds in high-dimensional settings where the dimension can be much larger than the sample size. It also holds when the number of components grows to the smaller of the dimension and the sample size, even under severe imbalance among component sizes. Computationally, the method is extremely fast: for example, it processes ten million samples in one hundred dimensions within one minute. Extensive experimental studies confirm its accuracy in challenging settings such as high dimensionality, many components, and severe class imbalance.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
S2MAM: Semi-supervised Meta Additive Model for Robust Estimation and Variable Selection
Xuelin Zhang, Hong Chen, Yingjie Wang et al. (+2 more)
Semi-supervised learning with
manifold regularization is a classical framework for jointly learning from both labeled and unlabeled data, where the key requirement is that the support of the unknown marginal distribution has the
geometric structure of a Riemannian
manifold. Typically, the Laplace-Beltrami operator-based
manifold regularization can be approximated empirically by the
Laplacian regularization associated with the entire training data and its corresponding graph
Laplacian matrix. However, the graph
Laplacian matrix depends heavily on the prespecified similarity metric and may lead to inappropriate penalties when dealing with redundant or noisy input variables. To address the above issues, this paper proposes a new Semi-Supervised Meta Additive Model (S$^2$MAM) based on a bilevel optimization scheme that automatically identifies informative variables, updates the similarity
matrix, and simultaneously achieves interpretable predictions. Theoretical guarantees are provided for S$^2$MAM, including the computing convergence and the statistical generalization bound. Experimental assessments across 4 synthetic and 12 real-world
datasets, with varying levels and categories of corruption, validate the robustness and interpretability of the proposed approach.
Read the full abstract ▾
All authors Xuelin Zhang, Hong Chen, Yingjie Wang, Tieliang Gong, Bin Gu
Statistics Theory
The interplay between network transitivity and community structure
Mingao Yuan, Irin Rahman, Chengay S Wangchuk et al. (+1 more)
Recent empirical observations suggest that network transitivity is highly correlated with community structure in many real-world networks. In this paper, we theoretically investigate this relationship by deriving the limits of the global and average clustering coefficients for the
geometric block model (GBM). Both limits exhibit a phase transition; specifically, the functional forms of the limit functions differ between the weak and strong community structure strength regimes. For a GBM with balanced communities, the limits of the global and average clustering coefficients are identical, whereas these limits differ for unbalanced communities. In general, the clustering coefficients do not exhibit a monotonic relationship with community structure strength. Particularly, for a balanced GBM where the within-community edge probability is a constant multiple of the between-community edge probability, the limit decreases from $3/4$ to $3/5$ and subsequently increases toward an asymptotic upper bound of $3/4$ as the multiple grows from one. A similar pattern is observed for the global clustering coefficient in unbalanced settings, where both limits exhibit an explicit dependence on community size.
Read the full abstract ▾
All authors Mingao Yuan, Irin Rahman, Chengay S Wangchuk, Minglian Lin
Statistics Applications · Cross-listed from cs.LG
Age-Dependent Heterogeneity in the Association Between Physical Activity and Mental Distress: A Causal Machine Learning Analysis of 3.2 Million U.S. Adults
Yuan Shan
Physical activity (PA) is widely recognized as protective against mental distress, yet whether this benefit varies systematically across population subgroups remains poorly understood. Using pooled data from ten consecutive annual waves of the U.S. Behavioral Risk Factor Surveillance System (2015-2024; n = 3,242,218), we investigate heterogeneity in the association between leisure-time PA and frequent mental distress (FMD, >=14 days/month) across age groups. Survey-weighted logistic regression reveals a striking age gradient: the adjusted odds ratio for PA ranges from 0.89 among young adults (18-24) to 0.50 among adults aged 55-64, with the protective association strengthening monotonically with age. Temporal analysis across all ten years shows that the young-adult PA effect has been eroding over the past decade, with the 18-24 OR reaching 1.01 (null) in both 2018 and 2024 – paralleling the deepening youth mental health crisis. Causal Forest via Double Machine Learning independently identifies age as the dominant driver of treatment effect heterogeneity (feature importance = 0.39, 2.5x the next predictor). E-value sensitivity analysis, propensity score overlap checks, placebo tests, and imputation comparisons confirm the robustness of the findings. These results suggest that the well-documented exercise–mental health link may not generalize to the youngest adult population, whose distress appears increasingly driven by stressors that PA alone cannot mitigate.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.GT
Last-Iterate Guarantees for Learning in Co-coercive Games
Siddharth Chandak, Ramanan Tamizholi and Nicholas Bambos
We establish finite-time last-iterate guarantees for vanilla
stochastic gradient descent in co-coercive games under noisy feedback. This is a broad class of games that is more general than strongly monotone games, allows for multiple Nash
equilibria, and includes examples such as quadratic games with negative semidefinite interaction
matrices and potential games with smooth concave potentials. Prior work in this setting has relied on relative noise models, where the noise vanishes as iterates approach
equilibrium, an assumption that is often unrealistic in practice. We work instead under a substantially more general noise model in which the second moment of the noise is allowed to scale affinely with the squared norm of the iterates, an assumption natural in learning with unbounded action spaces. Under this model, we prove a last-iterate bound of order $O(\log(t)/t^{1/3})$, the first such bound for co-coercive games under non-vanishing noise. We additionally establish almost sure convergence of the iterates to the set of Nash
equilibria and derive time-average convergence guarantees.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Local Linearity of LLMs Enables Activation Steering via Model-Based Linear Optimal Control
Julian Skifstad, Xinyue Annie Yang and Glen Chou
Inference-time LLM
alignment methods, particularly activation steering, offer an alternative to fine-tuning by directly modifying activations during generation. Existing methods, however, often rely on non-anticipative interventions that ignore how perturbations propagate through
transformer layers and lack online error feedback, resulting in suboptimal, open-loop control. To address this, we show empirically that, despite the nonlinear structure of
transformer blocks, layer-wise dynamics across multiple LLM architectures and scales are well-approximated by locally-linear models. Exploiting this property, we model LLM inference as a linear time-varying dynamical system and adapt the classical linear quadratic regulator to compute feedback controllers using layer-wise Jacobians, steering activations toward desired semantic setpoints in closed-loop with minimal computational overhead and no offline training. We also derive theoretical bounds on setpoint tracking error, enabling formal guarantees on steering performance. Using a novel adaptive semantic feature setpoint signal, our method yields robust, fine-grained behavior control across models, scales, and tasks, including state-of-the-art modulation of toxicity, truthfulness, refusal, and arbitrary concepts, surpassing baseline steering methods. Our code is available at: https://github.com/trustworthyrobotics/lqr-activation-steering
Read the full abstract ▾
Statistics Applications
Ground-Level Near Real-Time Modeling for PM2.5 Pollution Prediction
Zachary R. Fox, Janet O. Agbaje, Dakotah Maguire et al. (+6 more)
Air pollution is a worldwide public health threat that can cause or exacerbate many illnesses, including respiratory disease, cardiovascular disease, and some cancers. However, epidemiological studies and public health decision-making are stymied by the inability to assess pollution exposure impacts in near real time. To address this, developing accurate digital twins of environmental pollutants will enable timely data-driven analytics - a crucial step in modernizing health policy and decision-making. Although other models predict and analyze fine particulate matter exposure, they often rely on modeled input data sources and data streams that are not regularly updated. Another challenge stems from current models relying on predefined grids. In contrast, our deep-learning approach interpolates surface level PM2.5 concentrations between sparsely distributed US EPA monitoring stations in a grid-free manner. By incorporating additional, readily available
datasets - including topographic, meteorological, and land-use data - we improve its ability to predict pollutant concentrations with high spatial and temporal resolution. This enables model querying at any spatial location for rapid predictions without computing over the entire grid. To ensure robustness, we randomize spatial sampling during training to enable our model to perform well in both dense and sparse monitored regions. This model is well suited for near real-time deployment because its lightweight architecture allows for fast updates in response to streaming data. Moreover, model flexibility and scalability allow it to be adapted to various geographical contexts and scales, making it a practical tool for delivering accurate and timely air quality assessments. Its capacity to rapidly evaluate multiple scenarios can be especially valuable for decision-making during public health crises.
Read the full abstract ▾
All authors Zachary R. Fox, Janet O. Agbaje, Dakotah Maguire, Javier E. Santos, Jeremy Logan, Maggie Davis, Rima Habre, Jim VanDerslice, Heidi A. Hanson
Machine Learning (Statistics)
Beyond Bellman: High-Order Generator Regression for Continuous-Time Policy Evaluation
Yaowei Zheng, Richong Zhang, Shenxi Wu et al. (+5 more)
We study finite-horizon continuous-time policy evaluation from discrete closed-loop trajectories under time-inhomogeneous dynamics. The target value surface solves a backward
parabolic equation, but the Bellman baseline obtained from one-step recursion is only first-order in the grid width. We estimate the time-dependent generator from multi-step transitions using moment-matching coefficients that cancel lower-order truncation terms, and combine the resulting surrogate with backward regression. The main theory gives an end-to-end decomposition into generator misspecification, projection error, pooling bias, finite-sample error, and start-up error, together with a decision-frequency regime map explaining when higher-order gains should be visible. Across calibration studies, four-scale
benchmarks, feature and start-up ablations, and gain-mismatch stress tests, the second-order
estimator consistently improves on the Bellman baseline and remains stable in the regime where the theory predicts visible gains. These results position high-order generator regression as an interpretable continuous-time policy-evaluation method with a clear operating region.
Read the full abstract ▾
All authors Yaowei Zheng, Richong Zhang, Shenxi Wu, Shirui Bian, Haosong Zhang, Li Zeng, Xingjian Ma, Yichi Zhang