MLLGSep 24, 2021

Is the Number of Trainable Parameters All That Actually Matters?

arXiv:2109.11928v19 citations
Originality Synthesis-oriented
AI Analysis

This addresses the problem of efficiently scaling models for researchers and engineers, showing that incremental tricks to cheat scaling laws are ineffective.

The study investigated whether scaling laws for language models can be deceived by increasing effective parameters through methods like frozen random parameters or structured transforms, and found that test loss depends only on the actual number of trainable parameters, not spurious ones.

Recent work has identified simple empirical scaling laws for language models, linking compute budget, dataset size, model size, and autoregressive modeling loss. The validity of these simple power laws across orders of magnitude in model scale provides compelling evidence that larger models are also more capable models. However, scaling up models under the constraints of hardware and infrastructure is no easy feat, and rapidly becomes a hard and expensive engineering problem. We investigate ways to tentatively cheat scaling laws, and train larger models for cheaper. We emulate an increase in effective parameters, using efficient approximations: either by doping the models with frozen random parameters, or by using fast structured transforms in place of dense linear layers. We find that the scaling relationship between test loss and compute depends only on the actual number of trainable parameters; scaling laws cannot be deceived by spurious parameters.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes