Abstract
We present Junk DNA Hypothesis by adopting a novel task-centric angle for the pre-trained weights of large language models (LLMs). It has been believed that weights in LLMs contain significant redundancy, leading to the conception that a considerable chunk of the parameters can be removed by pruning without compromising performance. Contrary to this belief, this paper presents a counter-argument: small-magnitude weights of pre-trained model weights encode vital knowledge essential for tackling difficult downstream tasks - manifested as the monotonic relationship between the performance drop of downstream tasks across the difficulty spectrum, as we prune more pre-trained weights by magnitude. Moreover, we reveal that these seemingly inconsequential weights can result in irreparable loss of knowledge and performance degradation in difficult tasks, even when downstream continual training is allowed. Interestingly, our evaluations show that the other popular compression, namely quantization fail to exhibit similar “monotonic" effect and does not as convincingly disentangle this task-difficulty information. To study formally, we introduce several quantifiable metrics to gauge the downstream task difficulty: (a) within the same task category, and (b) across different task categories. Our extensive experiments substantiate the Junk DNA Hypothesis across a diverse range of model sizes, tasks, datasets, and even pruning methods. Codes are available at https://github.com/VITA-Group/Junk_DNA_Hypothesis.git.
| Original language | English |
|---|---|
| Title of host publication | Proceedings of the 41st International Conference on Machine Learning |
| Editors | Ruslan Salakhutdinov, Ziko Kolter, Katherine Heller, Adrian Weller, Nuria Oliver, Jonathan Scarlett, Felix Berkenkamp |
| Publisher | PMLR |
| Pages | 57053-57068 |
| Number of pages | 16 |
| Publication status | Published - 2024 |
| Event | 41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria Duration: 21 Jul 2024 → 27 Jul 2024 |
Publication series
| Name | Proceedings of Machine Learning Research (PMLR) |
|---|---|
| Volume | 235 |
| ISSN (Electronic) | 2640-3498 |
Conference
| Conference | 41st International Conference on Machine Learning, ICML 2024 |
|---|---|
| Abbreviated title | ICML 2024 |
| Country/Territory | Austria |
| City | Vienna |
| Period | 21/07/24 → 27/07/24 |
Funding
Z. Wang is in part supported by NSF Award DMS-02133861 and the NSF AI Institute for Foundations of Machine Learning (IFML).
Keywords
- Pruning
- LLMs
- Quantization