Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Recyclable Tuning for Continual Pre-training

About

Continual pre-training is the paradigm where pre-trained language models (PLMs) continually acquire fresh knowledge from growing data and gradually get upgraded. Before an upgraded PLM is released, we may have tuned the original PLM for various tasks and stored the adapted weights. However, when tuning the upgraded PLM, these outdated adapted weights will typically be ignored and discarded, causing a potential waste of resources. We bring this issue to the forefront and contend that proper algorithms for recycling outdated adapted weights should be developed. To this end, we formulate the task of recyclable tuning for continual pre-training. In pilot studies, we find that after continual pre-training, the upgraded PLM remains compatible with the outdated adapted weights to some extent. Motivated by this finding, we analyze the connection between continually pre-trained PLMs from two novel aspects, i.e., mode connectivity, and functional similarity. Based on the corresponding findings, we propose both an initialization-based method and a distillation-based method for our task. We demonstrate their feasibility in improving the convergence and performance for tuning the upgraded PLM. We also show that both methods can be combined to achieve better performance. The source codes are publicly available at https://github.com/thunlp/RecyclableTuning.

Yujia Qin, Cheng Qian, Xu Han, Yankai Lin, Huadong Wang, Ruobing Xie, Zhiyuan Liu, Maosong Sun, Jie Zhou• 2023

Related benchmarks

TaskDatasetResultRank
Sentiment AnalysisIMDB (test)
Accuracy90.4
248
Sentiment AnalysisSST-2 (test)
Accuracy92.5
136
Natural Language InferenceMNLI--
80
Natural Language InferenceWNLI
Accuracy72.2
40
Natural Language InferenceMNLI (test)
Accuracy0.783
38
Sentiment AnalysisRotten Tomato
Accuracy87.8
25
Natural Language InferenceSICK
Accuracy75.2
15
Hate Speech DetectionTwitter Hate
Accuracy62.4
9
Relation ClassificationChemProt (test)
Score0.76
6
Stance DetectionFeminist
Accuracy64.6
5
Showing 10 of 13 rows

Other info

Code

Follow for update