연합 학습의 파티클 군집 최적화 통합
Integrating Particle Swarm Dynamics into Federated Learning: A Promising Hybrid to Counter Client Drift under Non-IID Conditions
Marwa K. Farhan, Ruslan Saad Abdulrahman, Aseel B. Alnajjar 외 1인·International journal of intelligent engineering and systems·발표 2026.08· 0 인용
한국어 핵심 요약
연합 학습(FL)은 원본 데이터를 중앙 집중화하지 않고 분산 모델 훈련을 가능하게 하여 프라이버시에 민감한 IoT 환경에 적합합니다. 그러나 기존 FL 알고리즘은 통계적으로 이질적인(non-IID) 데이터 분포에서 클라이언트 드리프트로 인해 수렴 속도가 느려지고 성능이 저하되는 문제가 있습니다. 본 논문은 이러한 문제를 해결하기 위해 확률적 경사 하강법의 클라이언트 업데이트 규칙에 파티클 군집 최적화(PSO) 역학(속도 평활화, 개인 최적 기억, 전역 최적 조정)을 통합한 하이브리드 연합-군집 최적화(HFSO) 알고리즘을 제안합니다.
제안된 HFSO 알고리즘은 라운드당 통신 부하를 늘리지 않으면서 심각한 non-IID 조건에서 수렴을 가속화하고 견고성을 향상시키는 것을 목표로 합니다. CIFAR-10 및 FEMNIST 데이터셋에 대한 실험 결과, HFSO는 FedAvg 및 최신 이질성 인식 방법(FedNova, MOON, CCVR)보다 우수한 성능을 보였습니다. 특히 CIFAR-10에서 HFSO는 500라운드에서 74.84%의 정확도를 달성하여 FedAvg의 62.44%를 크게 상회했습니다.
분포 분석 결과, HFSO는 최악의 클라이언트 정확도를 개선하고 클라이언트 간 분산을 감소시키는 것으로 나타났습니다. 제한된 화이트박스 위협 모델에서의 경험적 프라이버시 평가에서는 공격에 대한 취약성이 감소함을 보여주었습니다. 이는 HFSO가 non-IID 환경에서 연합 학습의 효율성과 안정성을 크게 향상시킬 수 있음을 시사합니다.
섹션 미리보기
연구 배경
연합 학습은 데이터 프라이버시를 보호하며 분산 환경에서 모델을 훈련하는 유망한 기술입니다. 하지만 이질적인 데이터 분포(non-IID) 환경에서는 클라이언트 드리프트 문제로 인해 수렴 지연과 성능 저하가 발생합니다.
핵심 발견
제안된 HFSO 알고리즘은 파티클 군집 최적화(PSO)를 통합하여 non-IID 조건에서 연합 학습의 수렴 속도와 정확도를 크게 향상시켰습니다. HFSO는 FedAvg 대비 12%p 이상 높은 정확도를 달성하며, 최신 이질성 인식 방법들보다 우수한 성능을 보였습니다.
관련 컴퓨터 과학 논문
부분 관측 다중 모드 헬스케어 예측을 위한 차등 프라이빗 연합 학습
2026·0
미주신경성 실신 탐지: 프라이버시 보호 ML
2026·0
AI 생성 대화의 자연스러움: 턴 주고받기 및 역할 행동 분석
2026·0
해양 공간 데이터 인프라와 오픈 사이언스 연계
2026·0