The Exact Sample Complexity Gain from Invariances for Kernel Regression
Behrooz Tahmasebi, Stefanie Jegelka
Abstract
In practice, encoding invariances into models helps sample complexity. In this work, we tighten and generalize theoretical results on how invariances improve sample complexity. In particular, we provide minimax optimal rates for kernel ridge regression on any manifold, with a target function that is invariant to an arbitrary group action on the manifold. Our results hold for (almost) any group action, even groups of positive dimension. For a finite group, the gain increases the"effective"number of samples by the group size. For groups of positive dimension, the gain is observed by a reduction in the manifold's dimension, in addition to a factor proportional to the volume of the quotient space. Our proof takes the viewpoint of differential geometry, in contrast to the more common strategy of using invariant polynomials. Hence, this new geometric viewpoint on learning with invariances may be of independent interest.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 439f8324-33ea-4882-aef9-45dbc134a76dCited by top-tier papers13
- A Canonicalization Perspective on Invariant and Equivariant LearningGeorge Ma, Yifei Wang, Derek Lim, Stefanie Jegelka et al.NeurIPS 2024 · 38 citations
- Hardness of Learning Neural Networks under the Manifold HypothesisBobak T. Kiani, Jason Wang, Melanie WeberNeurIPS 2024 · 25 citations
- Symmetry Discovery Beyond Affine TransformationsBen Shaw, Abram Magner, Kevin R. MoonNeurIPS 2024 · 15 citations
- Achieving Approximate Symmetry Is Exponentially Easier than Exact SymmetryBehrooz Tahmasebi, Melanie WeberICLR 2026 · 8 citations
- GradMetaNet: An Equivariant Architecture for Learning on GradientsYoav Gelberg, Yam Eitan, Aviv Navon, Aviv Shamsian et al.NeurIPS 2025 · 8 citations
Builds on15
- Equivariant message passing for the prediction of tensorial properties and molecular spectraKristof Schütt, Oliver T. Unke, Michael GasteggerICML 2021 · 736 citations
- A Group-Theoretic Framework for Data AugmentationShuxiao Chen, Edgar Dobriban, Jane H. LeeNeurIPS 2020 · 254 citations
- Incorporating Symmetry into Deep Dynamics Models for Improved GeneralizationRui Wang, Robin Walters, Rose YuICLR 2021 · 201 citations
- Scalars are universal: Equivariant machine learning, structured like classical physicsSoledad Villar, David W. Hogg, Kate Storey-Fisher, Weichi Yao et al.NeurIPS 2021 · 185 citations
- On Learning Sets of Symmetric ElementsHaggai Maron, Or Litany, Gal Chechik, Ethan FetayaICML 2020 · 148 citations
Related papers
- Provably Strict Generalisation Benefit for Invariance in Kernel MethodsBryn ElesedyNeurIPS 2021 · 35 citations
- On the Sample Complexity of Learning under Geometric StabilityAlberto Bietti, Luca Venturi, Joan BrunaNeurIPS 2021 · 45 citations
- Sample complexity and effective dimension for regression on manifoldsAndrew D. McRae, Justin Romberg, Mark A. DavenportNeurIPS 2020 · 10 citations
- Learning with Exact Invariances in Polynomial TimeAshkan Soleymani, Behrooz Tahmasebi, Stefanie Jegelka, Patrick JailletICML 2025
- Sample Complexity Bounds for Estimating Probability Divergences under InvariancesBehrooz Tahmasebi, Stefanie JegelkaICML 2024 · 11 citations
