LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Huang, Chengsong, Liu, Qian, Lin, Bill Yuchen, Pang, Tianyu, Du, Chao, Lin, Min
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909289564602368
author Huang, Chengsong
Liu, Qian
Lin, Bill Yuchen
Pang, Tianyu
Du, Chao
Lin, Min
author_facet Huang, Chengsong
Liu, Qian
Lin, Bill Yuchen
Pang, Tianyu
Du, Chao
Lin, Min
contents Low-rank adaptations (LoRA) are often employed to fine-tune large language models (LLMs) for new tasks. This paper investigates LoRA composability for cross-task generalization and introduces LoraHub, a simple framework devised for the purposive assembly of LoRA modules trained on diverse given tasks, with the objective of achieving adaptable performance on unseen tasks. With just a few examples from a new task, LoraHub can fluidly combine multiple LoRA modules, eliminating the need for human expertise and assumptions. Notably, the composition requires neither additional model parameters nor gradients. Empirical results on the Big-Bench Hard benchmark suggest that LoraHub, while not surpassing the performance of in-context learning, offers a notable performance-efficiency trade-off in few-shot scenarios by employing a significantly reduced number of tokens per example during inference. Notably, LoraHub establishes a better upper bound compared to in-context learning when paired with different demonstration examples, demonstrating its potential for future development. Our vision is to establish a platform for LoRA modules, empowering users to share their trained LoRA modules. This collaborative approach facilitates the seamless application of LoRA modules to novel tasks, contributing to an adaptive ecosystem. Our code is available at https://github.com/sail-sg/lorahub, and all the pre-trained LoRA modules are released at https://huggingface.co/lorahub.
format Preprint
id arxiv_https___arxiv_org_abs_2307_13269
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition
Huang, Chengsong
Liu, Qian
Lin, Bill Yuchen
Pang, Tianyu
Du, Chao
Lin, Min
Computation and Language
Artificial Intelligence
Low-rank adaptations (LoRA) are often employed to fine-tune large language models (LLMs) for new tasks. This paper investigates LoRA composability for cross-task generalization and introduces LoraHub, a simple framework devised for the purposive assembly of LoRA modules trained on diverse given tasks, with the objective of achieving adaptable performance on unseen tasks. With just a few examples from a new task, LoraHub can fluidly combine multiple LoRA modules, eliminating the need for human expertise and assumptions. Notably, the composition requires neither additional model parameters nor gradients. Empirical results on the Big-Bench Hard benchmark suggest that LoraHub, while not surpassing the performance of in-context learning, offers a notable performance-efficiency trade-off in few-shot scenarios by employing a significantly reduced number of tokens per example during inference. Notably, LoraHub establishes a better upper bound compared to in-context learning when paired with different demonstration examples, demonstrating its potential for future development. Our vision is to establish a platform for LoRA modules, empowering users to share their trained LoRA modules. This collaborative approach facilitates the seamless application of LoRA modules to novel tasks, contributing to an adaptive ecosystem. Our code is available at https://github.com/sail-sg/lorahub, and all the pre-trained LoRA modules are released at https://huggingface.co/lorahub.
title LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2307.13269