Your paper timeline
Scroll AI takes the way you would scroll a great paper aggregator: quick signal first, deeper critique when something earns your attention, and challenges when a claim feels off.
109 papers in stat.ML
Trending mixes fresh papers with community signal.
0
cs.LGcs.AImath.OC Kaifei Wang, Zikang Shan, Liwei Wang et al. · Oct 14, 2026

Momentum is a pillar of modern optimization, yet its benefits in reinforcement learning with verifiable rewards (RLVR) remain unclear. This paper studies a KL-regularized bandit formulation of RLVR and proves that, with exact policy gradients, larger EMA coefficients strictly accelerate convergence within a safe overshoot-free range. Its central insight is that stochastic sampling obscures this gain unless the coefficient scales with the learning rate so that $\eta/(1-\beta)$ stays constant, a principle validated in bandits and in LLM post-training on Qwen models.

Momentum is a classical tool for accelerating convex optimization and a standard component of modern large language model training. Yet recent empirical work has questioned the usefulness of first-moment momentum in reinforcement learning with verifiable rewards (RLVR). We study this question through a bandit formulation of RLVR with bias-corrected exponential moving average (EMA) momentum, where the EMA coefficient $\beta$ controls how quickly the weights on past gradients decay. When policy gradients are evaluated exactly, we prove that increasing $\beta$ within a range that prevents overshoot strictly accelerates optimization. When gradients are estimated from sampled responses, a positive expected gain can be obscured by sampling fluctuations. We show that as the learning rate $\eta$ decreases, the expected gain with fixed $\beta$ vanishes faster than these fluctuations, whereas increasing $\beta$ to keep $\eta/(1-\beta)$ within the safe regime preserves a nonvanishing gain that eventually dominates them. Bandit simulations quantitatively validate these predictions. RLVR experiments on Qwen3-4B-Base using SGD with EMA and AdamW support our principle of choosing larger coefficients in practical LLM post-training. On Qwen2.5-32B with AdamW, simply increasing $\beta_1$ improves mean validation accuracy by nearly $2$ percentage points over the conventional choice, demonstrating the principle's effectiveness at a larger model scale.
0
cs.LGcs.AIcs.CL Aryan Das, Arnab Bhattacharya, Vinay Kumar Verma · Oct 14, 2026

Reinforcement learning with verifiable rewards suffers from coarse scalar supervision and moving-target instability in self-distillation. FSPO eliminates the persistent teacher model by storing sparse historical next-token distributions as fixed function-space targets, combining them with privileged disagreement correction and reliability-adaptive grounding. Across scientific reasoning and tool-use benchmarks, it achieves roughly 4% higher average accuracy than SDPO under a 5-hour budget while cutting peak memory from about 70 GB to 56 GB on Olmo3-7B-Instruct.

Reinforcement Learning with Verifiable Rewards (RLVR) improves reasoning in language models, but scalar outcomes provide coarse supervision over response trajectories. Self-distillation offers denser token-level guidance by comparing an unassisted policy with a feedback-informed self-teacher, yet the teacher evolves with the student, creating unstable targets and motivating additional regularization. We introduce Function-Space Policy Optimization (FSPO), a single-policy framework that combines current privileged adaptation with function-space temporal consolidation. FSPO stores sparse historical next-token distributions as fixed temporal targets, providing temporal separation without maintaining a persistent teacher model. It further introduces privileged disagreement correction to explicitly suppress locally overvalued actions, reliability-adaptive grounding that strengthens reference supervision when self-generated evidence repeatedly fails, and quality and coverage-aware temporal memory that retains verified behaviors and improves privileged demonstrations. Across reasoning and tool-use benchmarks, FSPO achieves the highest average accuracy under a 5-hour training budget, improving over the strongest baseline by approximately 4% points on average while reducing peak memory from about 70 GB to 56 GB on Olmo3-7B-Instruct.
0
cs.LGstat.ML Yuxuan Xia, Xiao-Tong Yuan · Oct 14, 2026

This paper claims the first global joint lower bound for zeroth-order nonconvex optimization under unrestricted spatial queries and sample reuse. The core idea is to force any algorithm to advance through $T\asymp\min\{\Delta H,L_0^2\}/\epsilon^2$ hidden geometric stages and to prove that each fresh stage requires $\Omega(dL_0^2/\epsilon^2)$ adaptive scalar evaluations, yielding an overall $\Omega(dL_0^2\min\{\Delta H,L_0^2\}/\epsilon^4)$ bound. If correct, this would settle the open problem posed by Kornowski and Shamir (2024), though the manuscript is dated 14 October 2026, cites several future works, and explicitly states that its mathematics was primarily AI-generated.

