Preprint
Large Language Models

PaLM

April 1, 2022

0

Citations

0

Influential Citations

Venue

2022

Year

Abstract

A 540-B parameter, densely activated, Transformer, trained using Pathways, (ML system that enables highly efficient training across multiple TPU Pods).

Analysis

Why This Paper Matters

PaLM represents a significant milestone in scaling dense Transformer models. At 540 billion parameters, it is one of the largest dense models ever trained, and its success demonstrates that dense architectures can still compete with sparse mixture-of-experts models when given sufficient compute and efficient infrastructure. The use of the Pathways system to coordinate training across multiple TPU Pods is a key engineering achievement, showing that the scaling bottleneck is as much about system design as model architecture.

This paper matters because it pushes the boundaries of what is possible with dense models, providing a strong baseline for future work. It also highlights the importance of co-designing hardware, software, and model architecture to achieve extreme scale.

Technical Contributions

  • 540B parameter dense Transformer: One of the largest dense models, showing that dense scaling remains viable.
  • Pathways training system: Enables efficient multi-TPU Pod training, overcoming communication and synchronization challenges.
  • Comprehensive evaluation: Tested on a wide range of NLP tasks, demonstrating broad capability.

Results

PaLM achieves state-of-the-art results on many benchmarks, including BIG-bench, reasoning tasks, and multilingual translation. It outperforms prior models like GPT-3 and GLaM on several metrics, showing the advantage of dense scaling when infrastructure supports it.

Significance

PaLM's success reinforces the trend toward ever-larger models and provides a template for training dense models at scale. It also underscores the importance of system-level innovations like Pathways, which may become standard for future large-scale AI training.