Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

OLoRA: Orthonormal Low-Rank Adaptation of Large Language Models

About

The advent of large language models (LLMs) has revolutionized natural language processing, enabling unprecedented capabilities in understanding and generating human-like text. However, the computational cost and convergence times associated with fine-tuning these models remain significant challenges. Low-Rank Adaptation (LoRA) has emerged as a promising method to mitigate these issues by introducing efficient fine-tuning techniques with a reduced number of trainable parameters. In this paper, we present OLoRA, an enhancement to the LoRA method that leverages orthonormal matrix initialization through QR decomposition. OLoRA significantly accelerates the convergence of LLM training while preserving the efficiency benefits of LoRA, such as the number of trainable parameters and GPU memory footprint. Our empirical evaluations demonstrate that OLoRA not only converges faster but also exhibits improved performance compared to standard LoRA across a variety of language modeling tasks. This advancement opens new avenues for more efficient and accessible fine-tuning of LLMs, potentially enabling broader adoption and innovation in natural language applications.

Kerim B\"uy\"ukaky\"uz• 2024

Related benchmarks

TaskDatasetResultRank
Math ReasoningGSM8K
Accuracy52.38
254
Commonsense ReasoningCommonsense Reasoning (BoolQ, PIQA, SIQA, HellaS., WinoG., ARC-e, ARC-c, OBQA)
BoolQ Accuracy71.34
223
Math ReasoningMATH
Accuracy8.22
160
Multi-turn dialogueMT-Bench
MT-Bench Score6.13
126
ChatMT-Bench
MT-Bench Score4.99
73
Sentiment AnalysisCR
CA93.16
54
Commonsense ReasoningCommonsense Reasoning Benchmark
BoolQ Accuracy74.41
22
Language ModelingWikitext-2 raw v1
Loss2.241
10
Natural Language UnderstandingGLUE
MRPC Score87.99
10
Mathematical ReasoningGSM8K (test)
Loss0.503
10
Showing 10 of 16 rows

Other info

Follow for update