Recent work has sought to transfer first-order nonconvex optimization guarantees to zeroth-order methods through gradient estimators, typically incurring a factor linear in $d$. \citet{kornowski2024} obtained an $\mathcal{O}(d\delta^{-1}\epsilon^{-3})$ oracle complexity for nonsmooth nonconvex optimization and asked whether its dependence on all three parameters is jointly optimal. Although the dimension dependence and the first-order dependence on $(\delta,\epsilon)$ are separately known to be unavoidable, a matching zeroth-order lower bound has remained open. We answer this question in the unrestricted full-space setting. For smooth stochastic nonconvex optimization, we establish a scalar-oracle lower bound of $\Omega\!\left(dL_0^2\min\{\Delta H,L_0^2\}/\epsilon^4\right)$ assuming a smooth, globally lower-bounded population objective and a second-moment bound on sample Lipschitz constants. A smooth reduction then yields a lower bound for Goldstein stationarity, proving the joint minimax optimality of the dependence on $(d,\delta,\epsilon)$ under explicit high-dimensional and small-accuracy conditions. Crucially, our results allow algorithms to query arbitrary points in $\mathbb{R}^d$, adaptively select sample-function labels, reuse the same sampled function at any number of locations, and output points without spatial restrictions. Thus, unlike lower bounds based on bounded query regions or fresh evaluation noise, our guarantees apply to a fully adaptive, reusable-sample scalar oracle. We also establish a separate noiseless exact-value lower bound against randomized algorithms, and hence also against deterministic methods. Our proof combines a globally defined sequential nonconvex construction with an exact Gaussian scalar channel, multistage random-field estimates, and an antithetic finite-sample realization that preserves information hardness under adaptive reuse.
0
stat.MLcs.LG Keisuke Hanada, Futoshi Futami · Oct 14, 2026

In medical and crowdsourced annotation, the absence of clean ground truth biases standard expected calibration error (ECE) evaluation because annotator errors confound predictor miscalibration. This paper proposes Noise Corrected ECE (NC-ECE), which inverts annotator-specific class-conditional noise channels to recover clean-label ECE, and provides methods to estimate those channels from repeated annotations alone. It then derives post-hoc recalibration algorithms with either clean-risk or calibration guarantees. The work is directly motivated by settings such as medical imaging, where expert disagreement is common and reliable uncertainty assessment is essential.

In medical annotation and crowdsourcing, multiple annotators assign labels to each example, but these labels may be incorrect. Treating such noisy labels as ground truth can bias calibration evaluation. To address this problem, we propose methods for evaluating and recalibrating a fixed binary predictor under the underlying clean-label distribution, using only noisy annotations. First, given annotator-specific noise probabilities, we propose Noise Corrected ECE (NC-ECE), a binned estimator of expected calibration error (ECE) under class-conditional noise. We derive non-asymptotic bounds on the estimator's bias. Then, we propose practical algorithms for estimating unknown noise probabilities under conditional independence and label-orientation assumptions. We also develop loss-based and histogram post-hoc recalibration with clean-risk and calibration guarantees, respectively. Experiments on semi-synthetic benchmarks show that the proposed ECE estimator closely tracks its clean-label oracle counterpart and that the recalibration methods substantially improve latent-label calibration, whereas uncorrected recalibration against noisy annotations can degrade it.
0
stat.MLcs.LG Kun Ryu, Yeo Jin Jung, Shyue Ping Ong et al. · Oct 14, 2026

Density functional theory (DFT) is accurate but expensive, while pretrained machine-learned interatomic potentials (MLIPs) are fast yet erratically wrong across configurations. This paper frames the decision of when to trust an MLIP and when to fall back to DFT as a risk-controlled selection problem solved with conformal risk control (CRC). A lightweight, chemistry-informed descriptor feeds a small neural network that estimates MLIP-DFT disagreement in energy and forces; CRC then calibrates a threshold so the expected retained error stays below a user-chosen fraction $\alpha$ of the fitting-set mean. The method cuts DFT costs—by 7–14% at $\alpha=0.3$ versus committee baselines—without requiring access to the target MLIP's internals, which matters for practitioners who want certified savings on a fixed, pretrained model.

