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

BA-LoRA: Bias-Alleviating Low-Rank Adaptation to Mitigate Catastrophic Inheritance in Large Language Models

About

Parameter-efficient fine-tuning (PEFT) has become a de facto standard for adapting large language models (LLMs). However, we identify a critical vulnerability within popular low-rank adaptation methods such as LoRA: they can exacerbate "Catastrophic Inheritance" - the unchecked propagation of biases, noise, and data imbalances from pre-training. This phenomenon can degrade model robustness and fairness, undermining the benefits of efficient adaptation. To address this, we introduce Bias-Alleviating Low-Rank Adaptation (BA-LoRA). Our approach is founded on a principled decomposition of Catastrophic Inheritance into three core challenges: Knowledge Drift, Representation Collapse, and Overfitting to Noise. BA-LoRA systematically mitigates these issues by incorporating a trio of targeted regularizers: consistency, diversity, and an SVD-based term, designed to preserve core knowledge, promote representational richness, and encourage robust, low-rank output representations, respectively. We conduct comprehensive evaluations on a suite of Natural Language Generation (NLG) and Natural Language Understanding (NLU) tasks using diverse, prominent open-source language models (e.g., LLaMA-2-7B and DeBERTa-v3-base). Our results show that BA-LoRA not only outperforms state-of-the-art LoRA variants in terms of performance and stability, but also demonstrates superior robustness and bias mitigation on targeted evaluations. These results provide evidence that BA-LoRA can counteract the adverse effects of Catastrophic Inheritance.

Yupeng Chang, Yi Chang, Yuan Wu• 2024

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval
Pass@123.58
1048
Natural Language UnderstandingGLUE (val)
SST-296.25
201
Mathematical ReasoningGSM8K (val)
Accuracy55.86
115
Mathematical ReasoningMATH (val)
Accuracy9.47
59
Code GenerationMBPP
Pass@1 Accuracy36.86
59
Conversational AbilityMT-Bench--
28
Showing 6 of 6 rows

Other info

Follow for update