← Search

Druv Pai

8 accepted papers

2025

Attention-Only Transformers via Unrolled Subspace Denoising

ICML 2025poster

Despite the popularity of transformers in practice, their architectures are empirically designed and neither mathematically justified nor interpretable. Moreover, as indicated by many empirical studies, some components of transformer architectures may be redundant. To derive a fully interpretable tr…

Cited by 0SourcePDFScholar
2025

Simplifying DINO via Coding Rate Regularization

ICML 2025poster

DINO and DINOv2 are two model families being widely used to learn representations from unlabeled imagery data at large scales. Their learned representations often enable state-of-the-art performance for downstream tasks, such as image classification and segmentation. However, they employ many empiri…

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

A Global Geometric Analysis of Maximal Coding Rate Reduction

ICML 2024poster

The maximal coding rate reduction (MCR$^2$) objective for learning structured and compact deep representations is drawing increasing attention, especially after its recent usage in the derivation of fully explainable and highly effective deep network architectures. However, it lacks a complete theor…

Cited by 6SourcePDFScholar
2024

Masked Completion via Structured Diffusion with White-Box Transformers

ICLR 2024poster

Modern learning frameworks often train deep neural networks with massive amounts of unlabeled data to learn representations by solving simple pretext tasks, then use the representations as foundations for downstream tasks. These networks are empirically designed; as such, they are usually not interp…

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…