The Cost of Down-Scaling Language Models: Fact Recall Deteriorates before In-Context Learning
Jin, Tian, Clement, Nolan, Dong, Xin, Nagarajan, Vaishnavh, Carbin, Michael, Ragan-Kelley, Jonathan, Dziugaite, Gintare Karolina
–arXiv.org Artificial Intelligence
How does scaling the number of parameters in large language models (LLMs) affect their core capabilities? We study two natural scaling techniques -- weight pruning and simply training a smaller or larger model, which we refer to as dense scaling -- and their effects on two core capabilities of LLMs: (a) recalling facts presented during pre-training and (b) processing information presented in-context during inference. By curating a suite of tasks that help disentangle these two capabilities, we find a striking difference in how these two abilities evolve due to scaling. Reducing the model size by more than 30\% (via either scaling approach) significantly decreases the ability to recall facts seen in pre-training. Yet, a 60--70\% reduction largely preserves the various ways the model can process in-context information, ranging from retrieving answers from a long context to learning parameterized functions from in-context exemplars. The fact that both dense scaling and weight pruning exhibit this behavior suggests that scaling model size has an inherently disparate effect on fact recall and in-context learning.
arXiv.org Artificial Intelligence
Oct-6-2023
- Country:
- Africa > South Africa (0.04)
- Oceania
- New Zealand (0.04)
- Australia (0.04)
- North America
- Dominican Republic (0.04)
- United States
- Washington > King County
- Seattle (0.04)
- New York > New York County
- New York City (0.04)
- Minnesota > Hennepin County
- Minneapolis (0.14)
- Massachusetts > Middlesex County
- Cambridge (0.04)
- Washington > King County
- Canada
- Ontario > Toronto (0.04)
- British Columbia > Metro Vancouver Regional District
- Vancouver (0.04)
- Europe
- United Kingdom > Scotland (0.04)
- Norway (0.04)
- Germany (0.04)
- Asia
- Middle East > Jordan (0.04)
- Philippines (0.04)
- China > Tianjin Province
- Tianjin (0.04)
- Genre:
- Research Report > New Finding (1.00)
- Industry:
- Leisure & Entertainment (0.46)
- Technology: