arXiv Machine Learning

Quantile-based Loss Filtering for Outlier-Robust Stochastic Gradient Descent

arXiv:2609. 13040v1 Announce Type: new Abstract: We study loss-based filtering for finite-sum optimization with a subset of corrupted component functions whose gradients may be highly unreliable.

Hugging Face Trending Papers
Jun 21

Distribution-Aware Robust Bilevel Optimization: Quantile-Guided Huber Updates in Two-Timescale Stochastic Approximation

Bilevel optimization (BLO) is fundamental to hierarchical decision-making but suffers from critical instability under heavy-tailed stochastic noise. Existing variance-reduction techniques typically rely on myopic magnitude checks, which fail to distinguish informative geometric signals from impulsive outliers.

arXiv Machine Learning
Sep 4

Occupancy-based Quantile Risk Control

Occupancy-based Quantile Risk Control (OQRC) is a new method that extends conformal risk control to quantile-based risk measures. It partitions the loss space using ordered calibration losses, estimates the distribution of test losses in each bin, and upper-bounds the risk by the maximum loss per bin. The approach guarantees finite-sample validity, achieving tight risk control bounds that converge at a rate of σ(n^{-1/2}) and reducing the risk gap by up to 78.64% in experiments.

By Zihao Shi, Huajun Xi, Bingyi Jing, Hongxin Wei
arXiv Machine Learning
4d ago

Learning Distributionally Robust First-Order Methods for Convex Optimization

The paper introduces a distributionally robust method for learning hyperparameters of first‑order convex optimization algorithms. By minimizing a Wasserstein‑robust performance estimation problem over a dataset of problem instances, the approach interpolates between classical learning‑to‑optimize (L2O) and worst‑case PEP design. The authors solve the resulting problem with stochastic gradient descent, provide high‑probability risk bounds, and demonstrate that the learned algorithms outperform both worst‑case optimal and vanilla L2O baselines on logistic regression, LASSO, and linear programming tasks.

By Vinit Ranjan, Jisun Park, Bartolomeo Stellato
Hugging Face Trending Papers
Sep 2

Occupancy-based Quantile Risk Control

Occupancy-based Quantile Risk Control (OQRC) is a new method that extends conformal risk control to quantile-based risk measures while avoiding excessive conservatism and providing rigorous finite-sample guarantees. It works by partitioning the loss space using ordered calibration losses, estimating the distribution of test losses in each bin, and bounding the risk by the maximum loss in each bin. The authors prove that OQRC achieves tight risk control bounds with a finite-sample guarantee that converges at a rate of π(n−½), and experiments show it can reduce the risk gap by up to 78.64% on common benchmarks.

arXiv Machine Learning
Jun 2

Robust Learning of a Group DRO Neuron

arXiv:2601. 18115v2 Announce Type: replace Abstract: We study the problem of learning a single neuron under standard squared loss in the presence of arbitrary label noise and group-level distributional shifts, for a broad family of covariate distributions.

By Guyang Cao, Shuyao Li, Sushrut Karmalkar, Jelena Diakonikolas