DC-Merge: Improving Model Merging with Directional Consistency
Organizations: School of Computer Science and Engineering, Southeast University, Nanjing 210096, China
Abstract
Model merging aims to integrate multiple task-adapted models into a unified model that preserves the knowledge of each task. In this paper, we identify that the key to this knowledge retention lies in maintaining the directional consistency of singular spaces between merged multi-task vector and individual task vectors. However, this consistency is frequently compromised by two issues: i) an imbalanced energy distribution within task vectors, where a small fraction of singular values dominate the total energy, leading to the neglect of semantically important but weaker components upon merging, and ii) the geometric inconsistency of task vectors in parameter space, which causes direct merging to distort their underlying directional geometry. To address these challenges, we propose DC-Merge, a method for directional-consistent model merging. It first balances the energy distribution of each task vector by smoothing its singular values, ensuring all knowledge components are adequately represented. These energy-balanced vectors are then projected onto a shared orthogonal subspace to align their directional geometries with minimal reconstruction error. Finally, the aligned vectors are aggregated in the shared orthogonal subspace and projected back to the original parameter space. Extensive experiments on vision and vision-language benchmarks show that DC-Merge consistently achieves state-of-the-art performance in both full fine-tuning and LoRA settings. The implementation code is available at https://github.com/Tobeginwith/DC-Merge.
Figures & tables
| Method | ViT-B-32 | ViT-B-16 | ViT-L-14 | ||||||
| 8 tasks | 12 tasks | 16 tasks | 8 tasks | 12 tasks | 16 tasks | 8 tasks | 12 tasks | 16 tasks | |
| Individual | |||||||||
| Task Arithmetic | |||||||||
| KnOTS-TIES | |||||||||
| WUDI-Merging | |||||||||
| TSV-M | |||||||||
| Method | ViT-B-32 | ViT-B-16 | ViT-L-14 | ||||||
| 8 tasks | 14 tasks | 20 tasks | 8 tasks | 14 tasks | 20 tasks | 8 tasks | 14 tasks | 20 tasks | |
| Individual | |||||||||
| Weight Averaging | |||||||||
| Task Arithmetic | |||||||||
| TIES-Merging | |||||||||
| Consensus TA | |||||||||
| Method | Seen Tasks | Unseen Tasks |
| Zeroshot | ||
| Individual | ||
| Multi Task | ||
| Task Arithmetic | ||
| DARE-Merging | ||
| TIES-Merging |
| Method | 8 tasks | 12 tasks | 16 tasks |
| No smoothing | 69.27 | 74.60 | 74.47 |
| Averaging | 73.09 ( +3.82 ) | 76.42 ( +1.82 ) | 75.51 ( +1.04 ) |
| Linear smoothing | 73.90 ( +4.63 ) | 77.22 ( +2.62 ) | 75.80 ( +1.33 ) |
| Method | Tasks | w/o mask | w/ mask |
| FFT | 8 tasks | 87.98 | 93.55 ( +5.57 ) |
| 14 tasks | 82.39 | 90.50 ( +8.11 ) | |
| 20 tasks | 77.63 | 88.18 ( +10.55 ) | |
| LoRA | 8 tasks | 73.61 | 73.90 ( +0.29 ) |
| 12 tasks | 75.94 | 77.22 ( +1.28 ) | |
| 16 tasks | 74.42 | 75.80 ( +1.38 ) |
| Method | 8 tasks | 12 tasks | 16 tasks |
| Vanilla TA | 61.73 | 69.12 | 68.94 |
| TA + ES | 69.12 ( +7.39 ) | 74.35 ( +5.23 ) | 73.01 ( +4.07 ) |
| TA + CSM | 68.13 ( +6.40 ) | 73.92 ( +4.80 ) | 72.64 ( +3.70 ) |
| TA + CSM + ES | 73.82 ( +12.09 ) | 77.16 ( +8.04 ) | 75.73 ( +6.79 ) |
| Vanilla TIES | 62.09 | 69.30 | 70.06 |
| TIES + ES | 69.94 ( +7.85 ) | 74.97 ( +5.67 ) | 74.74 ( +4.68 ) |
Appendix figures & tables22 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Definition |
| The weights of pretrained model | |
| The weights of tasks fine-tuned model | |
| The task vector of task | |
| Knowledge decomposition of | |
| -th knowledge vector of | |
| -th knowledge component of |
| Method | ViT-B-16 | ViT-L-14 | LLaVA-v1.5-7B | ||||
| 8 tasks | 12 tasks | 16 tasks | 8 tasks | 12 tasks | 16 tasks | 8 tasks | |
| No smoothing | 71.64 | 76.13 | 75.48 | 83.96 | 85.63 | 80.86 | 79.10 |
| Linear smoothing | 78.69 ( +7.05 ) | 80.35 ( +4.22 ) | 78.82 ( +3.34 ) | 89.22 ( +5.26 ) | 89.11 ( +3.48 ) | 85.38 ( +4.52 ) | 87.58 ( +8.48 ) |
| Averaging | 78.86 ( +7.22 ) | 80.56 ( +4.43 ) | 78.91 ( +3.43 ) | 89.42 ( +5.46 ) | 89.31 ( +3.68 ) | 85.71 ( +4.85 ) | 87.91 ( +8.81 ) |
| Method | Datasets | ||||||||
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Average | |
| Task Arithmetic | |||||||||
| KnOTS-TIES | |||||||||
| WUDI-Merging | |||||||||
| TSV-M | |||||||||
| Iso-CTS | |||||||||
| Method | ViT-B-32 | ViT-B-16 | ViT-L-14 |
| Zeroshot | |||
| Individual | |||
| Task Arithmetic | |||
| Fisher Merging | |||
| RegMean | |||
| PCB-Merging |
| Method | Seen Tasks | Unseen Tasks | ||||||||||||
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Average | CIFAR100 | Flowers | Pets | STL10 | Average | |
| Individual | ||||||||||||||
| Zeroshot | ||||||||||||||
| Task Arithmetic | ||||||||||||||
| KnOTS-TIES | ||||||||||||||
| WUDI-Merging | ||||||||||||||
| Method | Seen Tasks | Unseen Tasks | ||||||||||||
| Cars | DTD | EuroSAT | GTSRB | MNIST | RESISC45 | SUN397 | SVHN | Average | CIFAR100 | Flowers | Pets | STL10 | Average | |
| Individual | ||||||||||||||
| Zeroshot | ||||||||||||||
| Task Arithmetic | ||||||||||||||
| KnOTS-TIES | ||||||||||||||
| WUDI-Merging | ||||||||||||||
| Method | Seen Tasks | Unseen Tasks | ||||||||||||
| SciQA | Image | VQA | REC | OCR | VizWiz | Flickr | IconQA | Average | AVQA | Image-R | S2W | TabMWP | Average | |
| Individual | ||||||||||||||
| Zeroshot | ||||||||||||||
| Multi Task | ||||||||||||||
| Task Arithmetic | ||||||||||||||
| DARE-Merging | ||||||||||||||
| Method | 8 tasks | 14 tasks | 20 tasks |
| Zeroshot | |||
| Individual | |||
| TSV-M | |||
| WUDI-Merging | |||
| Iso-CTS | |||
| DC-Merge |
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