Preference-Aligned LoRA Merging: Preserving Subspace Coverage and Addressing Directional Anisotropy
Organizations: KAIST
Abstract
Merging multiple Low-Rank Adaptation (LoRA) modules is promising for constructing general-purpose systems, yet challenging because LoRA update directions span different subspaces and contribute unevenly. When merged naively, such mismatches can weaken the directions most critical to certain task losses while overemphasizing relatively less important ones, ultimately reducing the model's ability to represent all tasks faithfully. We revisit this problem through two perspectives: subspace coverage, which captures how broadly LoRA directions cover diverse representational directions, and anisotropy, which reflects the imbalance of influence across those directions. We propose TARA-Merging (Task-Rank Anisotropy Alignment), which aligns merging weights using a preference-weighted cross-entropy pseudo-loss while preserving task-relevant LoRA subspaces. This ensures broad subspace coverage and mitigates anisotropy via direction-wise reweighting. Across eight vision and six NLI benchmarks, TARA-Merging consistently outperforms vanilla and LoRA-aware baselines, demonstrating strong robustness and generalization, and highlighting the importance of addressing both subspace coverage and anisotropy in LoRA merging.
Figures & tables
| Method | Dataset | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Avg | |
| Per‑task absolute accuracies (%) | |||||||||
| Finetuned | 74.0 | 58.3 | 99.0 | 92.7 | 99.3 | 88.4 | 64.5 | 96.2 | 84.1 |
| Per‑task accuracies of merged models, normalized to finetuned (%) | |||||||||
| Vanilla Merging | |||||||||
| RegMean [ 40 ] | 80.2 | 71.3 | 37.9 | 47.3 | 43.1 | 70.5 | 93.9 | 43.0 | 60.9 |
| Method | Dataset | ||||||
|---|---|---|---|---|---|---|---|
| MNLI | QNLI | SNLI | RTE | SICK | SCITAIL | Avg | |
| Per-task absolute accuracies (%) | |||||||
| Finetuned | 90.8 | 95.3 | 92.1 | 84.8 | 91.3 | 87.3 | 90.2 |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||
| Vanilla Merging | |||||||
| TA [ 32 ] | 67.3 | 87.3 | 41.8 | 95.7 | 77.9 | 76.9 | 74.6 |
| Method | Hits@1 | Hits@3 | Hits@5 |
|---|---|---|---|
| TA [ 32 ] | 43.5 | 65.2 | 74.0 |
| TIES [ 99 ] | 43.6 | 65.3 | 73.9 |
| DARE-TIES [ 107 ] | 44.0 | 66.4 | 75.1 |
| AdaMerging [ 105 ] | 48.1 | 73.2 | 83.0 |
| KnOTS-TIES [ 79 ] | 46.8 | 68.1 | 76.3 |
| KnOTS-DARE-TIES [ 79 ] | 45.2 | 66.9 | 75.3 |
| Seen Tasks | Unseen Tasks | All Tasks | |||||||||
| Method | Cars | DTD | GTSRB | RESISC45 | SUN397 | SVHN | Avg Acc | EuroSAT | MNIST | Avg Acc | Avg Acc |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||||||
| TA [ 32 ] | 82.0 | 74.7 | 42.2 | 71.1 | 96.8 | 42.2 | 68.2 | 37.5 | 48.2 | 42.9 | 61.8 |
| TIES [ 99 ] | 81.5 | 73.2 | 40.3 | 64.4 | 95.4 | 45.7 | 66.7 | 37.2 | 61.8 | 49.5 | 62.4 |
| KnOTS-TIES [ 79 ] | 82.7 | 74.4 | 48.5 | 72.5 | 95.3 | 50.2 | 70.6 | 33.2 | 50.5 | 41.8 | 63.4 |
| KnOTS-DARE-TIES [ 79 ] | 81.7 | 72.5 | 49.4 | 71.9 | 94.6 | 52.5 | 70.4 | 31.6 | 51.2 | 41.4 | 63.2 |
| Method | Dataset | ||||||
|---|---|---|---|---|---|---|---|
| ScienceQA | VizWiz | IconQA | ImageNet | ChartQA | DocVQA | Avg | |
| Per-task absolute accuracies (%) | |||||||
| Finetuned | 86.6 | 69.7 | 65.2 | 96.0 | 39.3 | 41.2 | 66.4 |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||
| TA [ 32 ] | 81.6 | 71.3 | 57.2 | 43.6 | 66.6 | 79.5 | 66.6 |
| TIES [ 99 ] | 90.9 | 69.8 | 66.5 | 54.8 | 79.1 | 81.9 | 73.8 |
| LLaMA-3-8B | LLaVA-1.5-7B | |||
|---|---|---|---|---|
| Method | Time (min) | VRAM (MiB) | Time (min) | VRAM (MiB) |
| TA [ 32 ] | 3.6 | 16,150 | 10.6 | 14,814 |
| TIES [ 99 ] | 8.1 (4.5+3.6) | 16,150 | 18.3 (7.7+10.6) | 14,814 |
| KnOTS-TIES [ 79 ] | 10.8 (7.2+3.6) | 16,150 | 24.3 (13.7+10.6) | 14,814 |
| AdaMerging [ 105 ] | 8.0 | 19,918 | 13.9 | 19,788 |
| TARA -Variant A | 8.2 | 19,922 | 14.1 | 19,788 |
| Method | Time (min) | VRAM (MiB) |
|---|---|---|
| KnOTS-TIES [ 79 ] | 2.8 | 2262 |
| LoRA-LEGO [ 114 ] | 4.0 | 1416 |
| AdaMerging [ 105 ] | 4.1 | 4344 |
| TARA (Variant A) | 4.1 | 4344 |
| TARA (Variant B) | 5.1 | 5922 |
| Method | Dataset | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Avg | |
| Per‑task absolute accuracies (%) | |||||||||
| Finetuned | 76.2 | 72.5 | 98.6 | 98.3 | 99.2 | 93.1 | 73.7 | 96.7 | 88.5 |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||||
| Vanilla Merging | |||||||||
| TA [ 32 ] | 85.6 | 70.8 | 81.9 | 89.1 | 97.9 | 81.9 | 89.1 | 91.9 | 86.0 |
| Method | Dataset | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Avg | |
| Per‑task absolute accuracies (%) | |||||||||
| Finetuned (KnOTS) | 74.0 | 58.3 | 99.0 | 92.7 | 99.3 | 88.4 | 64.5 | 96.2 | 84.1 |
| Finetuned (Ours) | 76.2 | 72.5 | 98.6 | 98.3 | 99.2 | 93.1 | 73.7 | 96.7 | 88.5 |
| Per‑task accuracies of merged models, normalized to finetuned (%) | |||||||||
| TA [ 32 ] (KnOTS) | 82.0 | 73.6 | 48.8 | 42.1 | 53.1 | 71.5 | 97.5 | 41.2 | 63.7 |
| Seen Tasks | Unseen Tasks | All Tasks | |||||||||
| Method | Cars | DTD | GTSRB | RESISC45 | SUN397 | SVHN | Avg Acc | EuroSAT | MNIST | Avg Acc | Avg Acc |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||||||
| TA [ 32 ] | 85.6 | 73.0 | 94.7 | 82.4 | 88.4 | 95.0 | 86.5 | 47.3 | 78.1 | 62.7 | 80.6 |
| TIES [ 99 ] | 76.5 | 62.5 | 71.4 | 73.3 | 87.5 | 75.0 | 74.4 | 49.0 | 64.5 | 56.8 | 69.9 |
| KnOTS-TIES [ 79 ] | 85.0 | 69.4 | 83.9 | 81.1 | 89.4 | 84.8 | 82.3 | 56.6 | 73.4 | 65.0 | 78.0 |
| KnOTS-DARE-TIES [ 79 ] | 85.7 | 69.3 | 85.0 | 81.4 | 90.3 | 86.4 | 83.0 | 56.7 | 74.0 | 65.4 | 78.6 |
| Method | Dataset | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Avg | |
| Per-task absolute accuracies (%) | |||||||||
| Finetuned | 72.1 | 70.2 | 98.4 | 98.1 | 99.2 | 93.3 | 73.4 | 96.7 | 87.7 |
| Per-task accuracies of merged models, normalized to finetuned (%) | |||||||||
| Vanilla Merging | |||||||||
| TA [ 32 ] | 87.4 | 76.2 | 74.0 | 78.1 | 98.1 | 83.7 | 89.7 | 92.3 | 84.9 |
| Method | r=4 | r=16 | r=64 | r=256 | Full |
|---|---|---|---|---|---|
| TA [ 32 ] | 71.9(82.9) | 76.2(86.2) | 77.9(86.2) | 77.1(85.3) | 72.0(80.8) |
| Adamerging [ 105 ] | 76.3(87.2) | 78.7(88.5) | 80.2(88.7) | 79.8(88.6) | 79.1(88.2) |
| TARA | 77.5(88.8) | 80.5(90.6) | 81.9(90.5) | 80.9(89.7) | 80.5(89.5) |
| Method | Adamerging [ 105 ] | CALM [ 100 ] | FW-Merging [ 5 ] | TARA-variant B |
|---|---|---|---|---|
| Avg. Acc. | 78.7 | 80.0 | 77.9 | 80.5 |
| Time (min) | 4.1 | 20.0 | 13.4 | 5.1 |
| VRAM (MiB) | 4344 | 8066 | 7542 | 5922 |
| Backbone (# Tasks) | TA [ 32 ] | AdaMer. [ 105 ] | CALM [ 100 ] | FW-Mer. [ 5 ] | TARA |
|---|---|---|---|---|---|
| ViT-B/32 (8) | 76.2(86.2) | 78.7(88.5) | 80.0(90.6) | 77.9(87.6) | 80.5(90.6) |
| ViT-B/32 (14) | 73.6(83.6) | 75.0(84.8) | 74.7(84.6) | 75.4(85.2) | 75.4(85.2) |
| ViT-B/32 (20) | 64.6(72.7) | 66.6(74.8) | 66.9(75.2) | 68.2(77.2) | 68.7(77.2) |
| ViT-L/14 (8) | 88.0(95.8) | 88.9(96.7) | 83.0(90.3) | 87.0(94.7) | 88.9(96.7) |
| ViT-L/14 (14) | 85.0(92.7) | 85.2(93.0) | 80.5(87.7) | 85.7(93.6) | 85.9(93.7) |
| ViT-L/14 (20) | 80.7(87.2) | 81.3(87.8) | 78.5(84.5) | 77.4(84.1) | 81.5(88.0) |
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