Domain-Adaptive Model Merging Across Disconnected Modes
Organizations: Tongji University · Peking University · Southeast University · Nanchang University
Abstract
Learning across domains is challenging when data cannot be centralized due to privacy or heterogeneity, which limits the ability to train a single comprehensive model. Model merging provides an appealing alternative by consolidating knowledge from multiple specialized models into one, avoiding data sharing and reducing retraining cost. In this work, we present DMM, a data-free model merging framework designed to handle highly divergent models. DMM proceeds in three steps. First, domain-specific models are trained independently. Second, models with high similarity are merged using standard techniques to ensure stability. Third, we synthesize pseudo-data from normalization statistics and distill knowledge from divergent models into the merged model through a lightweight refinement guided by these samples. This approach preserves rare but critical knowledge while maintaining stability. Extensive experiments on unimodal and multimodal benchmarks show that DMM achieves state-of-the-art performance over existing merging methods.
Figures & tables
| Method | CIFAR-10 | CIFAR-100 | CrisisMMD | ||||||
|---|---|---|---|---|---|---|---|---|---|
| FedAvg [ 12 ] | 78.34 | 56.32 | 36.76 | 65.55 | 59.30 | 48.72 | 60.93 | 45.21 | 22.50 |
| FedProx [ 6 ] | 79.45 | 56.73 | 37.32 | 65.54 | 59.64 | 49.73 | 62.88 | 46.84 | 21.32 |
| FedBN [ 7 ] | 79.52 | 57.68 | 38.35 | 66.92 | 59.16 | 50.70 | 60.51 | 44.79 | 23.72 |
| FedAvg+DMM | 82.30 | 65.85 | 53.66 | 68.68 | 64.31 | 53.04 | 64.33 | 49.70 | 30.46 |
| PLeaS [ 13 ] | 60.51 | 45.77 | 33.30 | 49.74 | 44.97 | 39.56 | 52.18 | 40.11 | 23.67 |
| Method | CIFAR-10 | CIFAR-100 |
|---|---|---|
| Cat-Merge (baseline) | 47.67 | 45.51 |
| + Buffer Aggregation | 48.11 | 45.63 |
| + Inversion Augmentation | 53.32 | 47.16 |
| + DMM (full) | 55.86 | 48.42 |
| Method | CIFAR-10 | CIFAR-100 |
|---|---|---|
| FedAvg (baseline) | 36.76 | 48.72 |
| + Buffer Aggregation | 38.98 | 49.43 |
| + Inversion Augmentation | 50.38 | 51.10 |
| + DMM (full) | 53.66 | 53.04 |
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