Unified Scaling Laws for Compressed Representations

Dan Alistarh (IST Austria & NeuralMagic) · Vage Egiazarian (Higher School of Economics) · Andrei Panferov (Institute of Science and Technology Austria) · Alexandra Volkova (Institute of Science and Technology Austria) · Ionut-Vlad Modoranu (Institute of Science and Technology Austria (ISTA)) · Mher Safaryan (IST Austria)
accuracy potentialcompressed representationscomputational costempirical validationgaussian mean squared errorinferencelarge-scale trainingmodel compressionparameter efficiencyquantizationscaling frameworkscaling lawssparse-quantized formatssparsificationtraining algorithmsunified scaling laws

Scaling laws have shaped recent advances in machine learning by enabling predictable scaling of model performance based on model size, computation, and data volume. Concurrently, the rise in computational cost for AI has motivated model compression techniques, notably quantization and sparsification, which have emerged to mitigate the steep computational demands associated with large-scale training and inference. This paper investigates the interplay between scaling laws and compression strategies, exploring whether a unified scaling framework can accurately predict model performance when training occurs over various compressed representations, such as sparse, scalar-quantized, sparse-quantized or even vector-quantized formats. Our key contributions include proposing and validating a general scaling law formulation applicable both individually but also composably across compression types. We demonstrate both theoretically and empirically that a simple metric based on Gaussian mean squared error fitting can robustly predict parameter efficiency across compressed models. Additionally, we extend our formulation to directly compare the accuracy potential of different compressed formats, and to derive better algorithms for training over sparse-quantized formats. Finally, we identify conditions under which these unified scaling laws fail.