Automated materials discovery requires evaluating large pools of candidate configurations, with density functional theory (DFT) as a major throughput bottleneck. Pretrained machine-learned interatomic potentials (MLIPs) approximate DFT cheaply, but their errors vary by orders of magnitude across configurations, making it difficult to decide when to trust the MLIP predictions and when to fall back to DFT. We frame this decision as a risk-controlled selection problem and solve it within a conformal risk control (CRC) framework. Each configuration is mapped from atomic numbers and Cartesian coordinates to a fixed geometric descriptor, which a lightweight predictor uses to estimate an MLIP-DFT disagreement score in energy, average force, and maximum force. CRC then calibrates this score into a selection threshold for DFT labeling such that the MLIP error on the unselected configurations, which keep their MLIP labels, stays below a user-chosen fraction $\alpha$ of the expected error. Under an identical guarantee across eight dataset-MLIP pairs, our predictor routes 7% and 14% fewer configurations to DFT at $\alpha=0.3$ than committee baselines. Because committee disagreement overlooks errors the MLIPs share, our gains are largest when the DFT labels and MLIP training data differ in DFT functional and basis set, and narrow when the two agree. Without model access, the structure-only predictor also ranks errors as well as the target MLIP's latent representation and outperforms widely used descriptors such as Smooth Overlap of Atomic Positions (SOAP) and Atom-Centered Symmetry Functions (ACSF).
0
stat.MLcs.ITcs.LG Xuheng Li, Qingyue Zhao, Quanquan Gu · Oct 14, 2026

Feel-Good Thompson Sampling (FGTS) addresses slow exploration in vanilla TS by adding an optimism term to the posterior. This paper proves the first high-probability regret bounds for both cumulative and current-step FGTS under general realizable function approximation, showing that in linear contextual bandits the regret scales as $O(d\sqrt{T}\log(2T)+\sqrt{T}\log(1/\delta))$ with probability at least $1-\delta$. The core technique is applying Markov's inequality to exponential quantities derived from the posterior normalizer, avoiding explicit confidence-set constructions.

Feel-good Thompson sampling (FGTS) augments the posterior of vanilla Thompson sampling with an exploration term that favors reward functions with high predicted optimal values. This modification achieves minimax-optimal regret up to logarithmic factors in linear contextual bandits, but existing analyses of its two standard variants establish regret guarantees in expectation. In this paper, we derive the first high-probability regret bounds for both cumulative and current-step FGTS in contextual bandits with general realizable function approximation. Our bounds depend on the corresponding decoupling coefficients and the prior mass near the true reward function. In the $d$-dimensional linear setting, both variants satisfy $\mathrm{Regret}(T)=O\big(d\sqrt T\log(2T)+\sqrt T\log(1/\delta)\big)$ with probability at least $1-\delta$, where $T$ is the time horizon. This bound matches the dimension and horizon dependence of algorithms based on the upper confidence bounds, up to logarithmic factors. Relative to the expected-regret guarantees, the cost of controlling the regret with high probability is confined to additive terms with at most linear dependence on $\log(1/\delta)$. The proof ideas are partially developed by GPT-6 Astra under the authors' guidance.
0
stat.MEstat.ML Yundi Kong, Min Yang, Ping-Shou Zhong · Oct 14, 2026

This paper tackles measurement-constrained linear regression, where responses are expensive to collect and must be selected from a large candidate pool before being observed. The authors propose Expected Loss Optimal Selection (ELOS), which derives an A-optimality-type criterion that depends only on covariates and remains valid under potential model misspecification. The core insight is that averaging the expected prediction loss over uniformly random misspecification directions recovers the same trace-inverse criterion as under a correctly specified model, yielding a principled response-free selection rule.

In many modern applications, obtaining response variables is costly, time-consuming, or technically challenging, making it impractical to measure outcomes for all available units. In such settings, covariates may be available for a large pool of candidates, but responses can only be collected for a small, carefully selected subset, making the choice of which units to measure a consequential decision. Most existing subsampling methods are developed for massive datasets, emphasizing the trade-off between statistical accuracy and computational cost, and typically relying on the assumption of a correctly specified model. These features limit their suitability in measurement-constrained settings, where statistical efficiency is paramount and model misspecification is often unavoidable. To address these challenges, we develop a new subset selection framework for linear regression that accommodates potential model misspecification. The proposed approach targets predictive expected loss and yields an A-optimality-type criterion that depends only on covariate information, requiring no response observations at the selection stage. We refer to this method as Expected Loss Optimal Selection (ELOS). By focusing on predictive risk rather than model-based optimality, ELOS remains well-defined and robust under misspecification. Simulation studies and real data applications demonstrate that ELOS achieves strong predictive performance and compares favorably with existing methods across a range of scenarios.
0
stat.MLcs.LG Ruihang Han, Li-Hsiang Lin · Oct 14, 2026

