Work

FRAME: Boosting LLMs with A Four-Quadrant Multi-Stage Pretraining Strategy

Year: 2025
Type: conference-paper
Abstract: Large language models (LLMs) have significantly advanced human language understanding and generation, with pretraining data quality and organization being crucial to their performance.Multi-stage pretraining is a promising approach, but existing methods often lack quantitative criteria for data partitioning and instead rely on intuitive heuristics.In this paper, we propose the novel Four-quadRAnt Multistage prEtraining strategy (FRAME), guided by the established principle of organizing the pretraining process into four stages to achieve significant loss reductions four times.This principle is grounded in two key findings: first, training on high Perplexity (PPL) data followed by low PPL data, and second, training on low PPL difference (PD) data followed by high PD data, both causing the loss to drop significantly twice and performance enhancements.By partitioning data into four quadrants and strategically organizing them, FRAME achieves a remarkable 16.8% average improvement over random across MMLU and CMMLU for the 3B model, effectively boosting LLM performance.PPL and A high PPL .The data within each subset is uniformly distributed.We conduct two-stage training on the 3B model in the sequences A low PPL → A high PPL and A high PPL → A low PPL , and compare the results with
Language: English
Open Access status: gold
OpenAlex ID: W4412887757
FWCI:
Cites:
Cited by:
Related to: 10