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

Apr 16, 2026

Tight Bounds for Learning Polyhedra with a Margin

We give an algorithm for PAC learning intersections of kk halfspaces with a ρρ margin to within error ε\varepsilon that runs in time poly(k,ε−1,ρ−1)⋅exp⁡(O(nlog⁡(1/ρ)log⁡k))\textsf{poly}(k, \varepsilon^{-1}, ρ^{-1}) \cdot \exp \left(O(\sqrt{n \log(1/ρ) \log k})\right). Notably, this improves on prior work which had an exponential dependence on either kk or ρ−1ρ^{-1} and matches known cryptographic and Statistical Query lower bounds up to the logarithmic factors in kk and ρρ in the exponent. Our learning algorithm extends to the more general setting when we are only promised that most points have distance at least ρρ from the boundary of the polyhedron, making it applicable to continuous distributions as well.
Nov 3, 2026

Gaussian Partitioning for Reliable Learning of Multi-Index Models

We introduce Gaussian partitioning, an algorithmic primitive for distributions smooth with respect to a Gaussian. It partitions space into polyhedral regions so that almost all mass lies in thin strips restricting a few orthogonal directions. Within each strip, the conditional distribution approximately matches low-degree Gaussian moments in the unrestricted directions. Our construction builds on logconcave partitioning of Diakonikolas, Kane, and Ma (FOCS 2026). Under input marginals smooth with respect to the standard Gaussian, we obtain agnostic learners with error (1+c)opt+ε(1+c)\mathrm{opt}+ε for broad classes of Boolean multi-index models, including functions of halfspaces, low-dimensional convex concepts, and low-dimensional polynomial threshold functions. For functions of kk halfspaces, both agnostic and testable agnostic learners achieve this error with dimension dependence dpoly(k/c)d^{\mathrm{poly}(k/c)}, improving the previous exponential dependence on k/ck/c in the agnostic exponent. For learning from positive examples with imperfect unlabeled data (PUI), we give algorithms and Statistical Query (SQ) lower bounds that qualitatively characterize complexity through Gaussian moment matching. For fixed smoothness and positive-mass parameters, we learn intersections of kk halfspaces with dimension dependence dO(log⁡(k/ε))d^{O(\log(k/ε))} and matching SQ lower bounds, improving the previous dpoly(1/ε)d^{\mathrm{poly}(1/ε)} upper bound even for one halfspace. A general reduction from agnostic reliable learning to PUI yields negative reliable learners for these intersections with the same dimension dependence, for fixed smoothness. Finally, for PQ learning and testable learning with distribution shift, we prove qualitatively optimal SQ lower bounds of dΩ(1/ε2)d^{Ω(1/ε^2)}, even for intersections of two halfspaces with Gaussian training data and arbitrary test distributions.
Oct 23, 2026

Learning Positive Linear Combinations of ReLUs in Fully Polynomial Time

We study the fundamental problem of PAC learning positive linear combinations of kk arbitrary ReLU activations on Rd\mathbb{R}^d with respect to the Gaussian distribution, in L2L_2 loss. We give the first fully-polynomial time algorithm for this learning task. Specifically, our algorithm draws poly(d,k,1/ε)\mathrm{poly}(d, k, 1/ε) samples, runs in poly(d,k,1/ε)\mathrm{poly}(d, k, 1/ε) time, and outputs a hypothesis that is εε-close to the target concept with high probability. All prior algorithms for this problem require super-polynomial time in kk or 1/ε1/ε, or leverage strong well-conditioned assumptions on the weight matrix. On a technical level, our approach develops a novel boosting procedure of a weak'' learner that recovers the hidden weight vectors to constant error, in a list-decodable manner. To develop this weak'' learning procedure, we leverage implicit tensor computation techniques to construct a low-degree filtering procedure based on Hermite tensors of the target function. This procedure amplifies separation among (clusters of) weight vectors to learn each of them to constant error. We then argue that the same procedure can be used for improving the estimation of hidden weight vectors, using gradient descent-like updates on the Euclidean sphere. A crucial ingredient of our algorithm is a pruning approach that ensures only relevant vectors that approximate the hidden weight vectors at target accuracy are maintained by the algorithm.