cs.LGMay 20, 2026

Polynomial-Time Robust Multiclass Linear Classification under Gaussian Marginals

Authors: Ilias DiakonikolasGiannis IakovidisMingchen Ma

Organizations: University of Wisconsin-Madison

Abstract

We study the task of agnostic learning of multiclass linear classifiers under the Gaussian distribution. Given labeled examples (x,y)(x, y) from a distribution over Rd×[k]\mathbb{R}^d \times [k], with Gaussian xx-marginal, the goal is to output a hypothesis whose error is comparable to that of the best kk-class linear classifier. While the binary case k=2k=2 has a well-developed algorithmic theory, much less is known for k3k \ge 3. Even for k=3k=3, prior robust algorithms incur exponential dependence on the inverse of the desired accuracy in both complexity and representation size. In this work, we develop new structural results for multiclass linear classifiers and use them to design fully polynomial-time robust learners with dimension-independent error guarantees. Our first result shows that the standard multiclass perceptron algorithm requires super-polynomially many samples and updates, even with clean labels and Gaussian marginals, revealing a basic obstruction absent in the binary case. Our main positive result is a pairwise improper-learning framework which yields an efficient learner with error O~(k3/2opt)+ε\widetilde O(k^{3/2}\sqrt{\mathrm{opt}})+ε for general kk. Additionally, we develop a sharper localization-based framework which leads to error O(opt)+εO(\mathrm{opt})+ε for k=3k=3, and error poly(k)opt+ε\mathrm{poly}(k)\mathrm{opt}+ε for geometrically regular kk-class linear classifiers.

Explore similar work

May 26, 2026cs.DS

Proper Agnostic Learning of Functions of Halfspaces under Gaussian Marginals

We study the problem of computationally efficient proper agnostic learning of multidimensional concept classes under the Gaussian distribution. In this setting, given i.i.d. labeled samples from an unknown distribution over Rd×{±1}\mathbb{R}^d \times \{\pm 1\} whose marginal on Rd\mathbb{R}^d is Gaussian, the goal is to output a hypothesis from a target class F\mathcal{F} whose 0-1 loss is within εε of that of the best classifier in F\mathcal{F}. We give the first efficient proper agnostic learning algorithm for arbitrary Boolean functions of KK halfspaces under Gaussian marginals. Our algorithm runs in time dO(K2log(1/ε)/ε2)+(K/ε)O(K3/ε2.5)d^{O(K^2 \log(1/ε)/ε^2)} + (K/ε)^{O(K^3/ε^{2.5})}. Prior to our work, the only known algorithm for K2K \geq 2 was brute-force search, with run-time exponential in dd. Moreover, the dependence of our run-time on the dimension dd matches that of the best known improper learning algorithm, namely dO~(K2/ε2)d^{\widetilde{O}(K^2/ε^2)}. For the special case of a single halfspace (K=1K=1), the best previous run-time was dO(1/ε4)+(1/ε)O(1/ε6)d^{O(1/ε^4)} + (1/ε)^{O(1/ε^6)}. Our algorithm improves this to dO(1/ε2)+(1/ε)O(1/ε2.5)d^{O(1/ε^2)} + (1/ε)^{O(1/ε^{2.5})}. Once again, the dependence on dd matches that of the best known improper algorithm, namely dO(1/ε2)d^{O(1/ε^2)}. Furthermore, the dependence of our run-time on the dimension dd is essentially optimal in the statistical query model.
Sergei Tikhonov, Arsen Vasilyan
May 18, 2026cs.LG

Efficient and Noise-Tolerant PAC Learning of Multiclass Linear Classifiers

Noise-tolerant PAC learning of linear models has been of central interests in machine learning community since the last century. In recent years, many computationally-efficient algorithms have been proposed for the problem of learning linear threshold functions under multiple noise models. Yet, when the problem is considered under multiclass learning settings, i.e. when the number of classes kk is at least 33, it is unknown whether there exist computationally-efficient PAC learning algorithms when the data sets are maliciously corrupted. In this paper, we consider that the marginal distribution is a mixture of bounded variance distributions and the data sets satisfy a margin condition at the same time. We show that there exists a computationally-efficient algorithm that PAC learns multiclass linear classifiers {hw:xargmaxy[k]wyx,xRd,wRkd}\{h_w:x\mapsto \arg\max_{y\in[k]}w_y\cdot x, x\in \mathbb{R}^d, w\in\mathbb{R}^{kd}\} using at most O(k2(dlogd+logk))O(k^2\cdot (d\log d+\log k)) samples even under a constant rate of nasty noise. Our algorithm consists of two main ingredients: a cluster-based pruning scheme and a standard multiclass hinge loss minimization program. Even in the special case of binary setting, i.e. k=2k=2, our result is strictly stronger than all prior works.
Rita Adhikari, Shiwei Zeng
Sep 15, 2026cs.DS

Efficient Robust Learning at the Information-Theoretic Limit

In an important recent work, Blanc (2026) gave an algorithm for robustly learning Boolean concept classes with respect to a fixed distribution that outputs a (randomized) classifier achieving the optimal error of η+εη+ \varepsilon where ηη is the noise rate. In contrast, it is well known that deterministic hypotheses cannot achieve error less than 2η+ε.2η+ \varepsilon. Blanc's algorithm is computationally inefficient, and the main problem left open in his work is to find a polynomial-time algorithm given access to an oracle for empirical risk minimization (ERM). In this paper, we resolve this problem and give such an algorithm. Perhaps surprisingly, our techniques make crucial use of various types of no-regret learners. Additionally, we give an efficient algorithm (no ERM oracle required) for robustly learning any function class that admits sandwiching polynomials with respect to hypercontractive distributions. As one consequence, we give the first polynomial-time algorithm for robustly learning a halfspace with respect to Gaussian marginals that achieves error η+εη+ \varepsilon for any constant ε\varepsilon.
Adam R. Klivans, Konstantinos Stavropoulos, Sergei Tikhonov +1