← Search

Gagan Jain

3 accepted papers

2025

Masked Generative Nested Transformers with Decode Time Scaling

ICML 2025poster

Recent advances in visual generation have made significant strides in producing content of exceptional quality. However, most methods suffer from a fundamental problem - a bottleneck of inference computational efficiency. Most of these algorithms involve multiple passes over a transformer model to g…

Cited by 0SourcePDFScholar
2024

LookupViT: Compressing visual information to a limited number of tokens

ECCV 2024poster

"Vision Transformers (ViT) have emerged as the de-facto choice for numerous industry grade vision solutions. But their inference cost can be prohibitive for many settings, as they compute self-attention in each layer which suffers from quadratic computational complexity in the number of tokens. On t…

Cited by 11SourcePDFScholar
2024

Mixture of Nested Experts: Adaptive Processing of Visual Tokens

NeurIPS 2024poster

The visual medium (images and videos) naturally contains a large amount of information redundancy, thereby providing a great opportunity for leveraging efficiency in processing. While Vision Transformer (ViT) based models scale effectively to large data regimes, they fail to capitalize on this inher…

Cited by 8SourcePDFScholar