FedAttnPSO: Non-IID 환경에서 연합학습 정확도 향상을 위한 개선된 알고리즘 제안

Abstract

연합학습은 데이터를 공유하지 않고 분산된 환경에서 학습을 수행할 수 있도록 설계된 방법으로, 개인정보보호가 중요한 현대사회에서 실용성을 높이평가받고있다. 하지만 각클라이언트가 보유한데이터의양과분포및특성이 모두상이한Non-IID 환경에서 모델 성능이 저하되는 한계가 존재한다. 이 문제를 해결하기 위해 알맞은 클라이언트 선택과 가중치 집계문제를해결해야한다. 본논문에서는이문제들을해결하기위해FedAttnPSO를 제안한다. FedAttnPSO는 각 클라이언트의 손실값을 기준으로 성능이 우수한 상위 N개의 클라이언트를 선택한다. 선택된 클라이언트들 중Attention Mechanism을 통해 성능이 높은 클라이언트의 기여도를 높이고 성능이 낮은 클라이언트의 영향력을 제한하여 동적으로 집계 가중치를 부여한다. 또한 클라이언트 및 서버 측에서 PSO (Particle Swarm Optimization) 알고리즘을 활용한 학습 최적화를 통해 전체적인 모델의 성능을 향상시킨다. FedAttnPSO와다른 연합학습 알고리즘을 비교하기 위해 표준 이미지 데이터셋인 CIFAR-10, CIFAR-100을 사용하여 비교하였으며, Non-IID 환경에서의 실험을 위해 FEMNIST를 사용하여 FedAttnPSO를 평가 및 비교했다. 실험 결과, 모든 데이터셋에서 기존알고리즘대비성능이향상됨을확인하였다.

Keywords

Federated Learning, Particle Swarm Optimization, Attention Mechanism, Non-IID

Citation

장선영, 김종욱, 최미정. “FedAttnPSO: Non-IID 환경에서 연합학습 정확도 향상을 위한 개선된 알고리즘 제안.” KNOM Review, Vol. 28, No. 1, pp. 10–20, August 31, 2025.

원문 및 DOI