← Back to brief

Source archive

arXiv Statistical ML

arXiv's stat.ML (Statistical Machine Learning) category covers the statistical and theoretical foundations of machine learning - the mathematical underpinnings behind many applied AI methods.

14 AISurfing briefingsVisit official source ↗

Briefings where arXiv Statistical ML is the primary source

ResearchOfficialarXiv Statistical ML

Heavy-Tailed Flow Matching via Random Clocks

Researchers introduce Heavy-Tailed Flow Matching via Random Clocks (HTFM), a framework that models heavy-tailed data by representing sources as mixtures of Gaussian distributions conditioned on random clock paths. The method demonstrates improved mode coverage, sample quality, and recovery of tail statistics on imbalanced datasets such as CIFAR10-LT and weather fields, while maintaining efficient sampling. HTFM also enables practical control over the heaviness of generated tails by adjusting the clock law or tail parameter.

Why it matters: This approach offers a principled and practical way to generate and control heavy-tailed distributions, which is important for applications where rare events have significant impact, such as finance and climate modeling.

Jul 16, 2026

ResearchOfficialarXiv Statistical ML

Causal Analogical Researcher (CANA) Framework Enhances LLMs' Use of Historical Analogies for Foresight Analysis

A new preprint introduces Analogical Deep Research (ADR), a task designed to evaluate large language model (LLM) agents on their ability to retrieve and integrate historical analogies for foresight analysis. The authors find that LLMs often rely on surface-level similarities rather than underlying causal mechanisms when identifying analogies. To address this, they propose the Causal Analogical Researcher (CANA) framework, which uses structural decomposition and feedback to improve analogy identification. CANA demonstrates up to a 10% improvement over previous methods on the ADR-bench benchmark.

Why it matters: This work proposes a novel framework that addresses a key limitation in LLMs' causal reasoning and could improve AI-assisted strategic analysis.

Jul 16, 2026

ModelsOfficialarXiv Statistical ML

Algorithms Achieve Fixed-Parameter Tractability for Differentially Private Synthetic Data Generation

A new preprint establishes that generating synthetic data under differential privacy is fixed-parameter tractable (FPT) when parameterized by the treewidth of the query family's incidence graph. The authors introduce two algorithms that achieve optimal error rates: one based on linear programming and the FPT of the LP dual's separation problem, and another using a subsampled private multiplicative weights method with FPT Gibbs sampling. Both approaches are unified by a dynamic programming framework over tree decompositions.

Why it matters: This result advances the theoretical understanding of private synthetic data generation, potentially enabling more efficient privacy-preserving data analysis for complex query families.

Jul 16, 2026

ResearchOfficialarXiv Statistical ML

Fast Rates for Semi-Supervised Learning via Data-Augmentation Graph Regularization

A new theoretical analysis demonstrates that data augmentation in semi-supervised learning can achieve a fast O(1/n_L) error rate with respect to the number of labeled samples, improving over the standard supervised O(1/√n_L) rate. The bound explicitly connects error to the quality of augmentations, measured by the graph-cut mass of augmentations crossing label boundaries. This work provides a mechanistic explanation for how augmentation quality influences the trade-off between accuracy and label count.

Why it matters: This is the first theoretical result to explain the labeled-sample efficiency of self-supervised learning, offering insights that could help reduce annotation costs in practice.

Jul 16, 2026

ResearchOfficialarXiv Statistical ML

Hybrid Model Combines Differentiable PDE Solvers and Neural Networks for Sparse Measurement Reconstruction

A new hybrid modeling pipeline integrates Radial Basis Function reconstruction, a neural network correction, and a differentiable partial differential equation (PDE) solver to reconstruct dense physical fields from sparse measurements. The approach enables training the neural network without access to fully-resolved simulation states, by embedding the differentiable PDE solver directly in the training loop. Evaluated on fluid mechanics benchmarks, the method outperforms existing statistical and machine-learning-based reconstruction techniques.

Why it matters: This method allows for physics-informed reconstruction from sparse data without requiring complete simulation examples, addressing a common limitation in real-world applications.

Jul 16, 2026

ResearchOfficialarXiv Statistical ML

Generalised Exponentiated Gradient Algorithm Enhances Fairness in Multi-Class and Binary Classification

Researchers have introduced a Generalised Exponentiated Gradient (GEG) algorithm designed to improve fairness in both binary and multi-class classification tasks. The in-processing method can address multiple fairness constraints simultaneously and was empirically evaluated against six baseline methods on seven multi-class and three binary datasets, using several effectiveness and fairness metrics.

Why it matters: This work advances fairness mitigation techniques by providing a method applicable to multi-class classification, an area that has received less attention despite its growing importance in real-world AI applications.

Jul 16, 2026

ResearchOfficialarXiv Statistical ML

Fisher Rank Inflation: A Spectral Signature of Memorization under Label Noise

Researchers have identified 'Fisher Rank Inflation,' a spectral signature in neural network gradients that marks the transition from learning clean data to memorizing corrupted labels. The effective rank of per-example gradients expands during memorization and contracts afterward, with the peak rank increasing as label corruption becomes more severe. This phenomenon enables detection of corrupted examples and, in some cases, the onset of memorization can be detected before test performance degrades.

