University of Leicester
Browse

InA: Inhibition Adaption on pre-trained language models

journal contribution
posted on 2024-06-03, 15:23 authored by Cheng Kang, Jindrich Prokop, Lei Tong, Huiyu Zhou, Yong Hu, Daniel Novak

Β 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' Sciences

Version

  • AM (Accepted Manuscript)

Published in

Neural Networks

Pagination

106410

Publisher

Elsevier BV

issn

0893-6080

Copyright date

2024

Available date

2025-05-25

Language

en

Deposited by

Professor Huiyu Zhou

Deposit date

2024-05-28

Data Access Statement

The data that has been used is confidential.

Rights Retention Statement

  • No

Usage metrics

    University of Leicester Publications

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC