Fast Sketching of Polynomial Kernels of Polynomial Degree
Not on the map. Why?
Abstract (arXiv)
Kernel methods are fundamental in machine learning, and faster algorithms for kernel approximation provide direct speedups for many core tasks in machine learning. The polynomial kernel is especially important as other kernels can often be approximated by the polynomial kernel via a Taylor series expansion. Recent techniques in oblivious sketching reduce the dependence in the running time on the degree q of the polynomial kernel from exponential to polynomial, which is useful for the Gaussian kernel, for which q can be chosen to be polylogarithmic. However, for more slowly growing kernels, such as the neural tangent and arc-cosine kernels, q needs to be polynomial, and previous work incurs a polynomial factor slowdown in the running time. We give a new oblivious sketch which greatly improves upon this running time, by removing the dependence on q in the leading order term. Combined with a novel sampling scheme, we give the fastest algorithms for approximating a large family of slow-growing kernels.
Key facts. Cited by 33 papers in the map (data through 7 Oct 2026). Builds on 30 of its 59 references that are in the map. First posted on arXiv 21 Aug 2021.
Builds on
30 of 59 references.
References · 30
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
- 2015An Introduction to Matrix Concentration Inequalitiescited by 391
- 2014Sketching as a Tool for Numerical Linear Algebracited by 285
- 2018Learning Overparameterized Neural Networks via Stochastic Gradient Descent on Structured Datacited by 260
- 2018On the Convergence Rate of Training Recurrent Neural Networkscited by 82
- 2018Random Fourier Features for Kernel Ridge Regression: Approximation Bounds and Statistical Guaranteescited by 71
- 2013Divide and Conquer Kernel Ridge Regression: A Distributed Algorithm with Minimax Optimal Ratescited by 64
- 2019Quadratic Suffices for Over-parametrization via Matrix Chernoff Boundcited by 58
- 2020Training (Overparametrized) Neural Networks in Near-Linear Timecited by 58
- 2020An Improved Cutting Plane Method for Convex Optimization, Convex-Concave Games and its Applicationscited by 57
18 more
- 2019Solving Empirical Risk Minimization in the Current Matrix Multiplication Timecited by 53
- 2017Relative Error Tensor Low Rank Approximationcited by 45
- 2020Faster Dynamic Matrix Inverse for Faster LPscited by 44
- 2016Faster Kernel Ridge Regression Using Sketching and Preconditioningcited by 43
- 2016Diverse Neural Network Learns True Target Functionscited by 33
- 2016Preconditioning Kernel Matricescited by 32
- 2014Optimal CUR Matrix Decompositionscited by 29
- 2017Sketching for Kronecker Product Regression and P-splinescited by 29
- 2019Optimal Sketching for Kronecker Product Regression and Low Rank Approximationcited by 29
- 2020Generalized Leverage Score Sampling for Neural Networkscited by 26
- 2015Optimal Principal Component Analysis in Distributed and Streaming Modelscited by 24
- 2020Algorithms and Hardness for Linear Algebra on Geometric Graphscited by 22
- 2016Sharper Bounds for Regularized Data Fittingcited by 19
- 2018Subspace Embedding and Linear Regression with Orlicz Normcited by 19
- 2021FL-NTK: A Neural Tangent Kernel-based Framework for Federated Learning Convergence Analysiscited by 13
- 2020Near Input Sparsity Time Kernel Embeddings via Adaptive Samplingcited by 12
- 2017Near Optimal Sketching of Low-Rank Tensor Regressioncited by 11
- 2020Scaling up Kernel Ridge Regression via Locality Sensitive Hashingcited by 8
Built on by
Citing papers per year: 2021: 2; 2022: 8; 2023: 15; 2024: 4; 2025: 4.
33 papers.
Sort:
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023How to Capture Higher-order Correlations? Generalizing Matrix Softmax Attention to Kronecker Computationcited by 44
- 2023PolySketchFormer: Fast Transformers via Sketching Polynomial Kernelscited by 33
- 2023In-Context Learning for Attention Scheme: from Single Softmax Regression to Multiple Softmax Regression via a Tensor Trickcited by 32
- 2023Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Timecited by 31
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Timecited by 30
- 2024The Fine-Grained Complexity of Gradient Computation for Training Large Language Modelscited by 30
- 2023GradientCoin: A Peer-to-Peer Decentralized Large Language Modelscited by 23
- 2023Solving Attention Kernel Regression Problem via Pre-conditionercited by 20
- 2022Dynamic Tensor Product Regressioncited by 17
- 2023A Nearly-Optimal Bound for Fast Regression with ℓ_∞ Guaranteecited by 14
- 2022Improved Random Features for Dot Product Kernelscited by 11
21 more
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identificationcited by 11
- 2023A Theoretical Analysis Of Nearest Neighbor Search On Approximate Near Neighbor Graphcited by 10
- 2023Efficient Alternating Minimization with Applications to Weighted Low Rank Approximationcited by 9
- 2023Solving Tensor Low Cycle Rank Approximationcited by 8
- 2022Differentially Private Learning with Margin Guaranteescited by 7
- 2022Sublinear Time Algorithm for Online Weighted Bipartite Matchingcited by 7
- 2024Fast John Ellipsoid Computation with Differential Privacy Optimizationcited by 7
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapsecited by 7
- 2022Random Gegenbauer Features for Scalable Kernel Methodscited by 6
- 2022Leverage Score Sampling for Tensor Product Matrices in Input Sparsity Timecited by 6
- 2023Streaming Kernel PCA Algorithm With Small Spacecited by 5
- 2022Complex-to-Real Sketches for Tensor Products with Applications to the Polynomial Kernelcited by 3
- 2021Fast Graph Neural Tangent Kernel via Kronecker Sketchingcited by 2
- 2022On The Relative Error of Random Fourier Features for Preserving Kernel Distancecited by 2
- 2024Quantum Speedup for Spectral Approximation of Kronecker Productscited by 1
- 2024Log-concave Sampling from a Convex Body with a Barrier: a Robust and Unified Dikin Walkcited by 1
- 2025Kernel Learning for Mean-Variance Trading Strategiescited by 1
- 2025Towards Sampling Data Structures for Tensor Products in Turnstile Streamscited by 1
- 2023Online Adaptive Mahalanobis Distance Estimationcited by 0
- 2023Fast Heavy Inner Product Identification Between Weights and Inputs in Neural Network Trainingcited by 0
- 2025Semantic Distance Measurement based on Multi-Kernel Gaussian Processescited by 0
- 2025Semantic Distance Measurement based on Multi-Kernel Gaussian Processes13 Dec 2025
- 2025Towards Sampling Data Structures for Tensor Products in Turnstile Streams4 Oct 2025
- 2025Kernel Learning for Mean-Variance Trading Strategies14 Jul 2025
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapse22 May 2025
- 2024Log-concave Sampling from a Convex Body with a Barrier: a Robust and Unified Dikin Walk8 Oct 2024
- 2024Fast John Ellipsoid Computation with Differential Privacy Optimization12 Aug 2024
- 2024Quantum Speedup for Spectral Approximation of Kronecker Products10 Feb 2024
- 2024The Fine-Grained Complexity of Gradient Computation for Training Large Language Models7 Feb 2024
- 2023Fast Heavy Inner Product Identification Between Weights and Inputs in Neural Network Training19 Nov 2023
- 2023How to Capture Higher-order Correlations? Generalizing Matrix Softmax Attention to Kronecker Computation6 Oct 2023
- 2023PolySketchFormer: Fast Transformers via Sketching Polynomial Kernels2 Oct 2023
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Time14 Sep 2023
21 more
- 2023Online Adaptive Mahalanobis Distance Estimation2 Sep 2023
- 2023Solving Attention Kernel Regression Problem via Pre-conditioner28 Aug 2023
- 2023GradientCoin: A Peer-to-Peer Decentralized Large Language Models21 Aug 2023
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identification13 Jul 2023
- 2023In-Context Learning for Attention Scheme: from Single Softmax Regression to Multiple Softmax Regression via a Tensor Trick5 Jul 2023
- 2023Efficient Alternating Minimization with Applications to Weighted Low Rank Approximation7 Jun 2023
- 2023Solving Tensor Low Cycle Rank Approximation13 Apr 2023
- 2023A Theoretical Analysis Of Nearest Neighbor Search On Approximate Near Neighbor Graph10 Mar 2023
- 2023Streaming Kernel PCA Algorithm With Small Space8 Mar 2023
- 2023Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Time21 Feb 2023
- 2023A Nearly-Optimal Bound for Fast Regression with ℓ_∞ Guarantee1 Feb 2023
- 2022Dynamic Tensor Product Regression8 Oct 2022
- 2022On The Relative Error of Random Fourier Features for Preserving Kernel Distance1 Oct 2022
- 2022Sublinear Time Algorithm for Online Weighted Bipartite Matching5 Aug 2022
- 2022Differentially Private Learning with Margin Guarantees21 Apr 2022
- 2022Leverage Score Sampling for Tensor Product Matrices in Input Sparsity Time9 Feb 2022
- 2022Random Gegenbauer Features for Scalable Kernel Methods7 Feb 2022
- 2022Complex-to-Real Sketches for Tensor Products with Applications to the Polynomial Kernel4 Feb 2022
- 2022Improved Random Features for Dot Product Kernels21 Jan 2022
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Time14 Dec 2021
- 2021Fast Graph Neural Tangent Kernel via Kronecker Sketching4 Dec 2021
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023How to Capture Higher-order Correlations? Generalizing Matrix Softmax Attention to Kronecker Computationcited by 44
- 2023PolySketchFormer: Fast Transformers via Sketching Polynomial Kernelscited by 33
- 2023In-Context Learning for Attention Scheme: from Single Softmax Regression to Multiple Softmax Regression via a Tensor Trickcited by 32
- 2023Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Timecited by 31
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Timecited by 30
- 2024The Fine-Grained Complexity of Gradient Computation for Training Large Language Modelscited by 30
- 2023GradientCoin: A Peer-to-Peer Decentralized Large Language Modelscited by 23
- 2023Solving Attention Kernel Regression Problem via Pre-conditionercited by 20
- 2022Dynamic Tensor Product Regressioncited by 17
- 2023A Nearly-Optimal Bound for Fast Regression with ℓ_∞ Guaranteecited by 14
- 2022Improved Random Features for Dot Product Kernelscited by 11
12 most cited of 33 All 33 (JSON)
- 2025Semantic Distance Measurement based on Multi-Kernel Gaussian Processes13 Dec 2025
- 2025Towards Sampling Data Structures for Tensor Products in Turnstile Streams4 Oct 2025
- 2025Kernel Learning for Mean-Variance Trading Strategies14 Jul 2025
- 2025Only Large Weights (And Not Skip Connections) Can Prevent the Perils of Rank Collapse22 May 2025
- 2024Log-concave Sampling from a Convex Body with a Barrier: a Robust and Unified Dikin Walk8 Oct 2024
- 2024Fast John Ellipsoid Computation with Differential Privacy Optimization12 Aug 2024
- 2024Quantum Speedup for Spectral Approximation of Kronecker Products10 Feb 2024
- 2024The Fine-Grained Complexity of Gradient Computation for Training Large Language Models7 Feb 2024
- 2023Fast Heavy Inner Product Identification Between Weights and Inputs in Neural Network Training19 Nov 2023
- 2023How to Capture Higher-order Correlations? Generalizing Matrix Softmax Attention to Kronecker Computation6 Oct 2023
- 2023PolySketchFormer: Fast Transformers via Sketching Polynomial Kernels2 Oct 2023
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Time14 Sep 2023
12 newest of 33 All 33 (JSON)