← Search

Herve Jegou

14 accepted papers

2026

Revisiting [CLS] and Patch Token Interaction in Vision Transformers

ICLR 2026poster

Vision Transformers have emerged as powerful, scalable and versatile representation learners. To capture both global and local features, a learnable [CLS] class token is typically prepended to the input sequence of patch tokens. Despite their distinct nature, both token types are processed identical…

Cited by 0SourceScholar
2026

Short Window Attention Enables Long-Term Memorization

ICLR 2026poster

Recent works show that hybrid architectures combining sliding window softmax attention layers with linear recurrent neural network (RNN) layers outperform both of these architectures taken separately. However, the impact of the window length and the interplay between softmax attention and linear RNN…

Cited by 0SourceScholar
2023

Birth of a Transformer: A Memory Viewpoint

NeurIPS 2023spotlight

Large language models based on transformers have achieved great empirical successes. However, as they are deployed more widely, there is a growing need to better understand their internal mechanisms in order to make them more reliable. These models appear to store vast amounts of knowledge from thei…

Cited by 86SourcePDFScholar
2023

Improving Statistical Fidelity for Neural Image Compression with Implicit Local Likelihood Models

ICML 2023poster

Lossy image compression aims to represent images in as few bits as possible while maintaining fidelity to the original. Theoretical results indicate that optimizing distortion metrics such as PSNR or MS-SSIM necessarily leads to a discrepancy in the statistics of original images from those of recons…

Cited by 29SourcePDFScholar
2021

Training data-efficient image transformers & distillation through attention

ICML 2021spotlight

Recently, neural networks purely based on attention were shown to address image understanding tasks such as image classification. These high-performing vision transformers are pre-trained with hundreds of millions of images using a large infrastructure, thereby limiting their adoption. In this work,…

2021

Training with Quantization Noise for Extreme Model Compression

ICLR 2021poster

We tackle the problem of producing compact models, maximizing their accuracy for a given model size. A standard solution is to train networks with Quantization Aware Training, where the weights are quantized during training and the gradients approximated with the Straight-Through Estimator. In this…

2021

XCiT: Cross-Covariance Image Transformers

NeurIPS 2021poster

Following their success in natural language processing, transformers have recently shown much promise for computer vision. The self-attention operation underlying transformers yields global interactions between all tokens ,i.e. words or image patches, and enables flexible modelling of image data bey…

Cited by 448SourcePDFScholar
2020

Radioactive data: tracing through training

ICML 2020poster

Data tracing determines whether particular data samples have been used to train a model. We propose a new technique, radioactive data, that makes imperceptible changes to these samples such that any model trained on them will bear an identifiable mark. Given a trained model, our technique detects th…

2019

Fixing the train-test resolution discrepancy

NeurIPS 2019poster

Data-augmentation is key to the training of neural networks for image classification. This paper first shows that existing augmentations induce a significant discrepancy between the size of the objects seen by the classifier at train and test time: in fact, a lower train resolution improves the clas…

2019

Large Memory Layers with Product Keys

NeurIPS 2019spotlight

This paper introduces a structured memory which can be easily integrated into a neural network. The memory is very large by design and significantly increases the capacity of the architecture, by up to a billion parameters with a negligible computational overhead. Its design and access pattern is ba…

2019

White-box vs Black-box: Bayes Optimal Strategies for Membership Inference

ICML 2019oral

Membership inference determines, given a sample and trained parameters of a machine learning model, whether the sample was part of the training set. In this paper, we derive the optimal strategy for membership inference with a few assumptions on the distribution of the parameters. We show that optim…

Cited by 432SourcePDFScholar