본문으로 건너뛰기

© 2026 Molayo

arXiv논문2026. 05. 04. 20:03

FedKPer: 의료 연방 학습의 일반화 및 개인화를 지식 개인화를 통해 해결하다

요약

FedKPer는 의료 분야의 연방 학습(FL)에서 발생하는 통계적 이질성 문제를 해결하기 위해 지식 개인화(Knowledge Personalization)를 도입한 새로운 프레임워크입니다. 기존 FL 모델은 전역적인 일반화와 개별 병원의 특수한 데이터 분포에 대한 적응(개인화) 사이의 균형을 맞추는 데 어려움을 겪어 망각 현상을 초래했습니다. FedKPer는 로컬 학습 단계에 지식 개인화를 적용하고, 신뢰성 높은 업데이트를 강조하는 전역 집계 과정을 통해 이 두 목표 간의 트레이드오프를 개선하여 성능과 데이터 보존력을 동시에 높입니다.

핵심 포인트

  • 연방 학습(FL)은 의료 분야에서 잠재력이 크지만, 기관 간 통계적 이질성으로 인해 일반화와 개인화에 어려움을 겪는다.
  • 기존 연구는 일반화와 개인화를 분리하여 다루었으나, FedKPer는 지식 개인화를 통해 두 목표를 통합적으로 해결한다.
  • FedKPer는 로컬 장치 학습 단계에 지식 개인화를 도입하고, 전역 모델 집계 과정에서 신뢰성 높은 업데이트를 강조하는 방식으로 작동한다.
  • 이 프레임워크는 일반화-개인화 트레이드오프를 개선하여 데이터 리텐션(Retention)을 희생하지 않으면서 성능 향상을 입증했다.

연방 학습 (Federated Learning, FL) 은 의료 응용 분야에서 큰 잠재력을 가지고 있습니다. 그러나 의료 기관 간 통계적 이질성은 연방 학습에 대한 주요 도전 과제로 작용하며, 전역 모델은 보이지 않는 환자 인구 집단을 일반화하는 동시에 개별 병의 고유한 데이터 분포에 적응하는 데 어려움을 겪습니다. 이러한 이질성은 전역 및 지역 수준 모두에서 망각 (forgetting) 을 악화시키며, 모델 업데이트 후 이전에 학습된 환자 패턴이 오분류됩니다. 기존 연구는 일반적으로 일반화와 개인화를 별개의 도전 과제로 취급해 왔지만, 우리는 전역 모델과의 선택적 정렬과 수정된 집계 스키엄을 통해 두 가지 사이의 더 나은 균형을 달성할 수 있음을 보여줍니다. 이는 통계적 이질성의 영향을 완화합니다. 구체적으로, FedKPer 를 소개하며, 각 로컬 장치의 학습 단계에 지식 개인화 (Knowledge Personalization) 를 도입합니다. 그 후, 일반화는 신뢰성 있고 라벨 다양성이 높은 로컬 업데이트를 강조하는 전역 모델 집계 과정을 통해 고려됩니다. 우리는 FedKPer 의 성능을 평가하고, 망각의 일반적인 결과와 관련된 추가 지표를 개발했습니다. 전체적으로, FedKPer 는 리텐션 (Retention) 을 희생하지 않으면서 일반화-개인화 트레이드오프를 개선함을 입증합니다.

AI 자동 생성 콘텐츠

본 콘텐츠는 arXiv cs.LG의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.

원문 바로가기
1

댓글

0