Learning with Exact Invariances in Polynomial Time

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Soleymani, Ashkan, Tahmasebi, Behrooz, Jegelka, Stefanie, Jaillet, Patrick
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915771886600192
author Soleymani, Ashkan
Tahmasebi, Behrooz
Jegelka, Stefanie
Jaillet, Patrick
author_facet Soleymani, Ashkan
Tahmasebi, Behrooz
Jegelka, Stefanie
Jaillet, Patrick
contents We study the statistical-computational trade-offs for learning with exact invariances (or symmetries) using kernel regression. Traditional methods, such as data augmentation, group averaging, canonicalization, and frame-averaging, either fail to provide a polynomial-time solution or are not applicable in the kernel setting. However, with oracle access to the geometric properties of the input space, we propose a polynomial-time algorithm that learns a classifier with \emph{exact} invariances. Moreover, our approach achieves the same excess population risk (or generalization error) as the original kernel regression problem. To the best of our knowledge, this is the first polynomial-time algorithm to achieve exact (not approximate) invariances in this context. Our proof leverages tools from differential geometry, spectral theory, and optimization. A key result in our development is a new reformulation of the problem of learning under invariances as optimizing an infinite number of linearly constrained convex quadratic programs, which may be of independent interest.
format Preprint
id arxiv_https___arxiv_org_abs_2502_19758
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning with Exact Invariances in Polynomial Time
Soleymani, Ashkan
Tahmasebi, Behrooz
Jegelka, Stefanie
Jaillet, Patrick
Machine Learning
Artificial Intelligence
We study the statistical-computational trade-offs for learning with exact invariances (or symmetries) using kernel regression. Traditional methods, such as data augmentation, group averaging, canonicalization, and frame-averaging, either fail to provide a polynomial-time solution or are not applicable in the kernel setting. However, with oracle access to the geometric properties of the input space, we propose a polynomial-time algorithm that learns a classifier with \emph{exact} invariances. Moreover, our approach achieves the same excess population risk (or generalization error) as the original kernel regression problem. To the best of our knowledge, this is the first polynomial-time algorithm to achieve exact (not approximate) invariances in this context. Our proof leverages tools from differential geometry, spectral theory, and optimization. A key result in our development is a new reformulation of the problem of learning under invariances as optimizing an infinite number of linearly constrained convex quadratic programs, which may be of independent interest.
title Learning with Exact Invariances in Polynomial Time
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.19758