← Search

Xiaohua Zhai

32 accepted papers

2025

Recursive Inference Scaling: A Winning Path to Scalable Inference in Language and Multimodal Systems

NeurIPS 2025poster

Inspired by recent findings on the fractal geometry of language, we introduce Recursive INference Scaling (RINS) as a complementary, plug-in recipe for scaling inference time in language and multimodal systems. RINS is a particular form of recursive depth that significantly outperforms +55 other var…

Cited by 0SourceScholar
2024

CLIP the Bias: How Useful is Balancing Data in Multimodal Learning?

ICLR 2024poster

We study data-balancing for mitigating biases in contrastive language-image pretraining (CLIP), identifying areas of strength and limitation. First, we reaffirm prior conclusions that CLIP can inadvertently absorb stereotypes. To counter this, we present a novel algorithm, called Multi-Modal Moment…

Cited by 15SourcePDFScholar
2024

LocCa: Visual Pretraining with Location-aware Captioners

NeurIPS 2024poster

Image captioning was recently found to be an effective pretraining method similar to contrastive pretraining. This opens up the largely-unexplored potential of using natural language as a flexible and powerful interface for handling diverse pretraining tasks. In this paper, we demonstrate this with…

2024

No Filter: Cultural and Socioeconomic Diversity in Contrastive Vision-Language Models

NeurIPS 2024poster

We study cultural and socioeconomic diversity in contrastive vision-language models (VLMs). Using a broad range of benchmark datasets and evaluation metrics, we bring to attention several important findings. First, the common filtering of training data to English image-text pairs disadvantages commu…

2024

On Scaling Up a Multilingual Vision and Language Model

CVPR 2024poster

We explore the boundaries of scaling up a multilingual vision and language model both in terms of size of the components and the breadth of its training task mixture. Our model achieves new levels of performance on a wide-range of varied and complex tasks including multiple image-based captioning an…

Cited by 8SourcePDFScholar
2024

SILC: Improving Vision Language Pretraining with Self-Distillation

ECCV 2024poster

"Image-Text pretraining on web-scale image caption datasets has become the default recipe for open vocabulary classification and retrieval models thanks to the success of CLIP and its variants. Several works have also used CLIP features for dense prediction tasks and have shown the emergence of open…

Cited by 27SourcePDFScholar
2023

FlexiViT: One Model for All Patch Sizes

CVPR 2023poster

Vision Transformers convert images to sequences by slicing them into patches. The size of these patches controls a speed/accuracy tradeoff, with smaller patches leading to higher accuracy at greater computational cost, but changing the patch size typically requires retraining the model. In this pape…

2023

Getting ViT in Shape: Scaling Laws for Compute-Optimal Model Design

NeurIPS 2023poster

Scaling laws have been recently employed to derive compute-optimal model size (number of parameters) for a given compute duration. We advance and refine such methods to infer compute-optimal model shapes, such as width and depth, and successfully implement this in vision transformers. Our shape-opti…

2023

Image Captioners Are Scalable Vision Learners Too

NeurIPS 2023oral

Contrastive pretraining on image-text pairs from the web is one of the most popular large-scale pretraining strategies for vision backbones, especially in the context of large multimodal models. At the same time, image captioning on this type of data is commonly considered an inferior pretraining st…

2023

PaLI: A Jointly-Scaled Multilingual Language-Image Model

ICLR 2023top-5%

Effective scaling and a flexible task interface enable large language models to excel at many tasks. We present PaLI, a model that extends this approach to the joint modeling of language and vision. PaLI generates text based on visual and textual inputs, and with this interface performs many vision,…

2023

Scaling Vision Transformers to 22 Billion Parameters

ICML 2023oral

The scaling of Transformers has driven breakthrough capabilities for language models. At present, the largest large language models (LLMs) contain upwards of 100B parameters. Vision Transformers (ViT) have introduced the same architecture to image and video modelling, but these have not yet been suc…

Cited by 650SourcePDFScholar
2023

Three Towers: Flexible Contrastive Learning with Pretrained Image Models

NeurIPS 2023poster

We introduce Three Towers (3T), a flexible method to improve the contrastive learning of vision-language models by incorporating pretrained image classifiers. While contrastive models are usually trained from scratch, LiT (Zhai et al., 2022) has recently shown performance gains from using pretrained…

2023

Tuning Computer Vision Models With Task Rewards

ICML 2023poster

Misalignment between model predictions and intended usage can be detrimental for the deployment of computer vision models. The issue is exacerbated when the task involves complex structured outputs, as it becomes harder to design procedures which address this misalignment. In natural language proces…

2022

A Simple Single-Scale Vision Transformer for Object Detection and Instance Segmentation

ECCV 2022poster

