← Search

Benjamin David Haeffele

9 accepted papers

2025

A Convex Relaxation Approach to Generalization Analysis for Parallel Positively Homogeneous Networks

AISTATS 2025poster

We propose a general framework for deriving generalization bounds for parallel positively homogeneous neural networks--a class of neural networks whose input-output map decomposes as the sum of positively homogeneous maps. Examples of such networks include matrix factorization and sensing, single…

Cited by 0SourceScholar
2025

Token Statistics Transformer: Linear-Time Attention via Variational Rate Reduction

ICLR 2025spotlight

The attention operator is arguably the key distinguishing factor of transformer architectures, which have demonstrated state-of-the-art performance on a variety of tasks. However, transformer attention operators often impose a significant computational burden, with the computational complexity scali…

2024

Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models

ICLR 2024poster

The advent of large pre-trained models has brought about a paradigm shift in both visual representation learning and natural language processing. However, clustering unlabeled images, as a fundamental and classic machine learning problem, still lacks an effective solution, particularly for large-sca…

2024

Performance Bounds for Active Binary Testing with Information Maximization

ICML 2024poster

In many applications like experimental design, group testing, and medical diagnosis, the state of a random variable $Y$ is revealed by successively observing the outcomes of binary tests about $Y$. New tests are selected adaptively based on the history of outcomes observed so far. If the number of s…

Cited by 1SourcePDFScholar
2023

Learning Globally Smooth Functions on Manifolds

ICML 2023poster

Smoothness and low dimensional structures play central roles in improving generalization and stability in learning and statistics. This work combines techniques from semi-infinite constrained learning and manifold regularization to learn representations that are globally smooth on a manifold. To do…

2023

Variational Information Pursuit for Interpretable Predictions

ICLR 2023poster

There is a growing interest in the machine learning community in developing predictive algorithms that are interpretable by design. To this end, recent work proposes to sequentially ask interpretable queries about data until a high confidence prediction can be made based on the answers obtained (the…

2023

White-Box Transformers via Sparse Rate Reduction

NeurIPS 2023poster

In this paper, we contend that the objective of representation learning is to compress and transform the distribution of the data, say sets of tokens, towards a mixture of low-dimensional Gaussian distributions supported on incoherent subspaces. The quality of the final representation can be measur…

2022

Implicit Bias of Projected Subgradient Method Gives Provable Robust Recovery of Subspaces of Unknown Codimension

ICLR 2022spotlight

Robust subspace recovery (RSR) is the problem of learning a subspace from sample data points corrupted by outliers. Dual Principal Component Pursuit (DPCP) is a robust subspace recovery method that aims to find a basis for the orthogonal complement of the subspace by minimizing the sum of the distan…

Cited by 1SourcePDFScholar