Training-Free Model Merging for Multi-target Domain Adaptation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Wenyi, Gao, Huan-ang, Gao, Mingju, Tian, Beiwen, Zhi, Rong, Zhao, Hao
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913435635154944
author Li, Wenyi
Gao, Huan-ang
Gao, Mingju
Tian, Beiwen
Zhi, Rong
Zhao, Hao
author_facet Li, Wenyi
Gao, Huan-ang
Gao, Mingju
Tian, Beiwen
Zhi, Rong
Zhao, Hao
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2407_13771
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training-Free Model Merging for Multi-target Domain Adaptation
Li, Wenyi
Gao, Huan-ang
Gao, Mingju
Tian, Beiwen
Zhi, Rong
Zhao, Hao
Computer Vision and Pattern Recognition
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
title Training-Free Model Merging for Multi-target Domain Adaptation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.13771