Sequential conformal prediction struggles when the recent past is irrelevant and similar predictive states recur at nonadjacent times. This paper proposes State-Matched Sequence Conformal Prediction (SMSCP), which constructs an interpretable local-linear pre-target state and uses Mahalanobis similarity to select past calibration scores for ordinary conformal cutoff estimation. The work matters because it replaces fragile recency assumptions with an explicit state-matching mechanism, offering both modular implementation and a quantile-targeted theory that isolates when state matching helps and when it fails.

Sequential conformal prediction needs to decide which past errors should calibrate uncertainty for the current target. Recency and online error feedback are useful when recent observations remain relevant, but the recent past can be a poor reference when comparable states recur at nonadjacent times, with the recurrence pattern generally unknown. We propose State-Matched Sequence Conformal Prediction (SMSCP), which constructs an explicit one-sided local-linear state from pre-target information, uses state similarity to select a matched reference set of past calibration scores, and then applies the ordinary unweighted conformal order statistic. In contrast to existing approaches, SMSCP uses state similarity to identify calibration observations relevant to the target score quantile. This motivates a theoretical analysis of whether nearby states have comparable target score quantiles under temporal dependence. We establish control of the resulting conformal cutoff and characterize predictive information not captured by the chosen state. Simulations show that state matching is most effective when score-relevant states recur across nonadjacent times, while revealing settings in which its advantage weakens or reverses. Across solar irradiance and bike-demand benchmarks, SMSCP improves interval score over recency, weighted, adaptive, and HopCPT comparators, and achieves coverage closer to the nominal 0.90 level than ResCP on five of six benchmarks.
0
stat.MLcs.LG Isaias Banales, Arturo Jaramillo · Oct 14, 2026

This paper addresses non-parametric estimation of the background intensity in Hawkes and ETAS models by treating the background as an unknown finite positive measure with a regular density. The core idea is to derive a distribution-valued Wasserstein-Fisher-Rao (WFR) gradient from the Hawkes likelihood and then project it onto finite-dimensional statistical manifolds, yielding an explicit finite-dimensional ODE system. This avoids spatial discretization and is motivated by seismic applications, though the practical algorithmic and statistical guarantees remain partially conditional.

We consider the problem of estimating the background intensity of a Hawkes process when the triggering mechanism is known, treating the background as an unknown finite positive measure with a regular density. We propose a likelihood-based variational approach based on the Wasserstein-Fisher-Rao geometry. In this setting, the relevant first variations are naturally distribution-valued and therefore fall outside the classical gradient-flow formulation. We develop a distributional framework for the resulting dynamics and give them a finite-dimensional meaning through projections onto statistical manifolds. We identify the true background as the unique zero of a limiting generalized gradient and, under suitable stability, compactness, integrability and approximation conditions, establish a conditional consistency theorem for measurable estimators satisfying the empirical projected equations on events whose probabilities tend to one. Finally, we specialize the construction to Gaussian-mixture manifolds and obtain an explicit finite-dimensional scheme.
0
math.OCcs.LGstat.ML Zijun Chen, Zihan Zhang · Oct 14, 2026

Random reshuffling (RR) is the de facto standard for training machine learning models, yet theory has long lagged practice for last-iterate convergence. Existing guarantees required stepsizes as small as $1/(nL)$, while averaged-iterate analyses allowed $1/L$. Chen and Zhang close this gap for smooth convex objectives, proving that the last iterate of RR-SGD converges with stepsizes up to $1/(L(1+\log(nK)))$---removing the explicit $1/n$ barrier at the cost of only logarithmic factors. The result matters because practitioners return the final parameter, not an average, and large constant stepsizes are common in deep learning.

Random reshuffling stochastic gradient descent (RR-SGD) is widely used for finite-sum optimization in machine learning. For convex $L$-smooth components, recent work establishes convergence guarantees for an averaged iterate with component stepsizes as large as order $1/L$. However, the final iterate is a natural output in practice, and existing last-iterate guarantees in this setting require stepsizes that scale inversely with the number of components $n$. We establish an expected objective-gap bound for the last iterate of standard RR-SGD on unconstrained problems with a constant component stepsize satisfying $0<\eta\lesssim 1/(L(1 + \log(nK)))$, where $K$ is the number of epochs. By retaining the sampling bias at random comparison points, we remove the explicit $1/n$ factor from the stepsize restriction. Combined with existing small-step bounds, our result matches the corresponding averaged-iterate convergence rate up to logarithmic factors.
0
stat.MLcs.LGmath.ST Ruyu Zhou, Fang Liu · Oct 14, 2026