Why it matters: This work provides a principled and interpretable method to detect and quantify memorization of label noise in deep networks, which could improve data auditing and robust training practices.

Jul 15, 2026

ModelsOfficialarXiv Statistical ML

Particle-Based Algorithm Advances Learning of Latent Energy-Based Models

A new algorithm has been proposed for learning latent variable models with energy-based priors, utilizing interacting particle Langevin dynamics. The approach formulates the learning process as a system of stochastic differential equations, offering theoretical convergence guarantees. Empirical results on synthetic and image datasets indicate notable improvements in computational efficiency compared to existing methods.

Why it matters: This method could make training complex energy-based models more computationally feasible, potentially broadening their practical use in machine learning.

Jul 15, 2026

ResearchOfficialarXiv Statistical ML

DAG-FM: A Foundation Model for Causal Discovery Under Heterogeneous Mechanisms

Researchers introduce DAG-FM, a novel foundation model for causal discovery that decomposes the process into two auto-regressive stages using specialized Transformer-based modules. The model incorporates a Mixture-of-Leaf-Experts approach to adapt to diverse causal mechanisms and demonstrates state-of-the-art performance on both synthetic and real-world datasets, outperforming existing classical and foundation model approaches.

Why it matters: This work represents a significant advance in scalable causal discovery, enabling more accurate inference of cause-effect relationships from complex observational data.

Jul 14, 2026

ResearchOfficialarXiv Statistical ML

Audit Reveals Distributional Reinforcement Learning Agents' Risk Claims Often Unreliable

A new preprint audits distributional reinforcement learning (RL) agents—specifically QR-DQN, C51, and IQN—and finds that 40-95% of their strongest risk trade-off claims are statistically refuted at 95% confidence. The study shows that the 'risk' learned by these agents is largely a training artifact, not a reflection of true environment stochasticity, and that following the agents' risk-based advice can sometimes lead to outcomes worse than random choice. The authors provide a statistical toolkit for auditing such claims and highlight pitfalls that can lead to misleading audit results.

Why it matters: This work raises significant concerns about the reliability of risk estimates from distributional RL agents, which are increasingly considered for safety-critical applications.

Jul 14, 2026

ResearchOfficialarXiv Statistical ML

WSqD: A Horizon-Free Learning Rate Schedule for Large Model Training

Researchers introduce WSqD, a new learning rate schedule that replaces the constant phase of the Warmup-Stable-Decay (WSD) schedule with a shifted inverse-square-root base, making it independent of the training horizon. The method achieves minimax-optimal convergence in stochastic convex optimization and, in language model pretraining experiments, matches or outperforms tuned WSD across different training horizons using a single peak learning rate.

Why it matters: WSqD could reduce the need for repeated learning rate tuning when extending training, potentially saving computational resources in large-scale model training.

Jul 14, 2026

ResearchOfficialarXiv Statistical ML

Low-dimensional adaptation of diffusion models: Convergence in total variation

A new preprint provides theoretical analysis showing that diffusion models such as DDIM and DDPM can efficiently adapt to data with unknown low-dimensional structure. The authors prove that, under exact score functions, the number of sampling iterations needed to achieve a given total variation distance scales with the intrinsic dimension of the data, rather than the ambient dimension. The results are extended to cases where the score function is learned from data, with kernel-based estimators shown to maintain this adaptivity under certain conditions.

Why it matters: This work offers the first rigorous theoretical evidence that diffusion models can efficiently sample from high-dimensional data with low intrinsic dimension, supporting and explaining their empirical success.

Jul 14, 2026

ResearchOfficialarXiv Statistical ML

Markov Chain Monte Carlo with Diffusion Paths

Researchers introduce MAD-Path, a new Markov Chain Monte Carlo (MCMC) method that samples from multimodal distributions by interpolating along diffusion paths rather than using traditional tempering. This approach preserves the relative weights of modes and improves mixing, with a Metropolis adjustment ensuring the target distribution remains invariant even with approximate intermediate scores. Experimental results demonstrate that MAD-Path achieves better exploration and more accurate mode-weight estimation compared to tempering-based MCMC methods.

Why it matters: MAD-Path addresses a longstanding challenge in statistical inference by providing a more reliable and principled method for sampling from complex, multimodal Bayesian posteriors.

Jul 14, 2026

ResearchOfficialarXiv Statistical ML

Edge Cluster Expansion with Radial Rotary Attention Improves Machine Learning Interatomic Potentials

A new study introduces two novel interaction building blocks for machine learning interatomic potentials: the Edge Complex Product Basis and Radial Rotary Complex Attention. The proposed model, TECE-OAM-RRA-1.0, demonstrates state-of-the-art performance on the Matbench Discovery benchmark, surpassing previous methods. The work also presents improvements to the Atomic Cluster Expansion module and evaluates the approach on multiple datasets.

Why it matters: Improved accuracy in machine learning interatomic potentials can significantly enhance materials science simulations and discovery.

Jul 14, 2026