Review learning: Real world validation of privacy preserving continual learning across medical institutions
Created by
Haebom
저자
Jaesung Yoo, Sunghyuk Choi, Ye Seul Yang, Suhyeon Kim, Jieun Choi, Dongkyeong Lim, Yaeji Lim, Hyung Joon Joo, Dae Jung Kim, Rae Woong Park, Hyeong-Jin Yoon, Kwangsoo Kim
개요
본 논문은 전이 학습 기반 개인정보보호 딥러닝(PPDL) 응용에서 나타나는 심각한 문제인 급격한 망각(catastrophic forgetting)을 해결하기 위해, '검토 학습'(RevL)이라는 저비용 연속 학습 알고리즘을 제안합니다. RevL은 모델이 이전 데이터셋으로부터의 지식을 검토하는 데 사용되는 데이터 샘플을 생성합니다. 전자 건강 기록(EHR)을 이용한 진단 예측에 RevL을 적용하여 6개의 모의 실험과 3개 의료기관의 실제 데이터를 이용한 실험을 진행했습니다. 실제 환자 106,508명의 데이터를 사용한 실험 결과, RevL의 평균 글로벌 AUC는 0.710, 전이 학습(TL)의 경우 0.655로 나타나 RevL이 이전에 학습된 지식을 유지하고 실제 PPDL 시나리오에서 효과적임을 보여줍니다. 이 연구는 의료기관 간 모델 전이를 기반으로 한 PPDL 연구를 위한 현실적인 파이프라인을 구축하고, 개인 EHR 데이터를 사용한 실제 의료 환경에서 연속 학습의 실용성을 강조합니다.