Not All Directions Matter: Towards Structured and Task-Aware Low-Rank Model Adaptation
Organizations: University of Alabama at Birmingham · University of Virginia · Tulane University · Yale University ·
Abstract
Low-Rank Adaptation (LoRA) has become a cornerstone of parameter-efficient fine-tuning (PEFT). Yet, its efficacy is hampered by two fundamental limitations: semantic drift, by treating all update directions with equal importance, and structural incoherence, from adapting layers independently, resulting in suboptimal, uncoordinated updates. To remedy these, we propose StructLoRA, a framework that addresses both limitations through a principled, dual-component design: (1) an Information Bottleneck-guided filter that prunes task-irrelevant directions to mitigate semantic drift, and (2) a lightweight, training-only graph-based coordinator that enforces inter-layer consistency to resolve structural incoherence. Extensive experiments across large language model , vision language model, and vision model (including LLaMA, LLaVA, and ViT) demonstrate that StructLoRA consistently establishes a new state-of-the-art, outperforming not only vanilla LoRA but also advanced dynamic rank allocation and sparsity-based methods. Notably, the benefits are particularly pronounced in challenging low-rank and low-data regimes. Crucially, since our proposed modules operate only during training, StructLoRA enhances performance with zero additional inference cost, advancing the focus of PEFT -- from mere parameter compression to a more holistic optimization of information quality and structural integrity.
Figures & tables
| Method | Type | Reasoning | Image Classification | Captioning | Visual QA | ||
|---|---|---|---|---|---|---|---|
| BoolQ | PIQA | CIFAR-100 | ImageNet-1k | COCO Caption | VQAv2 | ||
| Full Fine-tuning | — | 82.6 | 85.3 | 85.9 | 78.8 | 123.5 | 76.2 |
| Linear Probing | — | 71.1 | 74.8 | 65.7 | 68.4 | 105.3 | 67.0 |
| Adapter Houlsby et al. (2019) | Additive | 79.2 | 82.5 | 81.3 | 75.7 | 116.5 | 73.0 |
| Prefix-Tuning Li and Liang (2021) | Additive | 79.6 | 82.8 | 81.7 | 76.1 | 117.0 | 73.3 |
| LoRA Hu et al. (2022a) | Reparam. | 79.1 | 82.4 | 81.5 | 76.2 | 116.2 | 73.5 |
| Method | MNLI | SST-2 | MRPC | CoLA | QNLI | QQP | RTE | STS-B | Average |
|---|---|---|---|---|---|---|---|---|---|
| LoRA | 87.3 | 93.5 | 87.1 | 58.8 | 93.0 | 90.5 | 79.4 | 91.0 | 85.1 |
| AdaLoRA | 87.3 | 93.6 | 87.3 | 59.0 | 93.1 | 90.6 | 79.6 | 91.2 | 85.2 |
| DyLoRA | 87.2 | 93.7 | 87.3 | 59.0 | 93.0 | 90.6 | 79.6 | 91.2 | 85.2 |
| Sensitivity-LoRA | 87.6 | 94.6 | 87.7 | 60.2 | 93.6 | 90.7 | 81.8 | 91.3 | 86.0 |
| StructLoRA (ours) | 88.1 | 95.0 | 88.5 | 61.5 | 94.1 | 91.0 | 82.3 | 91.5 | 86.5 |
| Rank ( ) | Trainable Params (%) | BoolQ (LLaMA-7B) | CIFAR-100 (ViT-B/16) | COCO Caption (LLaVA-1.5-7B) | |||
|---|---|---|---|---|---|---|---|
| LoRA | StructLoRA | LoRA | StructLoRA | LoRA | StructLoRA | ||
| 2 | 0.12 | 75.1 | 77.4 (+2.3) | 78.3 | 80.1 (+1.8) | 111.2 | 114.3 (+3.1) |
| 4 | 0.24 | 77.6 | 79.9 (+2.3) | 79.7 | 82.2 (+2.5) | 113.8 | 117.0 (+3.2) |
| 8 | 0.48 | 79.1 | 81.3 (+2.2) | 81.5 | 84.1 (+2.6) | 116.2 | 122.4 (+6.2) |
| 16 | 0.95 | 80.3 | 81.7 (+1.4) | 82.8 | 84.3 (+1.5) | 118.1 | 123.6 (+5.5) |
| 32 | 1.90 | 81.0 | 81.9 (+0.9) | 83.4 | 84.5 (+1.1) | 119.0 | 123.9 (+4.9) |
| Dataset | Metric | Method | 10% Data | 25% Data | 50% Data | 100% Data |
|---|---|---|---|---|---|---|
| BoolQ (LLaMA-7B) | Accuracy | LoRA | 68.5 | 73.2 | 76.4 | 79.1 |
| StructLoRA | 71.2 (+2.7) | 76.3 (+3.1) | 78.9 (+2.5) | 81.3 (+2.2) | ||
| CIFAR-100 (ViT-B/16) | Accuracy | LoRA | 73.6 | 78.0 | 80.5 | 81.5 |
| StructLoRA | 76.3 (+2.7) | 80.5 (+2.5) | 82.4 (+1.9) | 84.1 (+2.6) | ||
| COCO Caption (LLaVA-1.5-7B) | CIDEr | LoRA | 100.2 | 108.3 | 114.0 | 116.2 |
| StructLoRA | 103.7 (+3.5) | 112.4 (+4.1) | 117.9 (+3.9) | 122.4 (+6.2) |
| Ablation Setting | BoolQ (LLaMA-7B) | CIFAR-100 (ViT-B/16) | COCO Caption (LLaVA-1.5-7B) | |||
|---|---|---|---|---|---|---|
| Accuracy (%) | Accuracy (%) | CIDEr | ||||
| StructLoRA (Full) | 81.3 | - 0.0 | 84.1 | - 0.0 | 122.4 | - 0.0 |
| w/o IB Filter | 79.4 | -1.9 | 81.9 | -2.2 | 117.8 | -4.6 |
| w/o GNN Coordination | 80.1 | -1.2 | 82.6 | -1.5 | 119.4 | -3.0 |
| w/o Both (i.e., Standard LoRA) | 79.1 | -2.2 | 81.5 | -2.6 | 116.2 | -6.2 |
| GNN Configuration | Accuracy (%) |
|---|---|
| StructLoRA | 81.3 |
| Effect of Depth: | |
| 2-Layer GNN | 80.4 |
| 3-Layer GNN | 79.7 |
| Effect of Graph Construction: | |
| Adjacency Only | 80.5 |
| Method | Training Time | Peak Memory |
|---|---|---|
| / Epoch | (GB) | |
| LoRA | 1.00 | 16.8 |
| StructLoRA | 1.06 | 17.5 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Method | Magpie-Pro | OpenPlatypus | Avg. | ||||
|---|---|---|---|---|---|---|---|---|
| BLEU-4 | ROUGE-1 | ROUGE-L | BLEU-4 | ROUGE-1 | ROUGE-L | |||
| Qwen2.5-7B | HAdapter | 54.71 | 49.11 | 32.42 | 19.39 | 43.95 | 22.51 | 37.01 |
| PAdapter | 54.83 | 49.15 | 32.24 | 19.42 | 44.03 | 22.54 | 37.04 | |
| LoRA | 55.03 | 48.82 | 32.42 | 19.72 | 43.83 | 22.53 | 37.06 | |
| AdaLoRA | 55.66 | 49.13 | 32.75 | 19.87 | 44.24 | 22.67 | 37.39 | |
| DyLoRA | 55.59 | 49.21 | 32.82 | 19.86 | 44.18 | 22.59 | 37.37 | |
| Method | Task Score (↑) | (↓) | CosAdj (↑) | Extra Cost (train, %) |
|---|---|---|---|---|
| LoRA+Cos | 82.3 / 79.4 | 3.98 / 4.22 | 0.46 / 0.42 | +1.2 |
| LoRA+Lap | 83.0 / 80.1 | 3.72 / 3.95 | 0.49 / 0.45 | +2.4 |
| StructLoRA (GNN) | 84.1 / 81.3 | 3.05 / 3.41 | 0.56 / 0.52 | +5.3 |
| Task | Model | Context Length | Method | Performance | Metric | Trainable Params (%) | Peak Memory (GB) |
|---|---|---|---|---|---|---|---|
| BoolQ Reasoning | LLaMA-13B | 512 tokens | LoRA | 80.3 | Accuracy | 0.45 | 18.1 |
| StructLoRA | 81.7 | Accuracy | 0.45 | 18.8 | |||
| COCO Captioning | LLaVA-1.5-7B | 1024 tokens | LoRA | 110.6 | CIDEr | 0.48 | 20.3 |
| StructLoRA | 114.9 | CIDEr | 0.48 | 20.7 |
| Task Domain | Model | Base PEFT Method | Enhancement Method | Metric | Base Performance | Enhanced Performance |
|---|---|---|---|---|---|---|
| Language | LLaMA-7B | QLoRA Dettmers et al. (2023) | StructLoRA | Accuracy (%) | 80.0 | 81.6 (+1.6) |
| LLaMA2-7B | VeRA Kopiczko et al. (2024) | StructLoRA | Accuracy (%) | 82.6 | 83.9 (+1.3) | |
| Vision | ViT-B/16 | LoRA-FA Drost (2024) | StructLoRA | Accuracy (%) | 81.8 | 83.6 (+1.8) |
| Multimodal | LLaVA-1.5-7B | AdapterFusion Pfeiffer et al. (2021) | StructLoRA | CIDEr | 118.3 | 121.0 (+2.7) |
| Model | Method | Trainable Params (M) | Ratio (%) | Quantization | Activation Precision | Notes / Equivalent Setting |
|---|---|---|---|---|---|---|
| LLaMA-7B | LoRA | 67.2 | 0.96 | None | FP16 | Rank , |
| QLoRA | 66.9 | 0.95 | 4-bit NF4 | FP16 | Same rank, quantized base weights | |
| DoRA | 67.2 | 0.96 | None | FP16 | Direction-only adaptation | |
| AdaLoRA | 67.4 | 0.97 | None | FP16 | Dynamic rank allocation (avg ) | |
| Adapter | 68.0 | 0.98 | None | FP16 | Bottleneck dim 32 (matches LoRA params) | |
| Prefix-Tuning | 67.6 | 0.97 | None | FP16 | Prefix length 20 (equivalent budget) |
| Model / Layer Pair | LoRA | StructLoRA | Change ( ) |
|---|---|---|---|
| Layers 1–2 | 0.31 | 0.57 | +0.26 |
| Layers 2–3 | 0.27 | 0.55 | +0.28 |
| Layers 3–4 | 0.34 | 0.63 | +0.29 |
| Layers 4–5 | 0.38 | 0.66 | +0.28 |
| Layers 5–6 | 0.41 | 0.69 | +0.28 |
| Layers 6–7 | 0.36 | 0.64 | +0.28 |
Low-Rank Adaptation
Low-rank adaptation (LoRA) fine-tunes pretrained models with compact weight updates, reducing trainable parameters and deployment costs.
See all 542 papers