Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Chronic Kidney Disease Prognosis Prediction Using Transformer

Created by
  • Haebom
Category
Empty

저자

Yohan Lee, DongGyun Kang, SeHoon Park, Sa-Yoon Park, Kwangsoo Kim

개요

만성 신장 질환(CKD) 진행 예측을 위한 Transformer 기반 프레임워크인 ProQ-BERT를 제시합니다. 서울대학교병원 OMOP 공통 데이터 모델의 멀티 모달 전자 건강 기록(EHR)을 사용하여, 인구 통계, 임상 및 실험실 데이터를 통합합니다. 연속적인 실험실 값에 대한 양자화 기반 토큰화와 해석 가능성을 위한 어텐션 메커니즘을 사용합니다. ProQ-BERT는 3a 단계에서 5단계로의 진행을 예측하는 이진 분류 작업에 대해 사전 학습 및 미세 조정되었습니다. 91,816명의 환자 코호트에서 평가한 결과, 단기 예측에서 ROC-AUC 최대 0.995, PR-AUC 최대 0.989를 달성하며 CEHR-BERT를 능가했습니다.

시사점, 한계점

Transformer 아키텍처와 시간적 설계 선택의 효과를 강조하여 임상 예후 모델링에 유망한 방향 제시
ROC-AUC 최대 0.995, PR-AUC 최대 0.989로 우수한 예측 성능 달성
단기 예측에 초점을 맞춰 장기적인 예후 예측에 대한 추가 연구 필요
단일 기관(서울대학교병원) 데이터에 기반하여, 일반화 가능성을 높이기 위해 다른 데이터셋에서의 검증 필요
모델의 해석 가능성은 높지만, 임상적 적용을 위한 추가적인 임상적 검증 필요
👍