Superpose Task-specific Features for Model Merging
Created by
Haebom
저자
Haiquan Qiu, You Wu, Dong Li, Jianmin Guo, Quanming Yao
개요
본 논문은 추가적인 훈련 없이 신경망의 강력한 기능을 가능하게 하는 모델 병합에 대한 새로운 관점을 제시합니다. 선형 표현 가설(linear representation hypothesis) 즉, 신경망이 특징 벡터의 선형 결합을 통해 정보를 인코딩한다는 가정을 바탕으로, 개별 모델의 작업별 특징을 병합된 모델에 중첩하는 방법을 제안합니다. 특히 심층 신경망에서 특징 활성화 및 추출에 중요한 선형 변환 행렬을 타겟으로 하여, 모델 병합 과정을 선형 시스템으로 공식화함으로써 개별 모델의 작업별 특징을 보존하고 기존 방법보다 효과적으로 다중 작업 기능을 유지하는 병합 모델을 생성합니다. 다양한 벤치마크와 모델에 대한 광범위한 실험을 통해 제안된 방법이 기존 기술보다 우수함을 보여줍니다. 코드는 https://github.com/LARS-research/STF 에서 확인할 수 있습니다.