Lune

NeurIPS2025Top-tier venue

Learning single index models via harmonic decomposition

Nirmit Joshi, Hugo Koubbi, Theodor Misiakiewicz, Nati Srebro

2025Year
8Citations
5Top-tier citations

Abstract

We study the problem of learning single-index models, where the label y∈Ry \in \mathbb{R} depends on the input x∈Rd\boldsymbol{x} \in \mathbb{R}^d only through an unknown one-dimensional projection ⟨w∗,x⟩\langle \boldsymbol{w}_*,\boldsymbol{x}\rangle. Prior work has shown that under Gaussian inputs, the statistical and computational complexity of recovering w∗\boldsymbol{w}_* is governed by the Hermite expansion of the link function. In this paper, we propose a new perspective: we argue that sphericalspherical harmonicsharmonics -- rather than HermiteHermite polynomialspolynomials -- provide the natural basis for this problem, as they capture its intrinsic rotationalrotational symmetrysymmetry. 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.

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.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

Cited by top-tier papers5

Ask how each one uses it

Builds on23

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines