Learning single index models via harmonic decomposition
Nirmit Joshi, Hugo Koubbi, Theodor Misiakiewicz, Nati Srebro
摘要
We study the problem of learning single-index models, where the label depends on the input only through an unknown one-dimensional projection . Prior work has shown that under Gaussian inputs, the statistical and computational complexity of recovering is governed by the Hermite expansion of the link function. In this paper, we propose a new perspective: we argue that -- rather than -- provide the natural basis for this problem, as they capture its intrinsic . Building on this insight, we characterize the complexity of learning single-index models under arbitrary spherically symmetric input distributions. We introduce two families of estimators -- based on tensor unfolding and online SGD -- that respectively achieve either optimal sample complexity or optimal runtime, and argue that estimators achieving both may not exist in general. When specialized to Gaussian inputs, our theory not only recovers and clarifies existing results but also reveals new phenomena that had previously been overlooked.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了最后一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper5
- Positive Distribution Shift as a Framework for Understanding Tractable LearningMarko Medvedev, Idan Attias, Elisabetta Cornacchia, Theodor Misiakiewicz 等ICML 2026 · 被引用 3 次
- Full-Batch Gradient Descent Outperforms One-Pass SGD: Sample Complexity Separation in Single-Index LearningFilip Kovačević, Hong Chang Ji, Denny Wu, Mahdi Soltanolkotabi 等ICML 2026 · 被引用 2 次
- From Information to Generative Exponent: Learning Rate Induces Phase Transitions in SGDKonstantinos C. Tsiolis, Alireza Mousavi-Hosseini, Murat A. ErdogduNeurIPS 2025 · 被引用 2 次
- Convex Basins in Single-Index Model Loss Landscapes: Applications to Robust Recovery under Strong Adversarial CorruptionSANTANU DAS, Sagnik Chatterjee, jatin batraICML 2026
- Improved high-dimensional estimation with Langevin dynamics and stochastic weight averagingStanley Wei, Alex Damian, Jason D. LeeICLR 2026
它引用的顶会 Paper23
- High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the RepresentationJimmy Ba, Murat A. Erdogdu, Taiji Suzuki, Zhichao Wang 等NeurIPS 2022 · 被引用 173 次
- Learning single-index models with shallow neural networksAlberto Bietti, Joan Bruna, Clayton Sanford, Min Jae SongNeurIPS 2022 · 被引用 119 次
- Near-Optimal SQ Lower Bounds for Agnostically Learning Halfspaces and ReLUs under Gaussian MarginalsIlias Diakonikolas, Daniel Kane, Nikos ZarifisNeurIPS 2020 · 被引用 80 次
- The staircase property: How hierarchical structure can guide deep learningEmmanuel Abbe, Enric Boix-Adserà, Matthew S. Brennan, Guy Bresler 等NeurIPS 2021 · 被引用 74 次
- Phase retrieval in high dimensions: Statistical and computational phase transitionsAntoine Maillard, Bruno Loureiro, Florent Krzakala, Lenka ZdeborováNeurIPS 2020 · 被引用 73 次
相关 Paper
- Smoothing the Landscape Boosts the Signal for SGD: Optimal Sample Complexity for Learning Single Index ModelsAlex Damian, Eshaan Nichani, Rong Ge, Jason D. LeeNeurIPS 2023 · 被引用 67 次
- Learning Orthogonal Multi-Index Models: A Fine-Grained Information Exponent AnalysisYunwei Ren, Jason D. LeeNeurIPS 2025 · 被引用 7 次
- Neural network learns low-dimensional polynomials with SGD near the information-theoretic limitJason D. Lee, Kazusato Oko, Taiji Suzuki, Denny WuNeurIPS 2024 · 被引用 49 次
- Sample and Computationally Efficient Robust Learning of Gaussian Single-Index ModelsPuqian Wang, Nikos Zarifis, Ilias Diakonikolas, Jelena DiakonikolasNeurIPS 2024 · 被引用 5 次
- On Single-Index Models beyond Gaussian DataAaron Zweig, Loucas Pillaud-Vivien, Joan BrunaNeurIPS 2023 · 被引用 17 次
