1
Return

Bayesian Parameter-Efficient Fine-Tuning for Overcoming Catastrophic Forgetting

delete2024-01-01
delete0
delete
OA
AI
H
Haolin Chen *
P
Philip N. Garner
DOI:10.1109/TASLP.2024.3463395delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
We are motivated primarily by the adaptation of text-to-speech synthesis models; however we argue that more generic parameter-efficient fine-tuning (PEFT) is an appropriate framework to do such adaptation. Nevertheless, catastrophic forgetting remains an issue with PEFT, damaging the pre-trained model's inherent capabilities. We demonstrate that existing Bayesian learning techniques can be applied to PEFT to prevent catastrophic forgetting as long as the parameter shift of the fine-tuned layers can be calculated differentiably. In a principled series of experiments on language modeling and speech synthesis tasks, we utilize established Laplace approximations, including diagonal and Kronecker-factored approaches, to regularize PEFT with the low-rank adaptation (LoRA) and compare their performance in pre-training knowledge preservation. Our results demonstrate that catastrophic forgetting can be overcome by our methods without degrading the fine-tuning performance, and using the Kronecker-factored approximation produces a better preservation of the pre-training knowledge than the diagonal ones.
Keywords:
Adaptation models
Data models
Biological system modeling
Bayes methods
Tuning
Transfer learning
Training
Parameter-efficient fine-tuning
Bayesian transfer learning
Laplace approximation
catastrophic forgetting

Journal

I
IEEE-ACM Transactions on Audio Speech and Language Processing
IF:
5.1
Papers:
2.6K
Citations:
1.1W

Organization

S
swiss federal institutes of technology domain
Scholars:
9.0W
Papers: 8.0W
Citations: 163
Cited Papers

Cited Papers

Citing Papers

Citing Papers