Bayesian posterior sampling offers a potentially 'free' source of differential privacy, but existing guarantees are fragmented and often require strong boundedness assumptions. This paper develops a unified Rényi-divergence framework that characterizes the inherent privacy loss of One-Posterior-Sampling (OPS) through exponential moments of the single-record log-likelihood ratio. The results yield exact, tighter bounds for a broader class of models and clarify how prior strength, sample size, and neighboring relations jointly shape privacy.

Differential Privacy (DP) provides a rigorous framework for quantifying privacy guarantees. While most methods achieve DP by injecting calibrated random noise, the intrinsic randomness of certain procedures such as sampling can yield DP guarantees ``for free''. In this work, we develop a unified R\'enyi divergence framework to characterize the inherent DP guarantees of sampling from a Bayesian posterior distribution. We show that the privacy loss incurred by One-Posterior-Sampling (OPS) is governed by posterior exponential moments of the single-record log-likelihood ratio, and establish explicit DP guarantees under uniformly bounded, sub-Gaussian, or sub-exponential tail regimes. We apply the framework to representative Bayesian models, including categorical likelihoods with arbitrary priors, Gaussian-Gaussian conjugate pair, generalized linear models from the exponential family and linear regression with Gaussian likelihood, both with Gaussian priors on the regression coefficients. Our results elucidate sample size, model structure, neighboring relations in DP (bounded or unbounded), and how prior if applicable, jointly determine the inherent DP guarantees of OPS. Furthermore, our results recover the existing DP guarantees for OPS as special cases -- often with tighter privacy loss bounds -- and substantially broaden the class of Bayesian models for which the inherent DP guarantees of OPS can be rigorously characterized.
0
cs.LGmath.OCmath.PR Nikolaos Makras, Sotirios Sabanis · Oct 1, 2026

The paper tackles sampling from Gibbs distributions on matrix spaces with potentials that are neither convex nor globally gradient-Lipschitz. Its core idea is to replace the quadratic kinetic energy of underdamped Langevin dynamics with a non-quadratic, relativistic-like energy $K_\delta(p)=\delta^{-1}\operatorname{Tr}[(I_n+\delta pp^\top)^{1/2}-I_n]$, whose gradient tames momentum singular values smoothly. By carefully coupling a momentum-dependent diffusion matrix $M_\delta(p)$ and its divergence correction $\Gamma_\delta(p)$, the authors preserve the exact target Gibbs measure, recover regularised Muon in the zero-temperature limit, and prove exponential convergence of the diffusion plus uniform-in-time moment bounds for an explicit Euler-Maruyama scheme.

We consider the problem of sampling from Gibbs distributions on matrix spaces whose potential energies are neither convex nor globally gradient-Lipschitz. We introduce a family of non-quadratic kinetic energies that lead to a new underdamped Langevin system with momentum preconditioning, in which the gradient of the kinetic energy acts as a smooth spectral taming of the momentum. We prove that, under these relaxed assumptions on the potential, the resulting dynamics leaves the target Gibbs measure invariant, and we establish exponential convergence to equilibrium in a weighted total variation distance. Finally, we show that the corresponding Euler-Maruyama discretization admits moment bounds that are uniform in time, without any modification of the potential gradient, which ensures the stability of the resulting sampling algorithm.
0
cs.LGmath.OCstat.ML Konstantinos Ziliaskopoulos, Alexander Vinel, Alice E. Smith · Oct 1, 2026

Contextual linear optimization faces a discontinuity: small prediction errors can flip the LP solution to a different vertex, making regret non-differentiable pointwise. This paper proves that, after averaging over the data distribution, the population regret is locally quadratic with curvature concentrated on the walls of the normal fan. The authors derive the exact curvature measure $K_0$, provide a tractable single-projection approximation $\mathsf{K}_{\lambda,s}$, and apply it to scenario generation, showing a $30.8\%$ regret reduction on a battery arbitrage task.

