Raise a Child in Large Language Model: Towards Effective and Generalizable Fine-tuning
About
Recent pretrained language models extend from millions to billions of parameters. Thus the need to fine-tune an extremely large pretrained model with a limited training corpus arises in various downstream tasks. In this paper, we propose a straightforward yet effective fine-tuning technique, Child-Tuning, which updates a subset of parameters (called child network) of large pretrained models via strategically masking out the gradients of the non-child network during the backward process. Experiments on various downstream tasks in GLUE benchmark show that Child-Tuning consistently outperforms the vanilla fine-tuning by 1.5~8.6 average score among four different pretrained models, and surpasses the prior fine-tuning techniques by 0.6~1.3 points. Furthermore, empirical results on domain transfer and task transfer show that Child-Tuning can obtain better generalization performance by large margins.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Natural Language Inference | RTE | Accuracy70.87 | 590 | |
| Natural Language Inference | SNLI | Accuracy84.41 | 218 | |
| Natural Language Understanding | GLUE (val) | -- | 201 | |
| Natural Language Inference | MNLI (matched) | Accuracy79.13 | 110 | |
| Natural Language Inference | SICK | Accuracy55.69 | 85 | |
| Natural Language Inference | MNLI | -- | 80 | |
| Regional Precipitation Forecasting (36 Hours lead time) | ERA5-CH regional forecast 0.25 degree | SEEPS0.672 | 32 | |
| Regional Precipitation Forecasting (12 Hours lead time) | ERA5-CH regional forecast 0.25 degree | SEEPS0.407 | 32 | |
| Regional Precipitation Forecasting (24 Hours lead time) | ERA5-CH regional forecast 0.25 degree | SEEPS56.5 | 32 | |
| Question Answering | SQuAD (val) | F1 Score88.5 | 26 |