Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Training-Free Model Merging for Multi-target Domain Adaptation

About

In this paper, we study multi-target domain adaptation of scene understanding models. While previous methods achieved commendable results through inter-domain consistency losses, they often assumed unrealistic simultaneous access to images from all target domains, overlooking constraints such as data transfer bandwidth limitations and data privacy concerns. Given these challenges, we pose the question: How to merge models adapted independently on distinct domains while bypassing the need for direct access to training data? Our solution to this problem involves two components, merging model parameters and merging model buffers (i.e., normalization layer statistics). For merging model parameters, empirical analyses of mode connectivity surprisingly reveal that linear merging suffices when employing the same pretrained backbone weights for adapting separate models. For merging model buffers, we model the real-world distribution with a Gaussian prior and estimate new statistics from the buffers of separately trained models. Our method is simple yet effective, achieving comparable performance with data combination training baselines, while eliminating the need for accessing training data. Project page: https://air-discover.github.io/ModelMerging

Wenyi Li, Huan-ang Gao, Mingju Gao, Beiwen Tian, Rong Zhi, Hao Zhao• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K A-150
mIoU34.92
188
Semantic segmentationPascal Context 59
mIoU60.82
164
Semantic segmentationBDD100K
mIoU54.78
78
Open Vocabulary Semantic SegmentationPascal Context PC-59
mIoU63.24
57
Open Vocabulary Semantic SegmentationADE20K A-150
mIoU37.25
54
Open Vocabulary Semantic SegmentationCityscapes (val)
mIoU62.18
37
Semantic segmentationNYU Depth V2
mIoU48.42
27
Semantic segmentationMapillary Vistas
mIoU30.37
22
Semantic segmentationACDC--
17
Semantic segmentationIDD
mIoU39.59
15
Showing 10 of 32 rows

Other info

Follow for update