← Search

Yuanzhong Xu

4 accepted papers

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
2022

GLaM: Efficient Scaling of Language Models with Mixture-of-Experts

ICML 2022spotlight

Scaling language models with more data, compute and parameters has driven significant progress in natural language processing. For example, thanks to scaling, GPT-3 was able to achieve strong results on in-context learning tasks. However, training these large dense models requires significant amount…

Cited by 765SourcePDFScholar
2022

Vector-quantized Image Modeling with Improved VQGAN

ICLR 2022poster

Pretraining language models with next-token prediction on massive text corpora has delivered phenomenal zero-shot, few-shot, transfer learning and multi-tasking capabilities on both generative and discriminative language tasks. Motivated by this success, we explore a Vector-quantized Image Modeling…

Cited by 575SourcePDFScholar
2021

GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding

ICLR 2021poster

Neural network scaling has been critical for improving the model quality in many real-world machine learning applications with vast amounts of training data and compute. Although this trend of scaling is affirmed to be a sure-fire approach for better model quality, there are challenges on the path s…

Cited by 1282SourcePDFScholar