Machine Learning (Statistics)
Covariate-dependent Joint Modeling of Multivariate Ordinal Preferences and Its Connections with Comparison Models
Yujie Chen, Antik Chakraborty and Anindya Bhadra
Multivariate ordinal data along with
covariates are commonly collected in problems ranging from
alignment of
language models with human preferences, as well as in recommender systems. For example, data sets such as MovieLens contain several movies rated on a scale 1–5 by human users, along with their demographic information such as age or gender. Similarly, data sets such as HelpSteer collect human feedback on several attributes such as "helpfulness" or "verbosity" of LLM response on an ordinal scale, with
covariates depending on the LLM
prompt–response pairs. Unfortunately, the standard approaches for modeling these data (a) look at the attributes individually rather than jointly, and (b) often convert the data into pairwise or list-wise win–loss comparisons for fitting models such as Bradley–Terry and Plackett–Luce. Both of these lead to a coarsening of what is actually observed, which we address via a joint
covariate-dependent consecutive ratio Markov random field model. We also show pairwise or listwise comparison models are obtained under restrictions of our joint model, and that joint modeling improves comparisons. We also develop a maximum likelihood inference procedure even in the presence of an intractable normalizer.
Read the full abstract ▾
Machine Learning (Statistics)
Learning Transition Kernels of Jump-Diffusion Processes with Conditional Diffusion Models
Yuzhen Zhao, Yating Liu and Quentin Guibert
We study the problem of learning transition kernels for time-homogeneous jump-
diffusion processes using conditional
diffusion models, with the goal of generating new sample paths from training data consisting of N independent trajectories observed on a high-frequency discrete time grid. On the theoretical side, we establish non-asymptotic bounds for the conditional score estimation error and for the KL divergence between the laws of the true and generated discretely observed paths. On the numerical side, we first evaluate our method on synthetic data to assess the theoretical findings and
benchmark its performance against the approach of Gao et al. (2025). We then apply our method to real-world data and investigate its performance on a probabilistic forecasting task.
Read the full abstract ▾
Machine Learning (Statistics)
Quadratic Weak-to-Strong Generalization in Random Feature Networks via Random Matrix Theory
Deborah Oliveira and Elliot Paquette
Weak-to-strong generalization is the phenomenon where a strong student model trained with labels produced by a weak teacher model is able to generalize better than the teacher. In this paper, we study this phenomenon in two-layer random feature networks where the model strength is determined by its width. Using tools from random
matrix theory, we derive deterministic equivalents for the population errors of an optimally trained teacher and a student trained with gradient flow. For ReLU activation and a pure spherical harmonic target, we obtain sharp asymptotics under a Gaussian universality assumption, showing a quadratic improvement: the student error scales as the square of the teacher error. These results attain the general lower bound of Medvedev at al (2025). We also analyze how the student behaves under more general stopping times and targets supported on multiple harmonic degrees, characterizing the regimes in which weak-to-strong generalization occurs and identifying the transition between quadratic, non-quadratic, and no improvement.
Read the full abstract ▾
Machine Learning (Statistics)
Careful Judge: Safe and Efficient Human-AI Collaborative Decision Making
Chenyu Zhang, Rachel Luo, Boyi Li et al. (+3 more)
In human-AI collaborative decision making, human review can prevent unsafe AI decisions, but each human judgment is costly. Treating human intervention after AI abstention as a one-off fallback misses the opportunity to improve future AI decisions for greater automation, yet AI adaptively learning from selectively queried human feedback breaks safety guardrails calibrated for old models. We approach this challenge with CARE—calibrated adaptive rectification and escalation—an end-to-end pipeline that combines AI models and human reviewers to guarantee safe, human-aligned decisions, while continuously learning from human feedback to achieve greater automation with fewer human queries. CARE is principled, general, modular, and works with any black-box AI model. Our novel adaptive calibration module guarantees risk control at every time step for any rectification module. We further show how CARE improves query efficiency when the AI model is well trained and the human-AI misalignment has a clear structure. Experiments on four safety-critical real-world
datasets spanning driving,
language, and robotics demonstrate that CARE achieves human-aligned decisions while reducing human queries by 25-81% relative to baselines.
Read the full abstract ▾
All authors Chenyu Zhang, Rachel Luo, Boyi Li, Anjali Parashar, Marco Pavone, Apoorva Sharma
Statistics Theory
Graph-monotone entrywise guarantees for MLE and Rank Centrality on general comparison graphs
Yuepeng Yang and Cong Ma
Pairwise comparisons are widely used to infer latent scores and identify top-ranked items. Although sharp statistical guarantees are available under uniform sampling, real data often induce irregular comparison graphs with heterogeneous observation counts across pairs. In this paper, we study an arbitrary fixed comparison graph under the Bradley–Terry–Luce model with minimal assumptions. We prove that both the standard maximum likelihood
estimator and Rank Centrality achieve a high-probability entrywise error rate of order $1/\sqrt{\lambda_{\mathcal{D}}}$ up to logarithmic and dynamic-range factors, where $\lambda_{\mathcal{D}}$ is the algebraic connectivity of the count-weighted observation graph. This guarantee is graph-monotone since $\lambda_{\mathcal{D}}$ cannot decrease when additional comparisons are added. Under heterogeneous sampling, where comparison pairs are sampled independently with unequal probabilities, our guarantee improves existing error bounds or requires weaker assumptions. We further extend our analysis to show that both
estimators are robust against outcome-adaptive augmentation, where an adversary can choose additional comparison pairs after observing the initial outcomes.
Read the full abstract ▾
Statistics Applications · Cross-listed from math.PR
When Lines Become Contagious: Analyzing the Spread of Infection in the M/M/1 Queue
Laurel Newman and Jamol Pender
In this paper, we construct and analyze a
stochastic model for infectious disease spread in service systems. Our model considers a $M/M/1$ queue where a proportion, $p$, of the arriving population is infectious and the remaining population is susceptible. When an infectious person overlaps in the queue with a susceptible customer for more than $\ell$ units of time, then the susceptible person becomes infected. We provide both a personalized perspective of how an individual susceptible customer is infected and a community perspective of how one infectious customer can infect the others around them. From the personalized perspective, we derive the probability that a susceptible person becomes infected and the probability they are infected by at least $k$ different people. From the community perspective we compute the mean and higher moments of the number of susceptible customers that are infected by one infectious individual. Finally, we use
stochastic simulation to verify our analytical results and show that our new formulas accurately describe the behavior our
stochastic model of infection and to expand our model into the consideration of different arrival and service regimes.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
LASER: Latent Space Adjoint Matching for Support-Constrained Entropy-Regularized Offline RL
Songyuan Zhang, Oswin So, Eric Yang Yu et al. (+3 more)
While offline reinforcement learning (RL) enables policy optimization from static
datasets without costly online interaction, it remains bottlenecked by the risk of executing out-of-distribution (OOD) actions. Recent approaches mitigate this by learning a behavior-cloning policy through flow matching and then performing RL within its constrained latent space. However, naively optimizing the latent policy can easily cause the policy to collapse into a brittle mode or exploit sharp artifacts of the learned critic. In this work, we find that
entropy regularization is essential in latent-space RL for addressing these challenges. We introduce
LASER, a novel offline RL
algorithm that applies latent-space adjoint matching to achieve
entropy-regularized latent-space RL with expressive flow policies while avoiding backpropagation through time. Through comprehensive experiments on 40 challenging OGBench tasks with varying
dataset qualities, we show that
LASER achieves state-of-the-art performance. Notably,
LASER uses fixed method-specific hyperparameters across all tasks and outperforms the evaluated baselines, including those with task- and
dataset-specific tuning, which highlights the robust applicability of
LASER. Project website: https://mit-realm.github.io/
laser/.
Read the full abstract ▾
All authors Songyuan Zhang, Oswin So, Eric Yang Yu, Matthew Cleaveland, Peter Crowley-Dolen, Chuchu Fan
Machine Learning (Statistics) · Cross-listed from cs.LG
The Best Optimizer Depends on Batch Size
Xingyu Dang, Kaiyue Wen and Sadhika Malladi
A plethora of new adaptive optimizers are designed to efficiently estimate and use minibatch gradient statistics to shape parameter updates, but they are typically benchmarked at a single batch size. Hyperparameter scaling rules promise to preserve performance as batch size and gradient noise change, suggesting that the best optimizer at one batch size should remain the best at another. We challenge this approach to developing and evaluating optimizers by showing: (1) no principled scaling rule for
Muon works consistently across training settings, and (2) the best optimizer for
language model pretraining changes with batch size even after extensive hyperparameter tuning. To understand these failures, we analyze a noisy quadratic model and show that the batch-size scaling rule that preserves performance is direction-dependent, shifting from square-root to linear as a direction's
curvature-to-noise ratio rises. We further prove that gradient noise drives a bias-variance tradeoff in preconditioning that can reverse optimizer rankings across batch sizes, even under optimal learning rate and momentum tuning. Consistent with this, when we increase the batch size $16\times$ partway through
language model pretraining, keeping small-batch updates in only the sharpest $0.001\%$ of directions removes up to $59.5\%$ of the resulting loss gap over the next $8\%$ of training, while a random subspace of the same size barely helps. Taken together, our results show that principled optimizer design must account for batch size and empirical benchmarking should be performed across batch sizes.
Read the full abstract ▾
Statistical Methodology
Projected Normal Models on Recursively Curved Tori
Surojit Biswas and Aron Persson
We develop a computationally tractable statistical framework for angular data supported on a recursively embedded curved
torus $\mathbb{T}^n\hookrightarrow\mathbb{R}^{n+1}$. Due to the recursive structure we build an exact uniform sampler by generalizing the process from the curved 2-
torus. We generalize projected normal distributions to the curved $n$-
torus and we develop inference theory for these distributions. We utilize both maximum-log-likelihood estimates and energy-based Fréchet estimates for the projected normal distribution for this purpose. This is done by establishing identifiability for isotropic projected normal distributions and by applying theory of maximum likelihood and intrinsic statistics for the curved
tori. We numerically verify the geometry, sampler, density transformation, and energy solver. Lastly, we demonstrate the applicability of the proposed model using real data.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Work While They Sleep: Exploiting Evaluation Latency for Fully Bayesian Optimization
Gustavo Sutter, Alejandro Comas-Leon, David Holzmüller et al. (+4 more)
Black-box optimization problems are ubiquitous across science and engineering, often dealing with expensive objective functions. This objective latency has two consequences during optimization: (i) the objective evaluation dominates execution time, and (ii) sample-efficient
algorithms are crucial to accelerate development and avoid wasting resources.
Bayesian optimization (BO) methods are the de facto choice of planners for suggesting the next point to try. Standard BO fits the surrogate model's hyperparameters with a point estimate. Alternatively, a fully
Bayesian approach uses model averaging to account for uncertainty over the hyperparameters, leading to better uncertainty estimates—useful in the low-data regime that is pervasive in BO. However, it is often prohibitively expensive and thus rarely used. In this work, we propose ELF-BO, an
algorithm that uses the objective evaluation latency to headstart the computation of the next suggestion, allowing for fully
Bayesian optimization without incurring substantial decision-time costs. This is done by sampling from the hyperparameter posterior while the objective is being evaluated, only requiring reweighting of the samples once the objective value is observed. Across synthetic functions and real-world applications, we show that ELF-BO matches the performance of fully
Bayesian methods while only incurring decision latency on par with or better than standard BO. Thus, ELF-BO makes fully
Bayesian optimization practical in real-world use cases.
Read the full abstract ▾
All authors Gustavo Sutter, Alejandro Comas-Leon, David Holzmüller, Hao Wang, Luis Ricardez-Sandoval, Pascal Poupart, Agustinus Kristiadi
Machine Learning (Statistics)
Trust-Region Optimization for Smooth Potential-Interaction Energies in Wasserstein Space
You Wan, Ting Gao and Jinqiao Duan
Finding low-energy configurations of interacting particles and approximating probability distributions lead to the minimization of potential-interaction energies in Wasserstein space. These energies can be nonconvex, making it important to exploit second-order information while controlling the reliability of local approximations. We study trust-region optimization of smooth potential-interaction energies on the Wasserstein space of probability measures with finite second moment. The method uses a quadratic model along pushforward curves, an $L^2(\rho)$ step radius, and a Steihaug-Toint subsolver with an explicit self-adjoint second-variation operator. A ratio test determines acceptance and guides the radius update. Under a lower energy bound and globally bounded Hessians of the potential and interaction kernel, we prove that the objective is nonincreasing, the Wasserstein-gradient norms converge to zero, and an $\varepsilon$-stationary iterate is reached within $O(\varepsilon^{-2})$ total outer trials, including rejected trials. If the potential is quadratically coercive, every weak accumulation point is stationary. The analysis applies to arbitrary initial measures with finite second moment. For empirical measures, the iteration is a finite-dimensional trust-region method in the $L^2(\rho_N)$ inner product, with complexity constants independent of particle number and dimension when the initial objective gaps are uniformly bounded. Numerical experiments include a smooth soft-particle energy, maximum-mean-discrepancy minimization for non-Gaussian targets, component ablations, and scaling studies in particle number and dimension.
Read the full abstract ▾
Statistics Theory · Cross-listed from cs.IT
Slow Beats Fast at the Kesten-Stigum Threshold: Minimax, Fisher-Information and Belief-Propagation Characterizations of the Information-Computation Gap in Sparse Stochastic Block Models
Soroor Ghandali
We study community recovery in the sparse symmetric
stochastic block model with $q$ communities, average degree $d$ and signal strength $\lambda$ through statistical decision theory and Fisher information, and obtain three characterizations of the Kesten-Stigum threshold $d\lambda^2=1$ and of the information-computation gap below it. First, on each community-size profile the minimax risk of any class of rules closed under averaging and vertex relabeling equals its Bayes risk under the uniform prior; the posterior mean is the unique Bayes rule and is admissible, and the Bayes risk of degree-$D$
polynomial rules is the trivial risk times $1-\mathrm{Corr}_D^2$. Combined with known low-degree and information-theoretic results, this gives the gap as a worst-case statement: for $q\ge 5$ there is a window below the threshold in which no low-degree rule beats the trivial risk asymptotically, while an exponential-time rule does on a set of labelings of probability $1-o(1)$. Second, the Fisher information about $\lambda$ carried by cycle counts is a series with terms of order $k(d\lambda^2)^k$, convergent exactly when $d\lambda^2<1$; below the threshold the relative error of every unbiased cycle-based
estimator of $\lambda^k$ stays above an explicit constant, and every cycle-count test has success probability bounded below one. Third, the derivative of belief propagation at its uninformative fixed point multiplies a random perturbation by $|\lambda|\sqrt{d}$ per iteration, and one EM step taken there leaves $\lambda$ unchanged. A signal-to-noise computation recovers the condition $d\lambda^{1/\chi}>1$ of Chin et al. for $q=n^\chi$ communities and identifies personalized PageRank as a walk count with suboptimal weights. Experiments on networks with up to $3\times 10^5$ vertices confirm the threshold for $q=2$, the hard window for $q=5$, and the many-community scaling.
Read the full abstract ▾
Statistics Applications · Cross-listed from q-fin.RM
Learned Monotone Recurrent Features in Governed Credit Scoring: The Price of the Frame and the Necessity of Macro Conditioning
Yew Lee Tan
Regulated credit scoring requires scores monotone non-decreasing in every exposure input. Deployed pipelines – hand-crafted monotone aggregates feeding sign-constrained gradient boosting – already meet this by composition; the open question is what learned temporal aggregation is worth inside one. We answer on five production-scale credit
datasets at matched admissibility (one priced baseline convention excepted), with a monotone recurrent architecture whose per-input guarantee we extend, with proofs, to vector-valued inputs and to exogenously macro-conditioned decay gates, severities, thresholds, and peak
memory. Two findings result. First, a strictness ladder: the value of learned monotone features rises with governance-frame strictness – zero on unconstrained engineered panels, maximal in summaries-only frames – replicated across two
datasets and an official temporal-stability metric, though unconditioned features degrade on externally adjudicated later weeks. Second, a conditioning-delivery asymmetry under regime shift. On a train-on-boom, test-on-crisis mortgage design, two public macroeconomic series hurt as input columns, yet conditioning the recurrence on them delivers the paper's only learned-block crisis-cohort uplifts. The confirmed effect: +0.006 to +0.013 AUC on an internally pre-registered Freddie Mac replication, at all five held-out seeds. The discovery estimate: +0.015 to +0.021 on Fannie Mae (three of five seeds post hoc), worth 10-27 basis points of defaulted balance at an 80% approval cutoff, and grows with early-prepaid loans excluded. A state-level test identifies the mechanism: between-cohort calibration transfer. A pandemic-band episode bounds scope: under forbearance-distorted labels the gain generalizes at a quarter to a third of crisis size on Fannie Mae, on Freddie Mac only against the capacity control.
Read the full abstract ▾
Statistics Theory · Cross-listed from math.PR
Estimation and Recovery of a Planted Dense Subgraph from a Single Network Cascade
Maximilien Dreveton, Paula Mürmann and Patrick Thiran
We study the inference of a planted dense component in a sparse random graph from a single spreading process. The graph has an Erdős-Rényi background with edge probability $p_n$ and contains a planted dense component of size $n^{\alpha}$, with $\alpha > 1/2$, whose internal edge density $\xi>0$ is constant. The edge set is unobserved; the data only consist of the successive infection times from a single realization of a continuous-time SI process with independent and exponentially distributed transmission times. We show that the planted dense component leaves a detectable signature in the spreading process: after the exploration enters the dense component, it undergoes a short phase of accelerated growth. By analyzing this phase, we localize its onset and endpoint. These localization results yield consistent
estimators of the component-size exponent $\alpha$, the background edge density $p_n$, and the internal edge density $\xi$ from the infection times alone. When the identities of the infected vertices are also observed, we further establish consistent recovery of the planted dense component.
Read the full abstract ▾
Statistical Methodology
Prediction-powered inference for time series across space
Shahzar Rizvi, David Burt, Vishwak Srinivasan et al. (+3 more)
The following motif is common in spatiotemporal settings: we have a sequence of
covariate and label pairs observed for a relatively short, recent time period. We have access to unlabeled
covariates over a longer time period. Data is observed over many spatial locations. For instance, crop yield might be observed over a large geographical area for recent years, but weather data (which is informative about crop yield) is available for a much longer period. The goal is to estimate, at each spatial location, the expected label (e.g., crop yield) in the future and provide a valid confidence interval for this value. The observed time period alone is too short for reliable estimates. Imputing missing labels with machine learning can cause substantial bias. Prediction-powered inference (PPI) can correct for this bias, but it relies on an i.i.d. assumption that breaks under our expected temporal dependencies. Heteroskedasticity and autocorrelation consistent (HAC) procedures account for temporal correlation, but have not been adapted to cases where some labels are imputed. We provide reliable point estimates and confidence intervals given: short labeled time series (across spatial locations), a longer unlabeled time series, and an imperfect predictor of labels given
covariates. We show our method outperforms natural alternatives.
Read the full abstract ▾
All authors Shahzar Rizvi, David Burt, Vishwak Srinivasan, Renato Berlinghieri, Stefano Del Col, Tamara Broderick
Statistics Theory
Asymptotic Null Distributions of Moran's $I$ and Assortativity in Large Networks
Karin Ait Braham, Louis-Paul Rivest and Thierry Duchesne
This study investigates the asymptotic behavior of two dependence measures defined on networks, Moran's $I$ statistic and Newman's assortativity, under the null hypothesis that a Gaussian node attribute $Y$ is independent of the network structure. We demonstrate that the structure of the network directly affects the convergence rate to normality of these measures as the size of the network increases. We further establish that, in some instances, the mean values of these dependence measures under the null hypothesis remain non-negligible asymptotically and must therefore be explicitly accounted for when calculating the test statistics. Applications to a variety of simulated and real networks also reveal that the normal approximation performs well only when the network is not strongly heterogeneous. Network
topology determines both the convergence rate to normality and whether the limiting distribution is Gaussian. In dense networks whose degree heterogeneity does not vanish, we further show that assortativity can fail to be a valid test statistic even though Moran's $I$ remains well behaved, whereas a dominating node invalidates both.
Read the full abstract ▾
Machine Learning (Statistics)
Steering Diffusion Models to Rare Events with Sequential Monte Carlo
Aavash Subedi, Tim Reichelt, Christopher Williams et al. (+3 more)
Diffusion models are increasingly used as surrogates for expensive simulators in weather prediction, molecular dynamics, and materials design. In these models, computing the probability $p_0[E]$ of an event $E$ is difficult, especially when the event of interest is rare. A stable estimate using Monte Carlo becomes computationally intractable, requiring a growing sample size $\propto\!1/p_0[E]$ to compensate for an increasing rarity. In this paper, we present
Diffusion rare event sampling with Sequential Monte Carlo (DireSMC), a sequential Monte Carlo scheme that guides a population of weighted samples towards the rare event, giving access not only to samples but also to a calibrated estimate of its probability. We set up our guidance using an analytical relaxation of the event set, allowing the method to easily extend to a wide range of user-defined rare events. We validate our method on a toy problem with analytical solutions and on a score-based
climate emulator, where we obtain accurate rare-event probabilities on a range of rarities from $10^{-3}$ to $10^{-5}$, achieving net speed-ups of $9\times$ to $1413\times$ over Monte Carlo.
Read the full abstract ▾
All authors Aavash Subedi, Tim Reichelt, Christopher Williams, Philip Stier, Yee Whye Teh, Saifuddin Syed
Statistics Theory
Spectral Recovery of Point Clouds from Noisy Geometric Graphs
Tatiana Brailovskaya, Nicholas A. Cook and Sofia Poinelli
We study the problem of recovering low-dimensional latent geometry from a random
geometric graph generated by noisy, high-dimensional data. Specifically, we analyze the performance of a spectral
embedding algorithm on the Signal+Noise Graph Model, in which vertices are associated to points perturbed by Gaussian noise, and edges are included for pairs whose inner product exceeds a specified
alignment threshold. In the high-dimensional regime where the number $n$ of points and the ambient dimension $d$ both tend to infinity, we show that under a spectral gap condition, the top eigenvectors and
eigenvalues of the graph's adjacency
matrix can be used to approximately recover the point cloud up to an orthogonal transformation. We illustrate our results on point clouds sampled from nested spheres and high-dimensional sinusoid curves.
Read the full abstract ▾
Machine Learning (Statistics)
Feature Information Dynamics in Diffusion
Jia-Shu Pan, Tao Zhang, Yufei Huang et al. (+2 more)
Diffusion models generate data through a continuum of denoising problems, and are widely observed to reveal coarse structure before fine detail. Yet, this intuition is mostly empirical and qualitative. We introduce feature information dynamics, an information-theoretic framework for localizing when a feature is generated during
diffusion. Using the I-MMSE identity, we connect the rate of feature mutual information change to a gap between optimal unconditional and feature-conditional denoising losses, yielding practical
estimators for feature information density. We further develop a chained decomposition that separates shared from incremental information in a feature hierarchy. We use this framework first to quantitatively confirm spectral autoregression in pixel
diffusion, and then to extend the analysis beyond frequency: under a class $\to$ mask $\to$ Canny conditioning chain, the per-feature information densities differ across pixel, SDVAE, VAVAE, and RAE, exposing fundamental differences between these representations and suggesting that ordered generation could be beneficial for training
diffusion models. Our code is available at https://github.com/AI4Science-WestlakeU/feature-information-dynamics.
Read the full abstract ▾
All authors Jia-Shu Pan, Tao Zhang, Yufei Huang, Yanjun Sheng, Tailin Wu
Machine Learning (Statistics) · Cross-listed from cs.LG
Early Memory Selection for Balanced Adam
Alberto Fernández-Hernández, Cristian Pérez-Corral, Jose I. Mestre et al. (+2 more)
We propose a method for choosing the shared
memory parameter $\beta_1=\beta_2=\beta$ in Adam from a short pilot training. The selected $\beta$ remains fixed during the subsequent full training. A local model of Adam's normalized direction balances sampling variability against the delay introduced by averaging past gradients. This balance gives a cubic
memory rule, whose two coefficients are estimated from gradient probes at a few pilot checkpoints. The
estimator uses the numerator and denominator jointly, preserving their covariance. With a 200-update pilot and sixteen probe gradients at each of four checkpoints, a seed-matched retrospective evaluation on eleven
vision and
language workloads reduces mean relative validation gap by 40.7% and worst-quarter mean gap by 44.3% against the grid representative of shared $\beta=0.95$. The mean gap is also 32.3% lower than that of the best constant $\beta$ chosen across all eleven workloads.
Read the full abstract ▾
All authors Alberto Fernández-Hernández, Cristian Pérez-Corral, Jose I. Mestre, Manuel F. Dolz, Enrique S. Quintana-Ortí
Machine Learning (Statistics) · Cross-listed from cond-mat.dis-nn
Classifications in modular restricted Boltzmann machines
Elena Agliari, Andrea Lepre and Edoardo Roscani
We consider a modular associative neural network made of $L$ Hopfield models (HMs), coupled so that intra-module interactions are Hebbian and inter-module interactions are anti-Hebbian; this competitive coupling is known to endow the network with pattern-disentanglement capabilities. The integral representation of this system coincides with an assembly of $L$ restricted Boltzmann machines (RBMs) whose hidden layers are coupled, thereby extending the HM-RBM duality to the modular setting. We then train this modular RBM, via one-step contrastive divergence, to perform a classification task in which a query encoded on the visible layers is mapped onto an $L$-tuple of labels read off the hidden layers. When the query is composed of $L$ patterns that are mutually orthogonal on average, we prove that the RBM weights obtained as empirical means over the training
dataset, as suggested by the HM-RBM equivalence, constitute a fixed point of the learning dynamics, and we derive an explicit, non-asymptotic bound on the residual drift. We then turn to a harder classification task in which each module is queried with a mixture of $L$ patterns and we show numerically that the same setting for the RBM weights still provides an effective set-up, letting the network jointly classify and disentangle the mixture.
Read the full abstract ▾
Statistical Methodology
When does conformal calibration need censoring weights? Cause-of-failure prediction sets under competing risks
Sunny Yang and Weiyan Zhao
Split conformal prediction sets for competing-risks labels at a fixed horizon require calibration labels that right censoring can leave unobserved. Complete-case calibration guarantees coverage for the label-complete subpopulation, but its population coverage can deviate in either direction, even at the true class probabilities. We study how selection changes the score distribution near the population
quantile. At the true score in our main simulation family, with independent draws, complete-case calibration covers 0.8723 at a nominal 0.900 when 22% of subjects are event-free at the horizon. In 4 of 48 further designs using per-draw normalisation, estimating cause incidences as one minus the exponential of the negative of the cause-specific Nelson-Aalen cumulative hazards, with the event-free probability as the clipped and renormalised remainder, puts complete-case coverage at least four standard errors below nominal. In these designs, weights from a correctly specified censoring model keep coverage near nominal. Extending the argument of Yi et al. (2025) to a cause label, we establish a finite-sample coverage lower bound with an explicit penalty for censoring-model error. Misspecifying the censoring model can also lead to under-coverage.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Reinforcement Learning for Hierarchical Reasoning Rewards: Minimax-Optimal Rates with Transformers
Naoki Nishikawa and Taiji Suzuki
Reinforcement learning (RL) has become a standard tool for post-training
language models on
reasoning tasks, where the policy is updated by reward feedback while exploring the space of responses. Despite its empirical success, theoretical understanding of RL post-training remains limited, in particular of why on-policy exploration combined with a neural reward model is effective. In this paper, we address this question by modeling the reward as a hierarchical function on the response space: the reward consists of infinitely many local components, each of which becomes relevant only after the preceding ones have been resolved. We show that a natural
Transformer-based actor–critic
algorithm, which alternates between sampling from the current KL-regularized policy, fitting a
Transformer critic to the observed rewards, and updating the policy, achieves the minimax optimal rates in the query budget and in the regularization strength up to logarithmic factors, and is minimax optimal for a fixed number of
prompts. In contrast, we prove that sampling from the fixed reference distribution, as in offline reward modeling, can limit regret decay to a logarithmic rate. These results show that on-policy exploration progressively zooms in on the region where the reward is concentrated, and quantify its benefit for RL post-training.
Read the full abstract ▾
Statistical Methodology · Cross-listed from econ.EM
Measuring Gift Card Program Incrementality via Causal Data Fusion
Justin Whitehouse, William Betz, Yan Zhang et al. (+3 more)
Businesses regularly offer gift card programs to drive customer spending and increase engagement. A central question is how much incremental revenue these programs generate, and which channels drive it most efficiently. Measuring the incremental revenue associated with a gift card program is a challenging problem in causal inference, requiring a firm to infer how much each customer would have spent if they never received a gift card. Observational data on past customer purchasing behavior reveal possession of a gift card only when a customer makes a purchase, thus leaving a customer's treatment status systematically censored. In this paper, we develop a novel data fusion approach to overcome this missing data challenge. We identify and estimate incrementality by combining a large observational
dataset with a smaller experimental
dataset from a different population. Our approach relies on a mild transferability condition, which posits that the conditional relative treatment effect of gift card receipt on the decision to purchase is invariant across the two populations. We develop a flexible, machine learning-based
estimator for the incremental revenue and establish its asymptotic normality. We apply our
estimator across both first- and third-party channels through which Airbnb distributes gift cards, finding heterogeneity in incrementality across segments of the population. In particular, we find not only that third-party channels are more incremental than first-party ones, but also that "self-gifters" (i.e., customers likely to have purchased their own gift cards) are more incremental than the broader population.
Read the full abstract ▾
All authors Justin Whitehouse, William Betz, Yan Zhang, Peter Coles, Ramesh Johari, Vasilis Syrgkanis
Statistics Theory · Cross-listed from math.PR
Sharp detection thresholds for random geometric graphs with polynomial average degrees
Hang Du, Cheng Mao, Nike Sun et al. (+2 more)
We study the random
geometric graph formed by connecting $n$ independent uniform points on $\mathbb S^{d-1}$ whenever their inner product exceeds a threshold chosen to give edge density $p$. We establish the conjectured dimensionality threshold for indistinguishability from the Erdös-Rényi graph $G(n,p)$ throughout the sparse regime with polynomially growing average degree. Specifically, for every fixed $\delta>0$, if $p\to0$, $p\ge n^{-1+\delta}$, and $d\gg (np\log(1/p))^3$, then the Kullback-Leibler (KL) divergence between the two distributions vanishes. Our proof builds on the posterior analysis framework introduced in previous work of the authors, combining a truncated second moment analysis with
Fourier and spherical harmonic expansions. As in prior works, we reduce the KL bound to controlling a truncated $\chi^2$-divergence $\Delta$ for the neighborhood of the last vertex conditional on the graph induced by the preceding vertices. We then bound $\Delta$ using a
Fourier expansion of the conditional likelihood ratio for this neighborhood, where the high-frequency contribution is made negligible by truncation to a typical-degree event, while a spherical harmonic expansion approximates the low-frequency contribution in terms of posterior moments of bounded-degree harmonic
polynomials. The key observation is that these moments and $\Delta$ satisfy a coupled system of cavity recurrences, yielding a self-bounding inequality that implies the desired KL bound.
Read the full abstract ▾
All authors Hang Du, Cheng Mao, Nike Sun, Yihong Wu, Jiaming Xu
Machine Learning (Statistics)
Information-Dense Synthesis for Molecular Discovery
Kasper K. Jakobsen and Eli N. Weinstein
Machine learning can accelerate molecular discovery by designing molecules and planning experiments. However, many scientific challenges demand molecules with very rare properties, and in this sparse setting, existing
algorithms offer little gain over random guessing. We propose a method to efficiently search large regions of molecular space using algorithmically controlled
stochastic synthesis. Rather than design, make and test individual molecules, we design and make complex mixtures, test them as a pool, then deconvolute the molecule-activity map. We optimize synthesis to encode maximal information. Theoretically, this approach can reduce the number of experiments required to find the optimal molecule among $d$ candidates from $\mathcal{O}(d)$ to $\mathcal{O}(\log d)$ or $\mathcal{O}(1)$. In simulation, on estimated
protein fitness landscapes, it finds active molecules with an order of magnitude fewer experiments than existing
Bayesian optimization methods.
Read the full abstract ▾
Statistical Methodology
Fridge: focused fine-tuning of ridge regression for personalize predictions
Kristoffer Herland Hellton and Nils Lid Hjort
Statistical prediction methods typically require some form of fine-tuning of tuning parameter(s), with $K$-fold cross-validation as the canonical procedure. For ridge regression there exist numerous procedures, but common for all, including cross-validation, is that one single parameter is chosen for all future predictions. We propose instead to calculate a unique tuning parameter for each individual for which we wish to predict an outcome. This generates an individualized prediction by focusing on the vector of
covariates of a specific individual. The focused ridge – fridge – procedure is introduced with a two-part contribution: 1) first we define an oracle tuning parameter minimizing the mean squared prediction error of a specific
covariate vector, 2) then we propose to estimate this tuning parameter by using plug-in estimates of the regression coefficients and error variance parameter. The procedure is extended to logistic ridge regression by utilizing parametric bootstrap. For high-dimensional data, we propose to use ridge regression with cross-validation as the plug-in estimate, and simulations show that fridge gives smaller average prediction error than ridge with cross-validation for both simulated and real data. We illustrate the new concept for both linear and logistic regression models in two applications of personalized medicine: predicting individual risk and treatment response based on gene expression data. The method is implemented in the R package "fridge".
Read the full abstract ▾
Machine Learning (Statistics)
One-Shot Private Confidence Regions via Resampling
Shourya Pandey, Purnamrita Sarkar, Po-Ling Loh et al. (+1 more)
We propose a simple framework for constructing differentially private confidence regions in one shot, i.e., by adding noise only to the final resampling
quantile instead of privatizing the
estimator computed on each resample. The cost of
privacy of our procedure is only logarithmic in the number of resamples $B$ under with-replacement ($m$-out-of-$n$) sampling and independent of $B$ under without replacement sampling (subsampling), avoiding the $\sqrt{B}$ factor that arises in previous works. We provide nonasymptotic Gaussian Differential
Privacy (GDP) and utility guarantees for both subsampling and $m$-out-of-$n$ resampling, covering mean-like
estimators with small global sensitivity as well as
estimators admitting efficiently computable smooth sensitivity bounds, including
quantiles and degenerate U-statistics. This allows us to also obtain private confidence regions for degenerate U-statistics where the private error is much smaller than the non-private error. In all, we provide a toolbox for widely applicable DP uncertainty quantification procedures under popular resampling strategies while avoiding the computational and
privacy costs of privatizing many intermediate resample statistics.
Read the full abstract ▾
All authors Shourya Pandey, Purnamrita Sarkar, Po-Ling Loh, Debepsita Mukherjee
Statistical Computation
Scalable Regularized Vector Multiplicative Error Models for Positive-valued Financial Time Series
Rohan Hemant Chhatre, Chiranjit Dutta, Nalini Ravishanker et al. (+1 more)
The logarithmic multiplicative error model (log-vMEM) has been useful in modeling and forecasting multivariate positive-valued financial time series. The number of parameters grow rapidly with the dimension of the system and the lag order, making estimation computationally demanding in high-dimensional settings. This paper describes regularized estimation via hierarchical lag structures (Nicholson et al., 2020) for log-vMEM models with multivariate gamma error distribution of Tsionas (2004). The parameter estimation is performed using a blockwise coordinate descent
algorithm with a Gauss-Seidel-style update scheme (Wright, 2015). This enables an efficient computation strategy compared to traditional penalized maximum likelihood approaches. The competing models are juxtaposed against each other by combining three hierarchical lag structures (componentwise, elementwise, own-other) and four penalties(group-lasso, adaptive group-lasso, group-mcp, and group-scad). Extensive simulation runs have been performed to test the parameter recovery for both the unpenalized and the penalized models. We apply the proposed methods to model the joint dynamics of robust intraday realized
volatility measures for Microsoft (NASDAQ: MSFT) for the competing models. The numerical integration step of the log-likelihood is identified to be the principal computational bottleneck. We address this issue by using GPU-accelerated quadrature integration thus improving computational scalability of the proposed models.
Read the full abstract ▾
All authors Rohan Hemant Chhatre, Chiranjit Dutta, Nalini Ravishanker, Sumanta Basu
Machine Learning (Statistics) · Cross-listed from cs.LG
Symmetry-Aware Feature Learning: A Polynomial Separation for Multi-Index Models
Jivan Waber, Vanessa Piccolo, Yatin Dandi et al. (+1 more)
We establish a
polynomial sample complexity separation between
symmetry-aware and
symmetry-agnostic feature learning. We study growing-rank multi-index models with high-dimensional Gaussian
covariates in $\mathbb{R}^d$ and $r=\Theta(d^\delta)$ teacher directions forming a cyclic
symmetry orbit, where $0<\delta<1/2$. We compare three ways of exploiting this structure: architectural weight sharing, data augmentation over the full
symmetry group, and learning without access to the
symmetry. In particular, we analyze a
symmetry-tied convolutional network, an untied network, and the same untied network trained with full-group data augmentation, using spherical online SGD with correlation loss. For a class of
polynomial links with information exponent $p\ge3$, we prove matching sample complexity bounds up to logarithmic factors: the tied and augmented learners achieve weak directional recovery in $\widetilde{\Theta}(d^{p-1})$ samples, whereas the
symmetry-agnostic learner requires $\widetilde{\Theta}(rd^{p-1})$. For the pure quadratic Hermite link, the same separation holds for weak recovery of the teacher subspace, with sample complexities $\widetilde{\Theta}(d)$ and $\widetilde{\Theta}(rd)$, respectively. Thus, full-group data augmentation matches the sample efficiency of architectural weight sharing, and both provide a
polynomial advantage over training without
symmetry. For $p\ge3$, the proof reveals a two-stage mechanism: fluctuations at initialization select one direction in the teacher orbit, after which localized growth amplifies its overlap to the weak recovery scale while competing overlaps remain near their initialization scale.
Read the full abstract ▾
All authors Jivan Waber, Vanessa Piccolo, Yatin Dandi, Florent Krzakala