← Search

Aidan Clark

6 accepted papers

2022

An empirical analysis of compute-optimal large language model training

NeurIPS 2022accept

We investigate the optimal model size and number of tokens for training a transformer language model under a given compute budget. We find that current large language models are significantly undertrained, a consequence of the recent focus on scaling language models whilst keeping the amount of trai…

Cited by 171SourcePDFScholar
2022

Improving Language Models by Retrieving from Trillions of Tokens

ICML 2022spotlight

We enhance auto-regressive language models by conditioning on document chunks retrieved from a large corpus, based on local similarity with preceding tokens. With a 2 trillion token database, our Retrieval-Enhanced Transformer (RETRO) obtains comparable performance to GPT-3 and Jurassic-1 on the Pil…

2022

Unified Scaling Laws for Routed Language Models

ICML 2022oral

The performance of a language model has been shown to be effectively modeled as a power-law in its parameter count. Here we study the scaling behaviors of Routing Networks: architectures that conditionally use only a subset of their parameters while processing an input. For these models, parameter c…

2020

High Fidelity Speech Synthesis with Adversarial Networks

ICLR 2020talk

Generative adversarial networks have seen rapid development in recent years and have led to remarkable improvements in generative modelling of images. However, their application in the audio domain has received limited attention, and autoregressive models, such as WaveNet, remain the state of the ar…

Cited by 321SourcecodeScholar
2020

Stabilizing Transformers for Reinforcement Learning

ICML 2020poster

Owing to their ability to both effectively integrate information over long time horizons and scale to massive amounts of data, self-attention architectures have recently shown breakthrough success in natural language processing (NLP). Harnessing the transformer’s ability to process long time horizon…

2020

V-MPO: On-Policy Maximum a Posteriori Policy Optimization for Discrete and Continuous Control

ICLR 2020poster

Some of the most successful applications of deep reinforcement learning to challenging domains in discrete and continuous control have used policy gradient methods in the on-policy setting. However, policy gradients can suffer from large variance that may limit performance, and in practice require c…

Cited by 136SourceScholar