Is More Data Worth the Cost? Dataset Scaling Laws in a Tiny Attention-Only Decoder

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wiegand, Götz-Henrik, Raichle, Lorena, Städeli, Rico, Hrycej, Tomas, Bermeitinger, Bernhard, Handschuh, Siegfried
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915930262470656
author Wiegand, Götz-Henrik
Raichle, Lorena
Städeli, Rico
Hrycej, Tomas
Bermeitinger, Bernhard
Handschuh, Siegfried
author_facet Wiegand, Götz-Henrik
Raichle, Lorena
Städeli, Rico
Hrycej, Tomas
Bermeitinger, Bernhard
Handschuh, Siegfried
contents Training Transformer language models is expensive, as performance typically improves with increasing dataset size and computational budget. Although scaling laws describe this trend at large scale, their implications in controlled, smaller-scale settings remain less explored. In this work, we isolate dataset-size effects using a strongly reduced attention-only decoder architecture. By training on progressively larger power-of-two subsets, we observe smooth performance improvements accompanied by clear diminishing returns, consistent with scaling-law behavior. Using only about 30% of the training data is sufficient to reach approximately 90% of the full-data validation token-level accuracy. These results provide actionable insights into dataset scaling in a controlled, component-isolated setting and offer practical guidance for balancing dataset size and computational cost in compute- and data-restricted environments, such as small research labs and exploratory model development.
format Preprint
id arxiv_https___arxiv_org_abs_2604_09389
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Is More Data Worth the Cost? Dataset Scaling Laws in a Tiny Attention-Only Decoder
Wiegand, Götz-Henrik
Raichle, Lorena
Städeli, Rico
Hrycej, Tomas
Bermeitinger, Bernhard
Handschuh, Siegfried
Machine Learning
Computation and Language
Training Transformer language models is expensive, as performance typically improves with increasing dataset size and computational budget. Although scaling laws describe this trend at large scale, their implications in controlled, smaller-scale settings remain less explored. In this work, we isolate dataset-size effects using a strongly reduced attention-only decoder architecture. By training on progressively larger power-of-two subsets, we observe smooth performance improvements accompanied by clear diminishing returns, consistent with scaling-law behavior. Using only about 30% of the training data is sufficient to reach approximately 90% of the full-data validation token-level accuracy. These results provide actionable insights into dataset scaling in a controlled, component-isolated setting and offer practical guidance for balancing dataset size and computational cost in compute- and data-restricted environments, such as small research labs and exploratory model development.
title Is More Data Worth the Cost? Dataset Scaling Laws in a Tiny Attention-Only Decoder
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2604.09389