arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2402.12220eess.AScs.LG

用于克服灾难性遗忘的贝叶斯参数高效微调

Bayesian Parameter-Efficient Fine-Tuning for Overcoming Catastrophic Forgetting

Idiap Research Institute(伊迪亚普研究所);École Polytechnique Fédérale de Lausanne(洛桑联邦理工学院)

Haolin Chen, Philip N. Garner

AI总结:

针对PEFT存在的灾难性遗忘问题,提出将贝叶斯学习技术结合拉普拉斯近似应用于带LoRA的PEFT,实验证明该方法可在不降低微调性能的前提下克服遗忘,且克罗内克因子分解近似的知识保留效果更优。

摘要:

我们的研究主要受文本到语音合成模型适配的驱动,但我们认为更通用的参数高效微调(PEFT,Parameter-Efficient Fine-Tuning)是开展此类适配的合适框架。尽管如此,灾难性遗忘仍是PEFT存在的一个问题,会损害预训练模型的固有能力。我们证明,只要微调层的参数偏移可以可微计算,现有的贝叶斯学习技术就可以应用于PEFT以防止灾难性遗忘。在针对语言建模和语音合成任务的一系列原则性实验中,我们利用已有的拉普拉斯近似方法(包括对角近似和克罗内克因子分解近似方法),对采用低秩适配(LoRA,Low-Rank Adaptation)的PEFT进行正则化,并比较了它们在预训练知识保留方面的性能。我们的结果表明,所提方法可以克服灾难性遗忘,且不会降低微调性能,并且使用克罗内克因子分解近似比对角近似能更好地保留预训练知识。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