Decision-focused learning for linear optimization is complicated by the discontinuity of the optimizer, where small cost errors may leave the decision unchanged or move it to a different vertex. We show that this non-smooth pointwise behavior becomes locally quadratic after averaging over the data distribution, and we derive the curvature in closed form, specifically, a matrix-valued measure supported on the walls of the normal fan. This measure depends only on the feasible set, with the data distribution entering only as a weight. We then offer a tractable approximation for this curvature, computable with just one projection to the feasible set. We prove that the approximation weakly converges to the true population curvature. We offer one application of our findings, a decision-aware scenario generation method for expected-cost linear optimization. Our experiments test the quadratic and weak convergence laws and show a 30.8% regret improvement over uniform allocation on battery arbitrage.
0
stat.MLcs.LG Leyang Wang, Yakun Wang, Song Liu et al. · Oct 1, 2026

Comparing high-dimensional discrete distributions is notoriously difficult because the state space grows exponentially and differences can hide in complex joint dependencies. This paper introduces Zero Flux, a flow-based discrepancy that extends the continuous Zero Flow criterion to discrete state spaces. The core idea is to construct a discrete bridge between two distributions via coordinate-wise independent coupling, measure local probability fluxes at the midpoint, and show that all fluxes vanish if and only if the distributions are identical. This yields a decomposable, sample-based discrepancy with finite-sample guarantees, offering both a global comparison score and interpretable coordinate-wise contributions.

Comparing two high-dimensional discrete distributions has always been a challenging task due to the exponentially growing state space and complex changes in interactions. A recent work suggests comparing distributions through a vector field trained using flow matching between two continuous distributions. The resulting vector field at mid-point vanishes if and only if two distributions identical. However, such a flow-based criterion does not naturally apply to discrete distributions. We extend this principle to the discrete domain and introduce the \emph{Zero Flux} criterion, a discrepancy based on local probability fluxes. Under independent coupling, we show that all local probability fluxes vanish at the midpoint if and only if two distributions are the same. This discrepancy decomposes the joint distributional difference into smaller, local contributions and can be efficiently estimated from samples. We establish finite sample error bounds for our estimator. Experiments on synthetic and real categorical data demonstrate reliable recovery of sparse dependence signals and stable tracking of distribution shifts in high dimensions.
0
cs.LGstat.ML Luca Geminiani, Nadja Klein · Oct 1, 2026

IQS-BO tackles the per-iteration cost of Bayesian optimisation by replacing surrogate fitting and acquisition-function maximisation with a single transformer forward pass. The core idea is to cast query selection as supervised classification: given a finite candidate set, the model predicts which candidate maximises the objective, trained with cross-entropy on synthetic priors where the true best candidate is known for free. The method can run fully in-context or on top of a fixed surrogate, and it introduces a mixture prior of GP and non-stationary functions to improve optimisation performance.

Bayesian Optimisation (BO) is a powerful framework for the optimisation of expensive black-box functions, but typically requires refitting a surrogate and maximising an acquisition function at every evaluation step. In-context approaches based on Prior-data Fitted Networks (PFNs) amortise part of this cost by pre-training transformers on functions drawn from synthetic priors. PFNs4BO amortises the surrogate but still relies on a numerically maximised acquisition function, while FIBO performs BO fully in-context by sampling optimiser locations from a learned density, which fixes the decision rule and admits no surrogate. Learned acquisition functions score a finite candidate set with a trained network, but, lacking a label for the query, learn the score by reinforcement learning on previously solved tasks. We propose IQS-BO, a PFN that learns the query decision by supervised learning on synthetic priors. In a single forward pass, IQS-BO predicts the probability that each candidate maximises the objective over the set, and we show that the minimiser of its objective is the posterior probability of this event. The model can be pre-trained without a surrogate for fully in-context BO, or take the predictions of a fixed probabilistic surrogate as additional input, amortising only the decision step. Our method proposes queries at a fraction of the cost of acquisition-based methods, while either matching or outperforming standard BO with Gaussian processes (GPs) and available in-context methods on synthetic and real-world benchmarks. Finally, we propose a mixture prior for pre-training PFNs which combines samples from GPs with functions exhibiting warped inputs, isolated narrow optima, or plateaus that are poorly modeled by stationary kernels common in GP surrogates. We show that pre-training on this prior can lead to improved optimisation performance.
0
cs.LGcs.AIcs.RO Sebastian Sanokowski, Alireza Sarmadi, Majid Khadiv · Oct 1, 2026

Policy-gradient methods suffer from high variance, while pathwise actor updates require accurate action derivatives that standard value-function training does not guarantee. FERPO addresses this by deriving an entropy- and KL-regularized target action distribution from critic values alone, then fitting the policy with forward KL via self-normalized importance sampling---never differentiating the critic with respect to actions. The method targets massively parallel on-policy robot learning, where reliable policy improvement is critical for both sample and wall-clock efficiency.

