cs.LGNov 3, 2026

Beyond Multi-Index Models: Learning Intersections of Degree-2 PTFs

Authors: Ilias Diakonikolas, Daniel M. Kane, Mingchen Ma, Lisheng Ren, Trung Tran

Organizations: University of Wisconsin-Madison · University of California, San Diego

Abstract

We give an algorithm for PAC learning intersections of kk degree-2 polynomial threshold functions over RdR^d under a Gaussian distribution. Our algorithm achieves classification error εε using dO(k)2Ok(log⁡3(1/ε))d^{O(k)}2^{O_k(\log^3(1/ε))} time and samples, improving over the dO(k2/ε2)d^{O(k^2/ε^2)} running time of the polynomial regression approach. To our knowledge, our result is the first to obtain a dimension exponent independent of the accuracy parameter, even for k=2k=2. We complement our algorithmic result with an SQ lower bound showing that the complexity dependence of dΩ(k)d^{Ω(k)} is unavoidable. Lifting the input to polynomial features turns the target into an intersection of kk halfspaces. This suggests using the first two labeled moments for subspace recovery. In the lifted space, however, these moments can reveal nothing. Our starting point is a Gaussian comparison theorem showing that, for sufficiently regular defining quadratics, the labeled moments needed for subspace recovery are approximately preserved. To handle arbitrary defining quadratics, we establish a new structural result showing that every direction with significant resampling influence can be detected through degree-O(k)O(k) moments. These results yield an initial dO(k)2polyk(1/ε)d^{O(k)}2^{\mathrm{poly}_k(1/ε)}-time learner using adaptive restrictions. To improve the exponential dependence on 1/ε1/ε, we develop a two-stage dimension-reduction approach that avoids the costly adaptive partitioning. After identifying a subspace containing all directions with large influence, we use generalized regularity and Gaussian comparison to recover the remaining relevant quadratic features through a new derivative-based moment operator. The recovered features support an intersection close to the target. Combining them with robust hinge-loss minimization gives our quasipolynomial-time learner.

Figures & tables

Explore similar work

CardsList
  1. Tight Bounds for Learning Polyhedra with a Margin

    Apr 16, 2026Shyamal Patel, Santosh VempalaMargin-Based LearningPAC Learning

  2. Gaussian Partitioning for Reliable Learning of Multi-Index Models

    Nov 3, 2026Ilias Diakonikolas, Giannis Iakovidis, Daniel M. Kane +1Learning with Distribution ShiftAgnostic Learning

  3. Learning Positive Linear Combinations of ReLUs in Fully Polynomial Time

    Oct 23, 2026Ilias Diakonikolas, Jelena Diakonikolas, Puqian Wang +1PAC LearningReLU Neural Networks