"This work presents a simple vision transformer design as a strong baseline for object localization and instance segmentation tasks. Transformers recently demonstrate competitive performance in image classification tasks. To adopt ViT to object detection and dense prediction tasks, many works inheri…

Cited by 65SourcePDFScholar
2022

Knowledge Distillation: A Good Teacher Is Patient and Consistent

CVPR 2022oral

There is a growing discrepancy in computer vision between large-scale models that achieve state-of-the-art performance and models that are affordable in practical applications. In this paper we address this issue and significantly bridge the gap between these two types of models. Throughout our empi…

Cited by 358PDFcodeScholar
2022

LiT: Zero-Shot Transfer With Locked-Image Text Tuning

CVPR 2022poster

This paper presents contrastive-tuning, a simple method employing contrastive training to align image and text models while still taking advantage of their pre-training. In our empirical study we find that locked pre-trained image models with unlocked text models work best. We call this instance of…

Cited by 627PDFScholar
2022

Revisiting Neural Scaling Laws in Language and Vision

NeurIPS 2022accept

The remarkable progress in deep learning in recent years is largely driven by improvements in scale, where bigger models are trained on larger datasets for longer schedules. To predict the benefit of scale empirically, we argue for a more rigorous methodology based on the extrapolation loss, instead…

2022

Simple Open-Vocabulary Object Detection with Vision Transformers

ECCV 2022poster

"Combining simple architectures with large-scale pre-training has led to massive improvements in image classification. For object detection, pre-training and scaling approaches are less well established, especially in the long-tailed and open-vocabulary setting, where training data is relatively sca…

2022

UViM: A Unified Modeling Approach for Vision with Learned Guiding Codes

NeurIPS 2022accept

We introduce UViM, a unified approach capable of modeling a wide range of computer vision tasks. In contrast to previous models, UViM has the same functional form for all tasks; it requires no task-specific modifications which require extensive human expertise. The approach involves two components:…

2021

A Unified Few-Shot Classification Benchmark to Compare Transfer and Meta Learning Approaches

NeurIPS 2021poster

Meta and transfer learning are two successful families of approaches to few-shot learning. Despite highly related goals, state-of-the-art advances in each family are measured largely in isolation of each other. As a result of diverging evaluation norms, a direct or thorough comparison of different a…

Cited by 19SourceScholar
2021

An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

ICLR 2021oral

While the Transformer architecture has become the de-facto standard for natural language processing tasks, its applications to computer vision remain limited. In vision, attention is either applied in conjunction with convolutional networks, or used to replace certain components of convolutional net…

2021

MLP-Mixer: An all-MLP Architecture for Vision

NeurIPS 2021poster

Convolutional Neural Networks (CNNs) are the go-to model for computer vision. Recently, attention-based networks, such as the Vision Transformer, have also become popular. In this paper we show that while convolutions and attention are both sufficient for good performance, neither of them are necess…

2021

On Robustness and Transferability of Convolutional Neural Networks

CVPR 2021poster

Modern deep convolutional networks (CNNs) are often criticized for not generalizing under distributional shifts. However, several recent breakthroughs in transfer learning suggest that these networks can cope with severe distribution shifts and successfully adapt to new tasks from a few training exa…

Cited by 165PDFcodeScholar
2021

Revisiting the Calibration of Modern Neural Networks

NeurIPS 2021poster

Accurate estimation of predictive uncertainty (model calibration) is essential for the safe application of neural networks. Many instances of miscalibration in modern neural networks have been reported, suggesting a trend that newer, more accurate models produce poorly calibrated predictions. Here,…

2020

Big Transfer (BiT): General Visual Representation Learning

ECCV 2020poster

Transfer of pre-trained representations improves sample efficiency and simplifies hyperparameter tuning when training deep neural networks for vision. We revisit the paradigm of pre-training on large supervised datasets and fine-tuning the model on a target task. We scale up pre-training, and propos…

2019

A Large-Scale Study on Regularization and Normalization in GANs

ICML 2019oral

Generative adversarial networks (GANs) are a class of deep generative models which aim to learn a target distribution in an unsupervised fashion. While they were successfully applied to many problems, training a GAN is a notoriously challenging task and requires a significant number of hyperparamete…

2019

High-Fidelity Image Generation With Fewer Labels

ICML 2019oral

Deep generative models are becoming a cornerstone of modern machine learning. Recent work on conditional generative adversarial networks has shown that learning complex, high-dimensional distributions over natural images is within reach. While the latest models are able to generate high-fidelity, di…

2019

Self-Supervised GANs via Auxiliary Rotation Loss

CVPR 2019poster

Conditional GANs are at the forefront of natural image synthesis. The main drawback of such models is the necessity for labeled data. In this work we exploit two popular unsupervised learning techniques, adversarial training and self-supervision, and take a step towards bridging the gap between cond…

Cited by 384PDFcodeScholar