Label-Free Cross-Task LoRA Merging with Null-Space Compression
Organizations: KAIST · *Equal contribution to this work.
Abstract
Model merging combines independently fine-tuned checkpoints without joint multi-task training. In the era of foundation-model, fine-tuning with Low-Rank Adaptation (LoRA) is prevalent, making LoRA merging a promising target. Existing approaches can work in homogeneous settings where all target tasks are classification but often fail when tasks span classification and regression. Approaches using entropy-based surrogates do not apply to regression and are costly for large language models due to long token sequences. We introduce Null-Space Compression (NSC) Merging, a label-free, output-agnostic method that sets merge weights from adapter geometry. Our key observation is that during LoRA finetuning the down-projection factor in compresses its null space, and the compression correlates with performance. NSC uses this as an optimization signal for merging that can generalize across classification, regression, and sequence generation. NSC achieves state-of-the-art performance across twenty heterogeneous vision tasks with balanced gains where prior methods overfit subsets of tasks. It also outperforms baselines on six NLI benchmarks and on vision-language evaluations for VQA and image captioning, demonstrating scalability and effectiveness.
Figures & tables
| NYUD-v2 (4) | PASCAL-Context (5) | Taskonomy (11) | Avg | ||||||||||||||||||
| Method | Depth | Semseg | Normal | Edge | Semseg | Parts | Saliency | Normal | Edge | DE | DZ | EO | ET | K2 | K3 | N | C | R | S2 | S2.5 | |
| RMSE | mIoU | Mean | L1 | mIoU | mIoU | mIoU | Mean | L1 | L1 | L1 | L1 | L1 | L1 | L1 | L1 | RMSE | L1 | L1 | L1 | ||
| Finetuned performance | |||||||||||||||||||||
| Finetuned | 0.657 | 37.66 | 25.98 | 0.051 | 70.07 | 54.77 | 80.00 | 18.36 | 0.046 | 0.016 | 0.016 | 0.101 | 0.171 | 0.162 | 0.082 | 0.217 | 0.710 | 0.136 | 0.170 | 0.144 | – |
| Merged models, normalized to finetuned (%) | |||||||||||||||||||||
| TA [ 30 ] | 24.0 | 1.3 | 54.9 | 105.2 | 4.7 | 20.6 | 36.4 | 62.5 | 104.3 | 100.1 | 99.8 | 101.7 | 104.6 | 106.4 | 102.2 | 100.2 | 103.6 | 106.2 | 107.4 | 98.8 | 77.2 |
| Method | MNLI | QNLI | SNLI | RTE | SICK | SciTail | Avg |
|---|---|---|---|---|---|---|---|
| Per-task absolute accuracy (%) | |||||||
| Finetuned | 90.8 | 94.9 | 91.8 | 87.0 | 90.9 | 94.8 | 91.7 |
| Merged models (normalized to finetuned baselines, %) | |||||||
| TA [ 30 ] | 92.8 | 86.8 | 93.3 | 93.6 | 83.8 | 95.0 | 90.9 |
| TIES [ 90 ] | 94.3 | 88.8 | 90.8 | 89.8 | 86.6 | 94.4 | 90.8 |
| DARE-TIES [ 98 ] | 94.3 | 88.8 | 90.8 | 89.8 | 86.6 | 94.4 | 90.8 |
| Method | IconQA | VizWiz | ChartQA | DocVQA | COCO | Flickr30k | Avg |
|---|---|---|---|---|---|---|---|
| Per-task absolute score | |||||||
| Zero-Shot | 17.9 | 55.2 | 18.2 | 24.3 | 109.5 | 79.2 | – |
| Finetuned | 67.8 | 69.3 | 39.0 | 40.7 | 130.5 | 91.3 | – |
| Avg. generated tokens | 2.1 | 2.8 | 4.7 | 6.7 | 12.5 | 14.2 | 7.17 |
| Merged models (normalized to fine-tuned baselines, %) | |||||||
| TA [ 30 ] | 56.8 | 83.2 | 73.0 | 85.2 | 92.9 | 97.6 | 81.4 |
| Seen | Avg (Seen) | Unseen | Avg (Unseen) | Avg | |||||||||||||||||||
| NYUD-v2 | PASCAL-Context | Taskonomy | NYUD-v2 | PASCAL-Context | Taskonomy | ||||||||||||||||||
| Method | Semseg | Edge | Parts | Sal | DZ | ET | K3 | N | R | S2.5 | Depth | Normal | Semseg | Normal | Edge | DE | EO | K2 | C | S2 | |||
| mIoU | L1 | mIoU | mIoU | L1 | L1 | L1 | L1 | L1 | L1 | RMSE | Mean | mIoU | Mean | L1 | L1 | L1 | L1 | RMSE | L1 | ||||
| Finetuned performance (absolute) | |||||||||||||||||||||||
| Finetuned | 37.66 | 0.051 | 54.77 | 80.00 | 0.0160 | 0.1713 | 0.0820 | 0.2169 | 0.1357 | 0.1435 | – | 0.657 | 25.98 | 70.07 | 18.36 | 0.046 | 0.016 | 0.101 | 0.162 | 0.710 | 0.170 | – | – |
| Merged models, normalized to finetuned (%) | |||||||||||||||||||||||
| Targeted Projection Matrix | Number of Activated Transformer Blocks | |||
|---|---|---|---|---|
| 12 | 6 | 3 | 1 | |
| QKVO | 83.7 | 84.4 | 83.9 | 83.0 |
| KVO | 84.0 | 84.6 | 84.0 | 83.0 |
| VO | 84.3 | 84.7 | 84.1 | 82.6 |
| O | 84.5 | 84.9 | 84.6 | 82.6 |
| Method | Requirements | Prep. | Opt. | Val. | Total | GPU Mem. | ||
|---|---|---|---|---|---|---|---|---|
| Prep. | Opt. | Val. | (min) | (min) | (min) | (min) | (GB) | |
| Gradient-Free | ||||||||
| TA [ 30 ] | ✗ | ✗ | ✓ | – | – | 16.6 | 16.6 | 14.4 |
| TIES [ 90 ] | ✓ | ✗ | ✓ | 4.5 | – | 16.6 | 20.1 | 14.4 |
| KnOTS-TIES [ 71 ] | ✓ | ✗ | ✓ | 6.8 | – | 16.6 | 23.4 | 14.4 |
| Gradient-Based | ||||||||
| Dataset | Provided Prompt |
|---|---|
| IconQA , VizWiz | Answer the question using a single word. |
| ChartQA , DocVQA | Answer the question using a single word or short phrase. |
| COCO , Flickr30K | Provide a one-sentence caption for the provided image. |
| Token Position Strategy | Dataset | Performance | ||||||||
| First-Token | Last-Token | Full-Sequence | IconQA | VizWiz | ChartQA | DocVQA | COCO | Flickr | Avg | Time (min) |
| ✓ | 59.5 | 82.8 | 78.0 | 87.0 | 91.5 | 97.3 | 82.7 | 83.7 | ||
| ✓ | 60.0 | 82.9 | 77.7 | 87.1 | 91.5 | 97.2 | 82.8 | 65.1 | ||
| ✓ | 59.7 | 82.9 | 78.1 | 87.1 | 91.7 | 96.8 | 82.7 | 13.3 | ||
| Input Modality | Dataset | Performance | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Text | Image | IconQA | VizWiz | ChartQA | DocVQA | COCO | Flickr | Avg | Time (min) |
| ✓ | 56.6 | 83.8 | 74.3 | 83.9 | 93.3 | 97.7 | 81.6 | 6.0 | |
| ✓ | 60.2 | 82.4 | 78.4 | 87.0 | 91.4 | 97.0 | 82.7 | 13.1 | |
| ✓ | ✓ | 59.7 | 82.9 | 78.1 | 87.1 | 91.7 | 96.8 | 82.7 | 13.3 |
| Process Step | AdaMerging | NSC (Ours) |
|---|---|---|
| Pre-computation (Gram-Inverse) | ✗ | 0.7 |
| Forward Pass | 396.7 | 407.2 |
| Entropy/NSC Calculation | 1.1 | 4.1 |
| Backpropagation + Optimizer Step | 393.4 | 386.7 |
| Targeted Projection Matrix | Number of Activated Transformer Blocks | |||
|---|---|---|---|---|
| 12 | 6 | 3 | 1 | |
| QKVO | 92.0 | 92.0 | 92.0 | 91.9 |
| KVO | 91.6 | 91.6 | 91.6 | 91.4 |
| VO | 92.0 | 92.0 | 92.0 | 91.6 |
| O | 91.7 | 91.7 | 91.8 | 91.7 |
| Benchmark | Method | Avg. Norm. Acc. (mean std) |
|---|---|---|
| LLM Tasks | AdaMerging | 90.08 0.44 |
| NSC (Ours) | 92.25 0.08 |
| # Samples per Task | 1 | 10 | 100 | Full Data |
|---|---|---|---|---|
| Avg. Norm. Acc. | 91.88 0.35 | 92.19 0.16 | 92.25 0.12 | 92.25 0.08 |
| Method | Dataset | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Avg | |
| Finetuned | 70.7 | 67.6 | 98.5 | 97.6 | 99.4 | 91.6 | 72.0 | 95.7 | 86.6 |
| TA [ 30 ] | 87.8 | 78.6 | 62.9 | 70.5 | 92.2 | 78.8 | 92.4 | 85.9 | 81.1 |
| TIES [ 90 ] | 90.6 | 80.4 | 78.8 | 66.1 | 91.6 | 81.2 | 92.5 | 82.5 | 82.9 |
| AdaMerging [ 95 ] | 90.1 | 76.8 | 87.8 | 86.0 | 94.7 | 85.6 | 90.8 | 82.8 | 86.8 |
| NSC (Ours) | 89.4 | 78.7 | 80.3 | 74.4 | 93.4 | 82.6 | 92.1 | 86.0 | 84.6 |
Model Merging
Model merging combines trained model weights or adapters to reuse capabilities while reducing the cost of joint training or ensembles.
See all 192 papers