InA: Inhibition Adaption on pre-trained language models.

Cheng Kang, Jindrich Prokop, Lei Tong, Huiyu Zhou, Yong Hu, Daniel Novak
Author Information
  1. Cheng Kang: Department of Cybernetics, Czech Technical University in Prague, Czech Republic. Electronic address: kangchen@fel.cvut.cz.
  2. Jindrich Prokop: Department of Cybernetics, Czech Technical University in Prague, Czech Republic. Electronic address: prokojin@fel.cvut.cz.
  3. Lei Tong: School of Informatics, University of Leicester, UK. Electronic address: lt228@leicester.ac.uk.
  4. Huiyu Zhou: School of Informatics, University of Leicester, UK. Electronic address: hz143@leicester.ac.uk.
  5. Yong Hu: Li Ka Shing Faculty of Medicine, University of Hong Kong, Hong Kong. Electronic address: yhud@hku.hk.
  6. Daniel Novak: Department of Cybernetics, Czech Technical University in Prague, Czech Republic. Electronic address: xnovakd1@fel.cvut.cz.

Abstract

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 BERT-large, RoBERTa-large, and DeBERTa-large for text classification and question-answering tasks.

Keywords

MeSH Term

Language
Humans
Neural Networks, Computer
Neural Inhibition
Neurons

Word Cloud

Created with Highcharts 10.0.0inhibitionfine-tuningpre-trainedlanguagemodelsInhibitionInALMsapproachtasksmethodsmethodknowledgeneuronsAdaptionFine-tuningmayalwayspracticaldownstreamadaptationshownpromisingresultsclearerexplanationmechanismstransmissioninformationneededaddressproposeAdaptationaimsreducenumberaddedtunableweightsappropriatelyreweightderivedinvolves1insertingsmalltrainablevectorTransformerattentionarchitecture2settingthresholddirectlyeliminateirrelevantdrawsinspirationshuntingallowsspecificgatefunctionalmechanismachievescompetitiveevensuperiorperformancecomparedBERT-largeRoBERTa-largeDeBERTa-largetextclassificationquestion-answeringInA:Efficient-parameterPre-trainedShunting

Similar Articles

Cited By