English

Data coarse graining can improve model performance

Statistical Mechanics 2025-09-19 v1 Disordered Systems and Neural Networks Machine Learning Neurons and Cognition Machine Learning

Abstract

Lossy data transformations by definition lose information. Yet, in modern machine learning, methods like data pruning and lossy data augmentation can help improve generalization performance. We study this paradox using a solvable model of high-dimensional, ridge-regularized linear regression under 'data coarse graining.' Inspired by the renormalization group in statistical physics, we analyze coarse-graining schemes that systematically discard features based on their relevance to the learning task. Our results reveal a nonmonotonic dependence of the prediction risk on the degree of coarse graining. A 'high-pass' scheme--which filters out less relevant, lower-signal features--can help models generalize better. By contrast, a 'low-pass' scheme that integrates out more relevant, higher-signal features is purely detrimental. Crucially, using optimal regularization, we demonstrate that this nonmonotonicity is a distinct effect of data coarse graining and not an artifact of double descent. Our framework offers a clear, analytical explanation for why careful data augmentation works: it strips away less relevant degrees of freedom and isolates more predictive signals. Our results highlight a complex, nonmonotonic risk landscape shaped by the structure of the data, and illustrate how ideas from statistical physics provide a principled lens for understanding modern machine learning phenomena.

Keywords

Cite

@article{arxiv.2509.14498,
  title  = {Data coarse graining can improve model performance},
  author = {Alex Nguyen and David J. Schwab and Vudtiwat Ngampruetikorn},
  journal= {arXiv preprint arXiv:2509.14498},
  year   = {2025}
}

Comments

7 pages, 4 figures

R2 v1 2026-07-01T05:42:57.570Z