Several state-of-the-art methods for online reinforcement learning in continuous control improve policies using action gradients of a learned critic. However, critics are typically trained to predict returns, and accurate value predictions do not necessarily yield accurate action derivatives, potentially leading to unreliable policy updates. We propose Forward Entropy-Regularized Policy Optimization (FERPO), an on-policy maximum entropy reinforcement learning algorithm that performs policy improvement using critic values without differentiating the critic with respect to actions. FERPO derives an optimal target action distribution from a policy-improvement objective regularized by entropy and Kullback-Leibler (KL) divergence. We then fit the actor to this target by minimizing a forward-KL objective, estimated using self-normalized importance sampling (SNIS) with actions drawn from the rollout policy. By limiting the target distribution's deviation from the rollout policy, the KL regularization helps keep these importance weights well behaved. In contrast to reverse-KL objectives, which can favor a subset of the target distribution's modes, the forward-KL objective encourages coverage of multiple high-value modes and thereby promotes exploration. Experiments and ablations on MuJoCo Playground and ManiSkill show competitive performance and sample-efficiency gains. Computational benchmarks also demonstrate faster actor updates than Relative Entropy Pathwise Policy Optimization (REPPO).
0
cs.LGstat.ML Haodong Wen, Kaiyue Wen, Jiaye Teng · Oct 1, 2026

This paper studies gradient descent with a large constant stepsize on linearly separable logistic regression, a canonical model for the Edge of Stability phenomenon. It proves that the worst-case exact transition time from oscillatory loss to monotone decrease scales as $\Theta((\log \eta)^{\min\{n-2,d-2\}})$ uniformly over dimension $d \ge 2$, disproving a conjecture that stepsize independence holds beyond $d=2$. The result translates into substantially stronger finite-horizon optimization guarantees by permitting exponentially larger constant stepsizes than prior $O(\eta)$ transient bounds.

We study logistic regression on linearly separable data under gradient descent with a large constant stepsize $\eta$. Such dynamics may exhibit a characteristic Edge of Stability phenomenon, in which the loss initially oscillates before transitioning to a stable phase of monotone decrease. Existing work provides a tight $\Theta(1)$ bound in dimension $d=2$ as $\eta \to \infty$ and conjectures a bound independent of $\eta$ in arbitrary dimensions $d\geq 2$. In this paper, we disprove this conjecture by showing that, for every fixed sample size $n\geq 2$ and sufficiently small margin $\gamma$, the worst-case transition time is $$\Theta\!\left((\log\eta)^{\min\{n-2,d-2\}}\right)$$ uniformly over $d\geq2$. The key challenge in establishing a tight bound is that the sample contributing most strongly to the gradient can change repeatedly across iterations. To address this issue, we control such changes by induction on dimension and sample size, and construct matching hard instances.
0
stat.MLcs.LGmath-ph Daniel McBride, Pratik Khandagale, Cristina Garcia-Cardona et al. · Oct 1, 2026

Many recent sampling algorithms built on Wasserstein gradient flows (WGF) and forward-only diffusion processes (FODP) come with exponential-convergence guarantees that are often read as proof of fast practical mixing. This paper argues that guarantee is misleading: the canonical KL-induced WGF and overdamped forward diffusion produce the same marginal Fokker--Planck evolution, so they inherit the same metastability and exponentially slow inter-mode transport known in statistical physics. Using spectral-gap and mean-first-passage-time analysis on multimodal targets, the authors show that well-separated modes induce mixing times that scale exponentially with barrier height, and that standard log-linear annealing does not remove this scaling. The conclusion is that purely local, gradient-driven sampling is structurally limited for multimodal distributions, motivating nonlocal transport mechanisms.

There has been a proliferation of sampling algorithms based on Wasserstein gradient flows (WGF) and forward-only diffusion processes (FODP), often accompanied by theoretical guarantees of exponentially fast convergence to the target distribution. These guarantees are frequently interpreted as evidence that such methods can efficiently sample complex multimodal distributions, often supported by empirical results. In this work, we argue that this interpretation is fundamentally misleading. By invoking the Jordan-Kinderlehrer-Otto (JKO) scheme and Otto calculus, we establish that the canonical WGF sampling dynamics and overdamped forward diffusion share the same density evolution and therefore inherit the same metastability and slow-mixing phenomena long understood in nonequilibrium statistical physics. We analyze this family of samplers using two complementary tools -- spectral analysis and mean first-passage time (MFPT) analysis -- and show that well-separated multimodality can induce exponentially long mixing times associated with small spectral gaps and rare inter-mode transitions. For the commonly adopted log-linear annealing schedule studied here, we find that introducing intermediate distributions does not remove the exponential scaling of the total transport time. The limitation is structural rather than implementation-specific: purely local, gradient-driven transport mechanisms can require exponentially long times to transport probability mass across well-separated modes. We argue that this represents a fundamental limitation of WGF- and FODP-based sampling in their standard forms, and motivates future development of fundamentally nonlocal mechanisms for efficient multimodal sampling.
0
math.STcs.LGstat.ML Shivam Kumar, Nabarun Deb · Oct 1, 2026

