Learning Polynomial Transformations via Generalized Tensor Decompositions
Sitan Chen, Jerry Li, Yuanzhi Li, Anru R. Zhang
Abstract
We consider the problem of learning high dimensional polynomial transformations of Gaussians. Given samples of the form f(x), where x∼N(0,Ir) is hidden and f: ℝr → ℝd is a function where every output coordinate is a low-degree polynomial, the goal is to learn the distribution over f(x). One can think of this as a simple model for learning deep generative models, namely pushforwards of Gaussians under two-layer neural networks with polynomial activations, though the learning problem is mathematically natural in its own right.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get fd157ff7-9017-41e0-8f37-a134dd7a071cCited by top-tier papers3
- Provably Learning a Multi-head Attention LayerSitan Chen, Yuanzhi LiSTOC 2025 · 3 citations
- DNA-SE: Towards Deep Neural-Nets Assisted Semiparametric EstimationQinshuo Liu, Zixin Wang, Xi-An Li, Xinyao Ji et al.ICML 2024
- Linear Causal Representation Learning by Topological Ordering, Pruning, and DisentanglementHao Chen, Lin Liu, Yuguang WangICML 2026
Related papers
- Learning (Very) Simple Generative Models Is HardSitan Chen, Jerry Li, Yuanzhi LiNeurIPS 2022 · 12 citations
- Learning Hierarchical Polynomials with Three-Layer Neural NetworksZihao Wang, Eshaan Nichani, Jason D. LeeICLR 2024 · 7 citations
- Efficient Algorithms for Learning Depth-2 Neural Networks with General ReLU ActivationsPranjal Awasthi, Alex Tang, Aravindan VijayaraghavanNeurIPS 2021 · 24 citations
- Mean-Field Analysis for Learning Subspace-Sparse Polynomials with Gaussian InputZiang Chen, Rong GeNeurIPS 2024 · 1 citation
- Towards Understanding Hierarchical Learning: Benefits of Neural RepresentationsMinshuo Chen, Yu Bai, Jason D. Lee, Tuo Zhao et al.NeurIPS 2020 · 61 citations
