Preprint
Machine Learning

Smol LM v2

November 1, 2024

0

Citations

0

Influential Citations

Venue

2024

Year

Abstract

A family of language models (135M, 360M, and 1.7B parameters), trained on 2T, 4T, and 11T tokens respectively from datasets including FineWeb-Edu, DCLM, The Stack, and curated math and coding datasets, with instruction-tuned versions created using Smol Talk dataset and DPO using UltraFeedback.

Analysis

Why This Paper Matters

Smol LM v2 challenges the prevailing trend of ever-larger language models by demonstrating that small architectures (135M to 1.7B parameters) can be trained on massive token budgets (2T to 11T tokens). This is significant because it suggests an alternative scaling path: instead of increasing model size, one can increase training data volume to improve performance. For practitioners, this could mean more accessible, cost-effective models that run on consumer hardware while still benefiting from extensive pretraining.

The paper also emphasizes data diversity, combining web text (FineWeb-Edu, DCLM) with code (The Stack) and curated math/coding datasets. This mirrors the data strategies used by much larger models, indicating that data quality and variety matter even for small models. The inclusion of instruction tuning via Smol Talk and DPO with UltraFeedback further aligns with modern alignment techniques, making these models potentially useful out of the box.

Technical Contributions

  • Scaling token count for small models: The paper systematically trains models of 135M, 360M, and 1.7B parameters on 2T, 4T, and 11T tokens respectively, far exceeding typical token budgets for such sizes.
  • Diverse data mixture: Uses FineWeb-Edu, DCLM, The Stack, and curated math/coding datasets, covering general knowledge, code, and reasoning.
  • Instruction tuning and DPO: Creates instruction-tuned versions using the Smol Talk dataset and applies Direct Preference Optimization (DPO) with UltraFeedback for alignment.
  • Open-source release: The models are presumably released to the community, enabling reproduction and further research.

Results

The abstract does not report any quantitative results, benchmark scores, or comparisons to other models. This is a major limitation for evaluating the paper's claims. Without metrics like perplexity, accuracy on downstream tasks, or human evaluation, it is impossible to assess whether the large token budgets actually improve performance relative to smaller budgets or other small models.

Significance

If the trained models perform competitively with larger models on a per-parameter basis, Smol LM v2 could influence how practitioners allocate compute resources. It suggests that investing in more data rather than larger models may be a viable strategy for many applications. The use of DPO for alignment also demonstrates that small models can benefit from modern alignment techniques. However, the lack of results in the abstract means the community must wait for full evaluation to confirm these promises.