pith. sign in

arxiv: 2406.00605 · v1 · pith:VWA7FD7Anew · submitted 2024-06-02 · 💻 cs.CL · cs.AI

LongSkywork: A Training Recipe for Efficiently Extending Context Length in Large Language Models

classification 💻 cs.CL cs.AI
keywords long-contextdatamodelcontextlanguagelongskyworkmethodsperformance
0
0 comments X
read the original abstract

We introduce LongSkywork, a long-context Large Language Model (LLM) capable of processing up to 200,000 tokens. We provide a training recipe for efficiently extending context length of LLMs. We identify that the critical element in enhancing long-context processing capability is to incorporate a long-context SFT stage following the standard SFT stage. A mere 200 iterations can convert the standard SFT model into a long-context model. To reduce the effort in collecting and annotating data for long-context language modeling, we develop two novel methods for creating synthetic data. These methods are applied during the continual pretraining phase as well as the Supervised Fine-Tuning (SFT) phase, greatly enhancing the training efficiency of our long-context LLMs. Our findings suggest that synthetic long-context SFT data can surpass the performance of data curated by humans to some extent. LongSkywork achieves outstanding performance on a variety of long-context benchmarks. In the Needle test, a benchmark for long-context information retrieval, our models achieved perfect accuracy across multiple context spans. Moreover, in realistic application scenarios, LongSkywork-13B demonstrates performance on par with Claude2.1, the leading long-context model, underscoring the effectiveness of our proposed methods.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 2 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Optimization Hyper-parameter Laws for Large Language Models

    cs.LG 2024-09 unverdicted novelty 6.0

    Opt-Laws predicts LLM final training loss from LR schedules via SDE-derived convergence and escape features, with 94% Top-2 hit rate on held-out schedules and F1=0.92 for divergence detection.

  2. Randomized YaRN Improves Length Generalization for Long-Context Reasoning

    cs.CL 2026-06 unverdicted novelty 5.0

    Randomized YaRN improves LLM reasoning performance on 16K-128K contexts when trained only on <8K data by randomizing YaRN positional encodings during short-context training.