Machine Learning (Statistics) · Cross-listed from q-fin.RM
Financial Tail Risk Beyond Lipschitz Continuity via Semi-Discrete Optimal Transport
Ryan M. Engel, Kibaek Lee and Namid Stillman
Financial returns are heavy-tailed, and accurate tail risk estimation is central to portfolio risk management. Modern neural generators sample by pushing a simple base distribution through a learned map, and for training stability that map is built from Lipschitz components. This is the binding constraint: a Lipschitz map of a Gaussian is sub-Gaussian, so heavier-tailed targets admit no exact match at any finite Lipschitz constant. The Monge–Ampère equation ties the Brenier map's local distortion to the density ratio $f/(g\circ T)$, so a deeper trough in the target density requires a higher-gain map and yields a higher-variance
estimator. The argument needs only bounded distortion, so it covers normalizing flows, flow matching, GANs, and
diffusion samplers alike. Semi-Discrete Optimal Transport (SDOT) relaxes the map's regularity rather than the source's tail class. Its power diagram gives every training observation a
cell holding exactly $1/N$ of the source measure, and tail observations are reached by crossing a
cell boundary rather than by stretching. Our primary experiment sweeps severity over a calibrated Merton jump-
diffusion spanning kurtosis 94 to 1,679. SDOT holds tail ratios at $0.85$–$0.94$ with cross-seed standard deviations below $0.025$, while every learned generator either compresses the tails or inflates them with a variance that grows alongside. Further experiments carry the result to real S&P~500 returns and to a 21-year backtest, where SDOT gives the best risk-adjusted
market-neutral strategy under CVaR optimization (Sharpe $0.70$, max drawdown $-2.60\%$, against $0.40$ for the next-best generator).
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from math.OC
Supply Chain Analytics: A Data-Driven Approach
Elioth Sanabria
Modern supply chain networks increasingly rely on real-time data to navigate structural uncertainties,
market volatility, and operational disruptions. This manuscript bridges the gap between statistical data-driven learning and robust decision-making frameworks in logistics and operations management. We present a comprehensive, mathematically rigorous treatment of supply chain analytics, moving from empirical demand forecasting to optimal inventory and network control under uncertainty. Key topics explored include sample minimization, dynamic programming recursions for time-varying inventory replenishment, network fulfillment frameworks, and advanced distributionally robust optimization (DRO) via transport theory to hedge against rare events. By integrating predictive statistical models with prescriptive control
algorithms, such as column generation for vehicle routing and non-homogeneous queueing regimes, this text provides the foundational tools necessary for designing resilient, data-driven automated systems. It serves as both a theoretical blueprint and an algorithmic guide for researchers and practitioners operating at the intersection of machine learning, mathematical optimization, and applied probability.
Read the full abstract ▾
Machine Learning (Statistics)
Expressivity In Multimodal Contrastive Learning
Andrew Stuart and Florian Wolf
Contrastive learning has become a cornerstone of modern representation learning, powering CLIP-style models that underpin text-to-image generation,
vision-
language models, and
retrieval across a rapidly growing range of modalities. Despite this empirical success, the expressive power of these architectures remains poorly understood. To gain insight, we study expressivity by adopting a population-level, density-estimation viewpoint: each architecture comprises a parameterized set of densities whose parameters may be chosen to approximate the joint distribution of the modalities. This isolates a question of pure representational capacity: which joint distributions can a given contrastive family of parameterizations approximate to arbitrary accuracy? We show that expressivity is sharply architecture-dependent. For two modalities, the simple two-tower CLIP architecture is a universal approximator. A natural generalization of CLIP, widely used in practice when three or more modalities are present, is based on a loss found by summing over all pairwise similarities. This provably cannot represent arbitrary joint distributions, although we prove that it remains expressive enough to match all pairwise conditionals. Motivated by this gap, we propose Hadamard-CLIP, which adds a single learned weight vector on top of the existing encoders and restores universal approximation of the joint for any number of modalities while preserving CLIP's fast, precomputable-
embedding retrieval.
Read the full abstract ▾
Machine Learning (Statistics)
Policy Optimization and Statistical Inference for Online Contextual Matrix Games
Liner Xiang, Yixin Wang and Hengrui Cai
Online decision making often requires navigating a landscape shaped by both dynamic contexts and strategic interactions. In competitive pricing, for example, hotels must account for both dynamic contextual factors and rivals' strategic responses. Existing approaches address only part of this challenge: contextual bandits optimize single-
agent decisions using observable features but ignore multi-player interactions, while online
matrix games capture strategic behavior through Nash
equilibrium but assume fixed payoffs, ignoring contextual information. How should
agents act then when strategic payoffs evolve with contextual signals? We introduce online contextual
matrix games to integrate contextual information into multi-player online games. We further propose OnGameLearn, an online learning
algorithm that efficiently balances exploration and exploitation across both player actions and contexts. This approach comes with statistical guarantees: tail bounds for the estimated payoff
matrix, the convergence of the estimated Nash
equilibrium, the asymptotic normality of the parameter
estimators, and the sublinear regret bound. We also develop the notion of policy value in
matrix games and develop a doubly robust, $\sqrt{T}$-consistent
estimator for it. Across simulated studies and a real-world hotel pricing application, we find that OnGameLearn effectively navigates the intertwined challenges of strategic and contextual decision-making.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.IT
Expected free energy as an information constraint on the Bethe Lagrangian
Wouter M. Kouw
Active inference selects actions by minimising an expected free energy functional over predicted futures. However, adding an expectation over yet-unobserved outcomes means the free energy functional no longer has a Kullback-Leibler structure, which hinders message passing treatments of inference procedures. We propose an alternative formulation based on a Bethe free energy functional, fully supporting inference by message passing. The epistemic drive is maintained by imposing an information constraint, next to normalisation, marginalisation and form constraints, insisting that the mutual information between future observations, states and parameters given actions must be at least as large as the
entropy of the goal prior. For a specific value of the corresponding Karush-Kuhn-Tucker multiplier, the stationary point of this constrained Bethe Lagrangian recovers the expected free energy solution. We show that, as the information demand is varied, the solved multiplier moves through its inactive, interior, and saturated regimes. In the inactive regime the
agent's epistemic drive switches off entirely, while in the saturated regime it is maximal. We compare the performance of the constrained Bethe
agent on three tasks against EFE and Q-MDP.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from cs.LG
Causal Discovery in Equal Variance Linear Gaussian DAGs via SURE-Tuned Ridge Regression
Sambit Mishra and Urbashi Mitra
Recovering the directed acyclic graph (DAG) of a structural equation model (SEM) from observational data is a central problem in causal discovery. The iterative gradient descent and per-problem hyperparameter tuning of continuous-optimization methods are poorly suited to two practically important regimes: the sample-limited regime, where the number of samples is comparable to or smaller than the number of nodes in the DAG, and the compute-limited regime. This work proposes SURE-Ridge, a non-iterative, closed-form
estimator for equal variance linear Gaussian SEM. The method performs parallel node-wise regressions with regularization parameters chosen adaptively by Stein's unbiased risk estimate (SURE), and applies an adaptive thresholding procedure to extract a DAG from the resulting soft adjacency
matrix. Numerical results show that SURE-Ridge achieves the lowest structural Hamming distance in the small-sample regime and the lowest run time across all sample sizes tested, compared with NOTEARS, DAGMA, and GBNSL baselines.
Read the full abstract ▾
Machine Learning (Statistics)
Diagonal Multi-omics Integration of Heterogeneous Datasets
Maksim V.Kukushkin, Mikhail S. Arbatskiy, Dmitriy E. Balandin et al. (+1 more)
In this paper, we consider methods for the diagonal multi-omics integration of heterogeneous
datasets. Several approaches to the nature of biological heterogeneity are analyzed and developed to comprehend more clearly the generated differences. Specifically, the extremal trace problems for the coupled
Laplacian on sets homeomorphic to the Stiefel
manifold embedded in the complex Euclidean space are investigated. The gradient ascent method for the maximization problem is elaborated in the classical terms of functional analysis, which is of significant interest in itself. On this basis, we introduce a novel characteristic of
dataset heterogeneity by employing the norm of the difference between the maximum and minimum points.
Read the full abstract ▾
All authors Maksim V.Kukushkin, Mikhail S. Arbatskiy, Dmitriy E. Balandin, Alexey V. Churov
Machine Learning (Statistics)
Non-Crossing Deep Quantile Regression for Distributional Survival Prediction
Shuai Huang, Zhe Qu, Zhaowei Hua et al. (+3 more)
In survival analysis the way
covariates act on the risk of an event often differs between early and late failure times, yet hazard- and mean-based summaries collapse this variation into a single number.
Quantile-based modeling instead describes the full conditional distribution on the original time scale, but existing censored-data methods are either inflexible or produce logically inconsistent crossing
quantile curves. We propose a Censored Non-crossing
Quantile (CNQ) framework for right-censored data that jointly estimates several conditional survival
quantiles and guarantees valid ordering by construction, with flexibility supplied by Kolmogorov-Arnold and
Transformer backbones, and we establish a finite-sample excess-risk bound holding jointly across all fitted
quantile levels. Across 27 simulation settings and six cohorts the framework attains lower pinball loss than
quantile-, hazard- and tree-based competitors whenever the conditional distribution is asymmetric, with interval coverage closer to nominal on all six. In two clinical case studies (METABRIC, breast cancer; FLCHAIN, population mortality) it recovers
covariate effects that vary across the survival distribution and would be hidden by a single hazard ratio, and yields coherent individualized
quantile milestones. Code: https://github.com/BIG-S2/deepcnq
Read the full abstract ▾
All authors Shuai Huang, Zhe Qu, Zhaowei Hua, Guohao Shen, Rui Tang, Hongtu Zhu
Machine Learning (Statistics)
Hide&Seek: Learning to Explain in an End-to-End Differentiable Network
Tal Ellinson, Hadi Mohasel Afshar and Sally Cripps
Instance-wise feature selection is a valuable tool for interpreting labeled data and the predictions of black-box models. In contrast to global feature selection techniques, instance-wise methods dynamically identify important features for each instance. A growing number of methods learn a selector, which identifies important features, and a predictor, which uses these to make predictions. However, these pioneering methods face challenges including information leakage and lack of differentiability, which can slow training. In this paper, we present Hide&Seek, an end-to-end differentiable model for instance-wise feature selection. We jointly learn feature selection and prediction under a single objective without information leakage. Hide&Seek outperforms existing state-of-the-art models across a range of experiments and is fast to train. We achieve this by reformulating feature removal as a differentiable operation where instead of discretely removing features, we replace a proportion of each feature. Training is further stabilized via a parsimony-weight annealing framework.
Read the full abstract ▾
Machine Learning (Statistics)
Density-Reweighted Entropic Optimal Transport: Decoupling Geometry from Sampling Density
Keyi Li, Yuval Kluger and Boris Landa
Dataset alignment is a central step in data analysis across science and engineering, where the goal is to match observations between
datasets. Entropic Optimal Transport (EOT) offers a computationally tractable framework for this task by encoding cross-
dataset affinities in a transport plan. However, when two
datasets are sampled from geometrically similar low-dimensional structures with substantially different sampling densities, the EOT plan may match points by relative sampling density rather than
geometric proximity, yielding geometrically misleading correspondences. To address this issue, we propose a density-reweighted EOT framework in which the influence of sampling density on the transport plan can be discounted to a desired degree, ranging from standard EOT to
alignment driven purely by underlying geometry. Under suitable regularity conditions, we establish convergence of the reweighted EOT plan to a family of population-level plans whose dependence on sampling density is made explicit. Through simulations, we show that our approach recovers geometrically faithful correspondences, improving over related EOT-based frameworks when
datasets exhibit substantial sampling density disparity.
Read the full abstract ▾
Machine Learning (Statistics)
Improved Regret Analysis for Parallel Gaussian Process Bandit Optimization
Shion Takeno and Shogo Iwazaki
This paper studies the regret analysis for parallel Gaussian process (GP) bandit optimization. The known regret upper bounds for the widely used GP batched upper confidence bound and GP batched Thompson sampling (GP-BTS) suffer from a multiplicative factor with respect to the batch size $Q$. To avoid this degradation, existing analyses require a
polynomial number of uncertainty sampling (US) for $Q$ at the beginning of optimization. However, this initial US phase is often ineffective in practice. This paper shows that the regret upper bound without the multiplicative factor on $Q$ can be achieved without the initial US phase, using GP-BTS as an example. Furthermore, we show much better regret upper bounds in the noiseless setting than in the noisy setting, as in the sequential GP bandit setting.
Read the full abstract ▾
Machine Learning (Statistics) · Cross-listed from stat.ME
Deep adaptive design with an evidential bias criterion
David Chen, Michael Evans, Xinwei Li et al. (+2 more)
Bayesian optimal experimental design (BOED) aims to collect informative data by optimizing an expected utility reflecting the goals of an experiment. However, this optimization is computationally challenging for common utilities and complex models. This is especially so for sequential or adaptive designs, where design and data collection alternate, so that feedback from already observed data must be taken into account. Most existing BOED research employs information gain as the utility, leading to the expected information gain (EIG) criterion. While EIG is widely useful, it may not always adequately reflect experimental goals. EIG can be viewed as rewarding experiments that produce large positive evidence for the truth on average, but it does not directly control the risk of an experiment producing misleading evidence. Here we consider an alternative criterion, which we call bias against (BA), that prioritizes such control. To address computational challenges when applying this criterion for adaptive design, we consider a policy-based deep adaptive design framework, which has previously been used for the EIG criterion. Minimizing a tractable upper bound on the BA objective is equivalent to maximizing a variance-penalized EIG criterion, and we optimize the latter by approximating it by Monte Carlo and learning design policies using
stochastic gradient methods. The differences between BA and EIG designs are demonstrated in several examples including the adaptive design of a complex discrete choice experiment.
Read the full abstract ▾
All authors David Chen, Michael Evans, Xinwei Li, Prateek Bansal, David J. Nott
Machine Learning (Statistics) · Cross-listed from math.NA
Convergence Analysis of Statistical Inverse Problems on Reproducing Kernel Banach Spaces
Darrel K Joseph and M P Rajan
Statistical inverse problems have garnered significant
attention in recent years due to the growing importance of statistical learning theory and functional analytic approaches in the fields of machine learning and artificial intelligence. In this paper, we investigate the stable approximation of the element $u^{\dagger}$ that satisfies the equation $Au = g$, where $A$ is a linear operator that maps a Banach space into an appropriate function space. The function $g$ is observed only through independently and identically distributed data points that are corrupted by noise and assumed to follow an unknown distribution $\rho$. We employ the Tikhonov regularization scheme, leveraging statistical learning techniques and the framework of reproducing kernel Banach spaces to estimate the solution. We establish convergence and derive the convergence rate of the estimated solution with respect to the true solution as the number of data points increases, with the rate expressed in probabilistic terms. The theoretical findings are further supported by numerical experiments that demonstrate the effectiveness of the proposed approach.
Read the full abstract ▾
Machine Learning (Statistics)
LiD-GLM: Lipschitz-constrained Deep Generalized Linear Models
Tom Splittgerber, Niklas Koenen, Marvin N. Wright et al. (+1 more)
The combination of traditional statistical models and neural network (NN) components into semi-structured hybrid models is an intriguing approach to construct models that, ideally, combine traditional interpretability with the unprecedented flexibility of NNs. In order to preserve interpretability, it is usually necessary to restrict the NN components to prevent them from dominating the model. However, existing methods that enforce structural constraints on their NN components severely limit their models' flexibility; in contrast, methods that only enforce weak, indirect constraints lose meaningful interpretability. The method we propose therefore leverages invertible residual neural networks (i-ResNets) to equip generalized linear models with both nonlinear parameter estimation and a flexible correction of their distributional assumptions while always retaining
stochastic monotonicity of the modeled distribution in the (formerly linear) predictor. The i-ResNets correspond to a controlled deviation from identity and by constraining their Lipschitz constant one can rigorously limit and quantify how far the hybrid model deviates from its traditional counterpart. This enables a user-specifiable compromise between flexibility and interpretability without limiting the structure of nonlinear and interaction effects that can be learned. Furthermore, we develop specific inherent interpretation techniques for our model and enforce model identifiability through an adapted post-hoc orthogonalization.
Read the full abstract ▾
All authors Tom Splittgerber, Niklas Koenen, Marvin N. Wright, Werner Brannath
Machine Learning (Statistics) · Cross-listed from cs.LG
Conditional Evaluation of Language Models with Cheap Auxiliary Signals
Zhi Zhang, Lingfeng Lyu, Yue Kang et al. (+1 more)
Aggregate accuracy hides where models succeed and fail. Estimating conditional performance profiles from gold labels alone is expensive, while cheap auxiliary signals such as LLM-judge scores, pairwise comparisons, confidence scores, and judge-disagreement features can be collected for every
benchmark item but are often biased or miscalibrated. We propose LACE (Local Augmented Control-Variate Evaluation), a semi-supervised
estimator for conditional LLM evaluation. The key step is local centering: after subtracting the conditional mean of a cheap signal within the target profile region, any linear augmentation has zero conditional mean and therefore cannot change the estimand. The augmentation coefficient is used only for efficiency, and a local ridge control variate combines a gold-label residual mean from the labeled subset with a cheap-signal mean from the full item pool. We prove calibration-free identification, unbiasedness for grouped profiles, local oracle optimality within centered linear augmentations, and first-order adaptivity to the estimated coefficient. The resulting gain formula is governed by a population local $R^2$, which characterizes how the efficiency attainable from the cheap signals varies across profile values. We also derive corresponding
estimators for direct paired model gaps and deployment-weighted scores. We empirically evaluate the primary performance-profile
estimator on MATH-500, ScienceQA, MMLU, WinoGrande, HellaSwag, TruthfulQA, GSM8K, and ARC.
Read the full abstract ▾
All authors Zhi Zhang, Lingfeng Lyu, Yue Kang, Doudou Zhou
Machine Learning (Statistics)
Coded Hankel Polynomial Chaos: Spectral Identification of Dominant Polynomial-Chaos Modes
Zhiliang Deng and Xiaomei Yang
Identification of dominant
polynomial-chaos modes is usually formulated as a sparse-regression problem on a sampled multivariate
polynomial dictionary. We develop coded Hankel
polynomial chaos (CH-PC), a complementary spectral formulation for dominant-mode identification. A finite generating transform converts PCE coefficients into a coefficient-generating
polynomial, and evaluation along a
geometric phase orbit produces a finite exponential sum. Its model order and spectral nodes are encoded by low-rank Hankel
matrices, while coordinate phase shifts attach root-of-unity labels from which the full
polynomial multi-indices are recovered. Coordinate-shifted probes are combined as common-node snapshots, and independent phase encodings provide redundant representations when a single spectral encoding is poorly conditioned. For finite observations, population, finite-data, and observed probes are kept distinct: sampling or quadrature error and observation error enter as separate Hankel perturbations, which are then connected to spectral stability, discrete decoding, and phase voting. For tensor-product candidate sets, the generating kernel factorizes into one-dimensional sums and can be evaluated without assembling the full multivariate PCE design
matrix. Numerical experiments on sparse Legendre
benchmarks and a
stochastic Darcy problem illustrate exact recovery, noise stabilization, unknown-order identification by phase persistence, and dominant-mode recovery for a PDE-generated quantity of interest.
Read the full abstract ▾
Machine Learning (Statistics)
EMS Coreset: An Efficient Expectation-Maximization Algorithm for Sinkhorn Coreset
Haoyun Yin, Chuanhui Liu and Xiao Wang
Coresets distill large
datasets into small, representative subsets for efficient downstream learning. Yet Optimal Transport (OT)-based selection typically requires intensive computation of transport plans, limiting scalability. We introduce a scalable Sinkhorn coreset method that permits closed-form updates of the entropically regularized OT coupling by allowing non-uniform coreset weights. This produces centroids that generalize k-means via soft assignments. We establish asymptotic consistency of the selected measure and Lipschitz stability to data perturbations, providing accuracy and robustness guarantees. Across synthetic and real-world
benchmarks, the proposed method achieves competitive or improved approximation quality while substantially reducing runtime compared to Wasserstein- and standard Sinkhorn-based coreset selection, especially at large scale.
Read the full abstract ▾