Effective Long-Context Scaling of Foundation Models
Authors: name: Wenhan Xiong; name: Jingyu Liu; name: Igor Molybog; name: Hejia Zhang; name: Prajjwal Bhargava; name: Rui Hou; name: Louis Martin; name: Rashi Rungta; name: Karthik Abinav Sankararaman; name: Barlas Oguz; name: Madian Khabsa; name: Han Fang; name: Yashar Mehdad; name: Sharan Narang; name: Kshitiz Malik; name: Angela Fan; name: Shruti Bhosale; name: Sergey Edunov; name: Mike Lewis; name: Sinong Wang; name: Hao Ma
Published: 2023-09-27 21:41:49

Abstract: We present a series of long-context LLMs that support effective context
windows of up to 32,768 tokens. Our model series are built through continual
pretraining from Llama 2 with longer training sequences and on a dataset where
long texts are upsampled. We perform extensive evaluation on language modeling,
synthetic context probing tasks, and a wide range of research benchmarks. On
research benchmarks, our models achieve consistent improvements on most regular
tasks and significant improvements on long-context tasks over Llama 2. Notably,
with a cost-effective instruction tuning procedure that does not require
human-annotated long instruction data, the 70B variant can already surpass
gpt-3.5-turbo-16k's overall performance on a suite of long-context tasks.
Alongside these results, we provide an in-depth analysis on the individual
components of our method. We delve into Llama's position encodings and discuss
its limitation in modeling long dependencies. We also examine the impact of
various design choices in the pretraining process, including the data mix and
the training curriculum of sequence lengths -- our ablation experiments suggest
that having abundant long texts in the pretrain dataset is not the key to
achieving strong performance, and we empirically verify that long context
continual pretraining is more efficient and similarly effective compared to
pretraining from scratch with long sequences.