IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v11y2023i20p4317-d1261404.html
   My bibliography  Save this article

Structure-Aware Low-Rank Adaptation for Parameter-Efficient Fine-Tuning

Author

Listed:
  • Yahao Hu

    (Command and Control Engineering College, Army Engineering University of PLA, Nanjing 210007, China)

  • Yifei Xie

    (Command and Control Engineering College, Army Engineering University of PLA, Nanjing 210007, China)

  • Tianfeng Wang

    (Command and Control Engineering College, Army Engineering University of PLA, Nanjing 210007, China)

  • Man Chen

    (Command and Control Engineering College, Army Engineering University of PLA, Nanjing 210007, China)

  • Zhisong Pan

    (Command and Control Engineering College, Army Engineering University of PLA, Nanjing 210007, China)

Abstract

With the growing scale of pre-trained language models (PLMs), full parameter fine-tuning becomes prohibitively expensive and practically infeasible. Therefore, parameter-efficient adaptation techniques for PLMs have been proposed to learn through incremental updates of pre-trained weights, such as in low-rank adaptation (LoRA). However, LoRA relies on heuristics to select the modules and layers to which it is applied, and assigns them the same rank. As a consequence, any fine-tuning that ignores the structural information between modules and layers is suboptimal. In this work, we propose structure-aware low-rank adaptation (SaLoRA), which adaptively learns the intrinsic rank of each incremental matrix by removing rank-0 components during training. We conduct comprehensive experiments using pre-trained models of different scales in both task-oriented (GLUE) and task-agnostic (Yelp and GYAFC) settings. The experimental results show that SaLoRA effectively captures the structure-aware intrinsic rank. Moreover, our method consistently outperforms LoRA without significantly compromising training efficiency.

Suggested Citation

  • Yahao Hu & Yifei Xie & Tianfeng Wang & Man Chen & Zhisong Pan, 2023. "Structure-Aware Low-Rank Adaptation for Parameter-Efficient Fine-Tuning," Mathematics, MDPI, vol. 11(20), pages 1-16, October.
  • Handle: RePEc:gam:jmathe:v:11:y:2023:i:20:p:4317-:d:1261404
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/20/4317/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/11/20/4317/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:11:y:2023:i:20:p:4317-:d:1261404. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.