InA: Inhibition Adaption on pre-trained language models
Β Fine-tuning pre-trained language models (LMs) may not always be the most practical approach for downstream tasks. While adaptation fine-tuning methods have shown promising results, a clearer explanation of their mechanisms and further inhibition of the transmission of information is needed. To address this, we propose an Inhibition Adaptation (InA) fine-tuning method that aims to reduce the number of added tunable weights and appropriately reweight knowledge derived from pre-trained LMs. The InA method involves (1) inserting a small trainable vector into each Transformer attention architecture and (2) setting a threshold to directly eliminate irrelevant knowledge. This approach draws inspiration from the shunting inhibition, which allows the inhibition of specific neurons to gate other functional neurons. With the inhibition mechanism, InA achieves competitive or even superior performance compared to other fine-tuning methods on π΅πΈπ πβπππππ, π ππ΅πΈπ ππβπππππ, and π·ππ΅πΈπ ππβπππππ for text classification and question-answering tasks.Β
Funding
Czech Technical University in Prague(grant number: SGS22/165/OHK3/3T/13), the Research Centre for Informatics (grant number: CZ.02.1.01/0.0/0.0/160_19/0000765), and the Brain Dynamics (grant number: CZ.02.01.01/00/22_008/0004643).
History
Author affiliation
College of Science & Engineering Comp' & Math' SciencesVersion
- AM (Accepted Manuscript)
Published in
Neural NetworksPagination
106410Publisher
Elsevier BVissn
0893-6080Copyright date
2024Available date
2025-05-25Publisher DOI
Language
enPublisher version
Deposited by
Professor Huiyu ZhouDeposit date
2024-05-28Data Access Statement
The data that has been used is confidential.Rights Retention Statement
- No