Many applications require sampling from high-dimensional categorical distributions with local dependence, but discrete diffusion models lack end-to-end statistical guarantees. This paper introduces a pinning decomposition of the discrete score under uniform noising, showing that time dependence separates multiplicatively from target dependence for Markov random fields of order $d$. Building on this, the authors design a weight-sharing neural score learner and prove a sample-complexity rate of $\widetilde O_{\mathbb P}(S^d/n)$, replacing the ambient dimension $D$ with the interaction order $d$.

Many applications in statistics, economics, and physics require sampling from high-dimensional categorical distributions with local dependence structures. Examples include finite memory language models, Ising and Potts systems in statistical physics and protein folding, etc. In modern machine learning, discrete diffusions have emerged as a flexible approach for sampling such data, with strong empirical performance. Motivated by this, we develop learning methods with end-to-end sample complexity bounds for discrete diffusion with uniform noising under local dependence, which we model through low order Markov random fields (MRFs). Our main technical insight is a new \emph{pinning decomposition} of the discrete score. It shows that unlike in continuous diffusions, the score decomposes into components where the dependence on time separates multiplicatively from the dependence on the target. Building on this decomposition, we propose a \emph{weight-sharing neural score learner} and combine it with $\tau$-leaping to obtain an end-to-end sampling procedure. Rather than treating score-learning error as a black-box input, as is common in existing sampling analyses, we study the score learning error from finite data and derive optimal sampling guarantees with explicit dependence on the vocabulary size, the interaction order of the MRF, and the sample size. Moreover, our strategy trains a single score network across uniform noise levels while leaving the sampling discretization to be chosen at inference-time. This allows the same trained model to trade accuracy for computational cost as inference-time budgets vary. Numerical experiments on Potts, Ising, and tree-structured models show that weight-sharing score networks outperform fully connected ones for sampling long sequences.
0
cs.LGstat.ML Nam Nguyen, Tuan Quang Dam · Oct 1, 2026

This paper studies fixed-confidence best-arm identification for Bernoulli bandits using the top-two $\beta$-EB-TCI rule, where the challenger is selected by a penalized Bernoulli GLR index. Its core contribution is a sharp non-asymptotic analysis: once the true best arm becomes the permanent leader and its sampling fraction stays near $\beta$, the stopping time is $T_\beta^\star(\mu)\log(1/\delta)$ up to lower-order terms. The paper also identifies exactly where the analysis stalls for the unguarded rule when suboptimal means are tied, and closes the gap with a mild $O(\sqrt{Kt})$ exploration guard.

Top-two algorithms are simple and effective for fixed-confidence best-arm identification, but their sharp non-asymptotic behavior is still not well understood. We study this problem for Bernoulli bandits through $\beta$-EB-TCI, the empirical-best top-two rule of Jourdan et al., whose challenger is chosen using a Bernoulli transportation cost with a logarithmic count penalty. We prove that, after the empirical leader has become the true best arm and its sampling fraction stays close to $\beta$, the stopping time is $T_{\beta}^{\star}(\mu)\log(1/\delta)$ up to lower-order concentration terms. We also show that, in this regime, every challenger is sampled linearly often. Thus, for the original algorithm without forced exploration, the main remaining difficulty is to control when the empirical leader becomes permanently correct. These results imply a non-asymptotic high-probability bound for all Bernoulli instances with a unique best arm. If the algorithm satisfies a finite-mean sufficient-exploration condition, the bound further yields the sharp expected sample complexity. In particular, this gives the sharp expectation result for the unguarded Bernoulli rule when all arm means are pairwise distinct, using the sufficient-exploration result of Jourdan et al. Finally, if we add a mild forced-exploration rule that contributes only $O(\sqrt{Kt})$ pulls up to time $t$, we obtain a self-contained expected sample-complexity theorem for any number of arms under the unique-best-arm assumption. We also identify a limitation of proof strategies that try to handle equal suboptimal means through a single index-comparison argument.