← Search

Behrooz Tahmasebi

12 accepted papers

2025

A Robust Kernel Statistical Test of Invariance: Detecting Subtle Asymmetries

AISTATS 2025oral

While invariances naturally arise in almost any type of real-world data, no efficient and robust test exists for detecting them in observational data under arbitrarily given group actions. We tackle this problem by studying measures of invariance that can capture even negligible underlying patterns.…

Cited by 0SourceScholar
2025

Generalization Bounds for Canonicalization: A Comparative Study with Group Averaging

ICLR 2025poster

Canonicalization, a popular method for generating invariant or equivariant function classes from arbitrary function sets, involves initial data projection onto a reduced input space subset, followed by applying any learning method to the projected dataset. Despite recent research on the expressive p…

Cited by 0SourcePDFScholar
2025

Geometric Algorithms for Neural Combinatorial Optimization with Constraints

NeurIPS 2025poster

Self-Supervised Learning (SSL) for Combinatorial Optimization (CO) is an emerging paradigm for solving combinatorial problems using neural networks. In this paper, we address a central challenge of SSL for CO: solving problems with discrete constraints. We design an end-to-end differentiable framewo…

Cited by 0SourceScholar
2025

Learning with Exact Invariances in Polynomial Time

ICML 2025spotlight

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 applicab…

Cited by 0SourcePDFScholar
2024

A Universal Class of Sharpness-Aware Minimization Algorithms

ICML 2024poster

Recently, there has been a surge in interest in developing optimization algorithms for overparameterized models as achieving generalization is believed to require algorithms with suitable biases. This interest centers on minimizing sharpness of the original loss function; the Sharpness-Aware Minimiz…

2024

Coded Computing for Resilient Distributed Computing: A Learning-Theoretic Framework

NeurIPS 2024poster

Coded computing has emerged as a promising framework for tackling significant challenges in large-scale distributed computing, including the presence of slow, faulty, or compromised servers. In this approach, each worker node processes a combination of the data, rather than the raw data itself. The…

Cited by 2SourcePDFScholar
2024

Sample Complexity Bounds for Estimating Probability Divergences under Invariances

ICML 2024poster

Group-invariant probability distributions appear in many data-generative models in machine learning, such as graphs, point clouds, and images. In practice, one often needs to estimate divergences between such distributions. In this work, we study how the inherent invariances, with respect to any smo…

Cited by 8SourcePDFScholar
2023

The Exact Sample Complexity Gain from Invariances for Kernel Regression

NeurIPS 2023spotlight

In practice, encoding invariances into models improves sample complexity. In this work, we study this phenomenon from a theoretical perspective. In particular, we provide minimax optimal rates for kernel ridge regression on compact manifolds, with a target function that is invariant to a group actio…

Cited by 21SourcePDFScholar
2023

The Power of Recursion in Graph Neural Networks for Counting Substructures

AISTATS 2023poster

To achieve a graph representation, most Graph Neural Networks (GNNs) follow two steps: first, each graph is decomposed into a number of subgraphs (which we call the recursion step), and then the collection of subgraphs is encoded by several iterative pooling steps. While recently proposed higher-ord…

Cited by 13SourcePDFScholar