Low-rank updates of pre-trained weights for multi-task learning - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Low-rank updates of pre-trained weights for multi-task learning

Résumé

Multi-Task Learning used with pre-trained models has been quite popular in the field of Natural Language Processing in recent years. This framework remains still challenging due to the complexity of the tasks and the challenges associated with fine-tuning large pre-trained models. In this paper, we propose a new approach for Multi-task learning which is based on stacking the weights of Neural Networks as a tensor. We show that low-rank updates in the canonical polyadic tensor decomposition of this tensor of weights lead to a simple, yet efficient algorithm, which without loss of performance allows to reduce considerably the model parameters. We investigate the interactions between tasks inside the model as well as the inclusion of sparsity to find the best tensor rank and to increase the compression rate. Our strategy is consistent with recent efforts that attempt to use constraints to fine-tune some model components. More precisely, we achieve equivalent performance as the state-of-the-art on the General Language Understanding Evaluation benchmark by training only 0.3 of the parameters per task while not modifying the baseline weights.

Dates et versions

hal-04360908 , version 1 (22-12-2023)

Identifiants

Citer

Alexandre Audibert, Massih Amini, Konstantin Usevich, Marianne Clausel. Low-rank updates of pre-trained weights for multi-task learning. Findings of the Association for Computational Linguistics, ACL 2023, Jul 2023, Toronto, Canada. pp.7544-7554, ⟨10.18653/v1/2023.findings-acl.476⟩. ⟨hal-04360908⟩
45 Consultations
0 Téléchargements

Altmetric

Partager

More