SyMerge: From Non-Interference to Synergistic Merging via Single-Layer Adaptation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Jung, Aecheon, Lee, Seunghwan, Han, Dongyoon, Hong, Sungeun
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913155858300928
author Jung, Aecheon
Lee, Seunghwan
Han, Dongyoon
Hong, Sungeun
author_facet Jung, Aecheon
Lee, Seunghwan
Han, Dongyoon
Hong, Sungeun
contents Model merging combines independently trained models into a single multi-task model. However, most existing approaches focus primarily on avoiding task interference. We argue that its greater potential lies in enabling task synergy, where tasks actively improve one another. We identify cross-task performance, defined by compatibility between encoders and predictors across tasks, as a key indicator of merge quality. We demonstrate that adapting only a single task-specific layer is sufficient to induce such synergy. This study proposes SyMerge, a lightweight framework that jointly optimizes merging coefficients and a single task-specific layer. We adopt an expert-guided self-labeling objective, providing stable supervision beyond entropy minimization. Intriguingly, we further show that SyMerge successfully merges models trained from different initializations, a regime where standard methods break down. Our minimalist yet principled method achieves state-of-the-art results across vision, dense prediction, and NLP benchmarks. Our code is available at https://aim-skku.github.io/SyMerge
format Preprint
id arxiv_https___arxiv_org_abs_2412_19098
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SyMerge: From Non-Interference to Synergistic Merging via Single-Layer Adaptation
Jung, Aecheon
Lee, Seunghwan
Han, Dongyoon
Hong, Sungeun
Machine Learning
Model merging combines independently trained models into a single multi-task model. However, most existing approaches focus primarily on avoiding task interference. We argue that its greater potential lies in enabling task synergy, where tasks actively improve one another. We identify cross-task performance, defined by compatibility between encoders and predictors across tasks, as a key indicator of merge quality. We demonstrate that adapting only a single task-specific layer is sufficient to induce such synergy. This study proposes SyMerge, a lightweight framework that jointly optimizes merging coefficients and a single task-specific layer. We adopt an expert-guided self-labeling objective, providing stable supervision beyond entropy minimization. Intriguingly, we further show that SyMerge successfully merges models trained from different initializations, a regime where standard methods break down. Our minimalist yet principled method achieves state-of-the-art results across vision, dense prediction, and NLP benchmarks. Our code is available at https://aim-skku.github.io/SyMerge
title SyMerge: From Non-Interference to Synergistic Merging via Single-Layer Adaptation
topic Machine Learning
url https://arxiv.org/abs/2412.19098