Agnostically Learning Single-Index Models using Omnipredictors
Aravind Gollakota, Parikshit Gopalan, Adam R. Klivans, Konstantinos Stavropoulos
Abstract
We give the first result for agnostically learning Single-Index Models (SIMs) with arbitrary monotone and Lipschitz activations. All prior work either held only in the realizable setting or required the activation to be known. Moreover, we only require the marginal to have bounded second moments, whereas all prior work required stronger distributional assumptions (such as anticoncentration or boundedness). Our algorithm is based on recent work by [GHK23] on omniprediction using predictors satisfying calibrated multiaccuracy. Our analysis is simple and relies on the relationship between Bregman divergences (or matching losses) and distances. We also provide new guarantees for standard algorithms like GLMtron and logistic regression in the agnostic setting.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1d422d5c-65e4-4fc8-89f5-7d41cb79f102Cited by top-tier papers13
- When is Multicalibration Post-Processing Necessary?Dutch Hansen, Siddartha Devic, Preetum Nakkiran, Vatsal SharanNeurIPS 2024 · 21 citations
- Robustly Learning Single-Index Models via Alignment SharpnessNikos Zarifis, Puqian Wang, Ilias Diakonikolas, Jelena DiakonikolasICML 2024 · 13 citations
- Learning Noisy Halfspaces with a Margin: Massart is No Harder than RandomGautam Chandrasekaran, Vasilis Kontonis, Konstantinos Stavropoulos, Kevin TianNeurIPS 2024 · 8 citations
- Algorithms and SQ Lower Bounds for Robustly Learning Real-valued Multi-Index ModelsIlias Diakonikolas, Giannis Iakovidis, Daniel Kane, Lisheng RenNeurIPS 2025 · 8 citations
- Improved Bounds for Swap Multicalibration and Swap OmnipredictionHaipeng Luo, Spandan Senapati, Vatsal SharanNeurIPS 2025 · 5 citations
Builds on4
- 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
- Agnostic Learning of a Single Neuron with Gradient DescentSpencer Frei, Yuan Cao, Quanquan GuNeurIPS 2020 · 68 citations
- Non-Convex SGD Learns Halfspaces with Adversarial Label NoiseIlias Diakonikolas, Vasilis Kontonis, Christos Tzamos, Nikos ZarifisNeurIPS 2020 · 38 citations
Related papers
- Omnipredicting Single-Index Models with Multi-index ModelsLunjia Hu, Kevin Tian, Chutong YangSTOC 2025 · 1 citation
- Robustly Learning Monotone Single-Index ModelsPuqian Wang, Nikos Zarifis, Ilias Diakonikolas, Jelena DiakonikolasNeurIPS 2025 · 3 citations
- Sample and Computationally Efficient Robust Learning of Gaussian Single-Index ModelsPuqian Wang, Nikos Zarifis, Ilias Diakonikolas, Jelena DiakonikolasNeurIPS 2024 · 5 citations
- Near-Optimal Algorithms for OmnipredictionPrincewill Okoroafor, Robert Kleinberg, Michael P. KimFOCS 2025 · 37 citations
- Omnipredictors for Constrained OptimizationLunjia Hu, Inbal Rachel Livni Navon, Omer Reingold, Chutong YangICML 2023 · 17 citations
