Testing Distributional Assumptions of Learning Algorithms
Ronitt Rubinfeld, Arsen Vasilyan
Abstract
There are many important high dimensional function classes that have fast agnostic learning algorithms when strong assumptions on the distribution of examples can be made, such as Gaussianity or uniformity over the domain. But how can one be sufficiently confident that the data indeed satisfies the distributional assumption, so that one can trust in the output quality of the agnostic learning algorithm? We propose a model by which to systematically study the design of tester-learner pairs (A, T ), such that if the distribution on examples in the data passes the tester T then one can safely trust the output of the agnostic learner A on the data.
To demonstrate the power of the model, we apply it to the classical problem of agnostically learning halfspaces under the standard Gaussian distribution and present a tester-learner pair with a combined runtime of n Õ(1/ǫ 4 ) . This qualitatively matches that of the best known ordinary agnostic learning algorithms for this task. In contrast, finite sample Gaussian distribution testers do not exist for the L 1 and EMD distance measures. Previously it was known that half-spaces are well-approximated with low-degree polynomials relative to the Gaussian distribution. A key step in our analysis is showing that this is the case even relative to distributions whose low-degree moments approximately match those of a Gaussian.
We also go beyond spherically-symmetric distributions, and give a tester-learner pair for halfspaces under the uniform distribution on 0, 1 n with combined run-time of n Õ(1/ǫ 4 ) . This is achieved using polynomial approximation theory and critical index machinery of [DGJ + 09].
Can one design agnostic learning algorithms under distributional assumptions and count on future technical work to produce, as a matter of course, tester-learner pairs with similar run-time? Our answer is a resounding no, as we show there exist some well-studied settings for which 2 Õ( √ n) run-time agnostic learning algorithms are available, yet the combined run-times of tester-learner pairs must be as high as 2 Ω(n) . On that account, the design of tester-learner pairs is a research direction in its own right independent of standard agnostic learning. To be specific, our lower bounds apply to the problems of agnostically learning convex sets under the Gaussian distribution and for monotone Boolean functions under the uniform distribution over 0, 1 n .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers21
- Efficient Testable Learning of Halfspaces with Adversarial Label NoiseIlias Diakonikolas, Daniel Kane, Vasilis Kontonis, Sihan Liu et al.NeurIPS 2023 · 24 citations
- Tester-Learners for Halfspaces: Universal AlgorithmsAravind Gollakota, Adam R. Klivans, Konstantinos Stavropoulos, Arsen VasilyanNeurIPS 2023 · 19 citations
- Adversarial Resilience in Sequential Prediction via AbstentionSurbhi Goel, Steve Hanneke, Shay Moran, Abhishek ShettyNeurIPS 2023 · 17 citations
- Tolerant Algorithms for Learning with Arbitrary Covariate ShiftSurbhi Goel, Abhishek Shetty, Konstantinos Stavropoulos, Arsen VasilyanNeurIPS 2024 · 17 citations
- An Efficient Tester-Learner for HalfspacesAravind Gollakota, Adam R. Klivans, Konstantinos Stavropoulos, Arsen VasilyanICLR 2024 · 16 citations
Builds on5
- Near-Optimal SQ Lower Bounds for Agnostically Learning Halfspaces and ReLUs under Gaussian MarginalsIlias Diakonikolas, Daniel Kane, Nikos ZarifisNeurIPS 2020 · 80 citations
- Statistical-Query Lower Bounds via Functional GradientsSurbhi Goel, Aravind Gollakota, Adam R. KlivansNeurIPS 2020 · 72 citations
- Non-Convex SGD Learns Halfspaces with Adversarial Label NoiseIlias Diakonikolas, Vasilis Kontonis, Christos Tzamos, Nikos ZarifisNeurIPS 2020 · 38 citations
- Optimal testing of discrete distributions with high probabilityIlias Diakonikolas, Themis Gouleakis, Daniel M. Kane, John Peebles et al.STOC 2021 · 1 citation
- A Moment-Matching Approach to Testable Learning and a New Characterization of Rademacher ComplexityAravind Gollakota, Adam R. Klivans, Pravesh K. KothariSTOC 2023
Related papers
- Testably Learning Polynomial Threshold FunctionsLucas Slot, Stefan Tiegel, Manuel WiedmerNeurIPS 2024 · 13 citations
- Agnostically Learning Multi-Index Models with QueriesIlias Diakonikolas, Daniel M. Kane, Vasilis Kontonis, Christos Tzamos et al.FOCS 2024 · 2 citations
- Reliable Learning of Halfspaces under Gaussian MarginalsIlias Diakonikolas, Lisheng Ren, Nikos ZarifisNeurIPS 2024 · 1 citation
- Near-Optimal Cryptographic Hardness of Agnostically Learning Halfspaces and ReLU Regression under Gaussian MarginalsIlias Diakonikolas, Daniel Kane, Lisheng RenICML 2023 · 40 citations
- Smoothed Agnostic Learning of Halfspaces over the HypercubeYiwen Kou, Raghu MekaNeurIPS 2025 · 3 citations
