cs.LGJun 30, 2026

Distributionally Robust Linear Regression With Block Lewis Weights

Authors: Naren Sarayu ManojKumar Kshitij Patel

Organizations: TTIC · Institute for Foundations of Data Science, Yale University

Abstract

We present an algorithm for the group distributionally robust (GDR) least squares problem. Given mm groups, a parameter vector in Rd\mathbb{R}^d, and stacked design matrices and responses A\mathbf{A} and b\mathbf{b}, our algorithm obtains a (1+ε)(1+\varepsilon)-multiplicative optimal solution using O~(min{rank(A),m}1/3ε2/3)\widetilde{O}(\min\{\mathsf{rank}(\mathbf{A}),m\}^{1/3}\varepsilon^{-2/3}) linear-system-solves of matrices of the form ABA\mathbf{A}^{\top}\mathbf{B}\mathbf{A} for block-diagonal B\mathbf{B}. Our technical methods follow from a recent geometric construction, block Lewis weights, that relates the empirical GDR problem to a carefully chosen least squares problem and an application of accelerated proximal methods. Our algorithm improves over known interior point methods for moderate accuracy regimes and matches the state-of-the-art guarantees for the special case of \ell_{\infty} regression. We also give algorithms that smoothly interpolate between minimizing the average least squares loss and the distributionally robust loss.

Explore similar work

May 7, 2026cs.LG

Distributionally-Robust Learning to Optimize

We propose a distributionally robust approach to learning hyperparameters for first-order methods in convex optimization. Given a dataset of problem instances, we minimize a Wasserstein distributionally robust version of the performance estimation problem (PEP) over algorithm parameters such as step sizes. Our framework unifies two extremes: as the robustness radius vanishes, we recover classical learning to optimize (L2O); as it grows, we recover worst-case optimal algorithm design via PEP. We solve the resulting problem with stochastic gradient descent, differentiating through the solution of an inner semidefinite program at each step. We prove high-probability bounds showing that the true risk of the learned algorithm is at most the in-sample L2O optimum plus a slack that shrinks with the sample size, and is no worse than the worst-case PEP bound. On unconstrained quadratic minimization, LASSO, and linear programming benchmarks, our learned algorithms achieve strong out-of-sample performance with certifiable robustness, outperforming both worst-case optimal and vanilla L2O baselines.
Vinit Ranjan, Jisun Park, Bartolomeo Stellato
Sep 10, 2026stat.ML

Generalization Analysis of Distributed Kernel-based Robust Gradient Descent Algorithms

In this paper, we investigate the generalization performance of distributed gradient descent algorithms in a reproducing kernel Hilbert space under a robust loss function lσl_σ. By exploiting the spectral characterization of gradient descent together with the intrinsic properties of robust loss functions, we establish optimal learning rates for the distributed kernel-based robust gradient descent (DKRGD) algorithm with an appropriately chosen scale parameter σσ. The proposed parameter choice of σσ simultaneously alleviates the saturation phenomenon and guarantees statistical robustness. A key technical contribution is a novel error analysis that provides substantially sharper bounds for products of operators, thereby significantly relaxing existing restrictions on the maximum number of local machines while retaining optimal learning rates. Finally, we develop a communication-efficient strategy that further improves the convergence performance of DKRGD.
Jun-Yi Meng, Zheng-Chu Guo, Yuan Mao
May 13, 2026cs.LG

Byzantine-Robust Distributed Sparse Learning Revisited

We revisit Byzantine robust distributed estimation for high-dimensional sparse linear models. By combining local 1\ell_1-regularized robust estimation with robust aggregation at the server, the framework applies to pseudo-Huber regression, quantile regression, and sparse SVM. We show that the resulting estimators yield non-asymptotic guarantees and attain near-optimal statistical rates under mild conditions, while remaining communication-efficient. Simulations confirm strong robustness in estimation, support recovery and classification accuracy under various Byzantine attacks.
Yuxuan Wang, Lixin Zhang, Kangqiang Li