Convex Basins in Single-Index Model Loss Landscapes: Applications to Robust Recovery under Strong Adversarial Corruption
SANTANU DAS, Sagnik Chatterjee, jatin batra
Abstract
In this paper, we tackle a fundamental problem in high-dimensional statistics, namely, learning Single Index Models (SIMs) robustly in the presence of heavy-tailed noise and an adversary that can corrupt a constant fraction of both covariates and responses. Prior research on efficient robust recovery only focuses on monotonic link functions or only limit themselves to Phase Retrieval. Provable efficient robust recovery guarantees for generic nonlinear link functions have remained elusive. In this paper, we obtain the first near-linear time, optimal-sample-complexity robust recovery algorithm for a wide class of nonlinear non-monotonic link functions. Critical to our result is an improved understanding of the squared-loss landscape: we identify a sufficient condition under which a broad class of non linear link functions admit a dimension-independent constant-radius convex basin around the ground truth, establishing statistical identifiability beyond previously known cases. We also leverage second-order Stein's identities to identify a structural condition, that we term Expected Squared Convexity (ESC), that acts as a sufficient condition for spectral initialization techniques to obtain an estimator within the convex basin with error , even under heavy-tailed noise and strong adversarial contamination. This robust initialization technique can be combined with a robust gradient descent phase to break the spectral error barrier, achieving an improved estimation error of . Our non-convex optimization framework gives the first efficient sample and time complexity robust recovery results for activation functions such as GeLU and Swish that act as building blocks of modern deep-learning architectures.
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.
Builds on6
- 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 citations
- Emergence and scaling laws in SGD learning of shallow neural networksYunwei Ren, Eshaan Nichani, Denny Wu, Jason D. LeeNeurIPS 2025 · 33 citations
- Streaming Algorithms for High-Dimensional Robust StatisticsIlias Diakonikolas, Daniel M. Kane, Ankit Pensia, Thanasis PittasICML 2022 · 25 citations
- On Single-Index Models beyond Gaussian DataAaron Zweig, Loucas Pillaud-Vivien, Joan BrunaNeurIPS 2023 · 17 citations
- Nearly-Linear Time and Streaming Algorithms for Outlier-Robust PCAIlias Diakonikolas, Daniel Kane, Ankit Pensia, Thanasis PittasICML 2023 · 11 citations
Related papers
- Sample and Computationally Efficient Robust Learning of Gaussian Single-Index ModelsPuqian Wang, Nikos Zarifis, Ilias Diakonikolas, Jelena DiakonikolasNeurIPS 2024 · 5 citations
- Cryptanalytic Extraction of Deep Neural Networks with Non-linear ActivationsRoderick Asselineau, Patrick Derbez, Pierre-Alain Fouque, Brice MinaudCRYPTO 2026 · 8 citations
- Interactive Learning of Single-Index Models via Stochastic Gradient DescentNived Rajaraman, Yanjun HanICLR 2026 · 1 citation
- Robust Regression Revisited: Acceleration and Improved Estimation RatesArun Jambulapati, Jerry Li, Tselil Schramm, Kevin TianNeurIPS 2021 · 18 citations
- Robustly Learning Monotone Single-Index ModelsPuqian Wang, Nikos Zarifis, Ilias Diakonikolas, Jelena DiakonikolasNeurIPS 2025 · 3 citations
