AI RESEARCH

Average Gradient Outer Product in kernel regression provably recovers the central subspace for multi-index models

arXiv CS.LG

ArXi:2605.15082v1 Announce Type: cross We study a prototypical situation when a learned predictor can discover useful low-dimensional structure in data, while using fewer samples than are needed for accurate prediction. Specifically, we consider the problem of recovering a multi-index polynomial $f^*(x)=h(Ux)$, with $U\in\mathbb{R}^{r\times d}$ and $r\ll d$, from finitely many data/label pairs. Importantly, the target function depends on input $x$ only through the projection onto an unknown $r$-dimensional central subspace.