Guided Model Merging for Hybrid Data Learning: Leveraging Centralized Data to Refine Decentralized Models
Created by
Haebom
Category
Empty
저자
Junyi Zhu, Ruicong Yao, Taha Ceritli, Savas Ozkan, Matthew B. Blaschko, Eunchung Noh, Jeongwon Min, Cho Jung Min, Mete Ozay
개요
본 논문은 중앙 집중식 및 분산형 데이터 환경이 공존하는 하이브리드 데이터 환경에서의 모델 훈련을 위한 새로운 프레임워크를 제안합니다. 이 프레임워크는 분산형 모델로부터 모델 아틀라스를 구축하고, 중앙 집중식 데이터를 활용하여 이 구조화된 공간 내에서 글로벌 모델을 개선합니다. 개선된 모델은 다시 분산형 모델을 재초기화하는 데 사용됩니다. 제안된 방법은 연합 학습과 모델 병합을 결합하여 하이브리드 데이터 환경에서 효과적인 훈련을 가능하게 하며, 이론적으로 분산형 데이터만 사용하는 방법보다 더 빠른 수렴 속도를 달성합니다. 실험을 통해 중앙 집중식, 분산형, 기존 하이브리드 방식을 능가하는 성능을 보이며, 중앙 집중식 및 분산형 데이터 도메인이 다르거나 분산형 데이터에 노이즈가 있는 경우에도 강건함을 유지합니다.