Daily Arxiv

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

Transformer-Based Sleep Stage Classification Enhanced by Clinical Information

Created by
  • Haebom
Category
Empty

저자

Woosuk Chung, Seokwoo Hong, Wonhyeok Lee, Sangyoon Bae

개요

수면다원검사(PSG) 기반 수면 단계 분류는 인력 투입이 많고, 분석자 간의 변동성이 크다. 본 논문은 딥러닝 모델의 발전을 위해, PSG 신호 외에 전문가들이 사용하는 임상적 맥락 정보를 활용하는 2단계 아키텍처를 제안한다. 해당 아키텍처는 Transformer 기반의 에포크별 인코더와 1D CNN 애그리게이터를 결합하여, 환자 수준의 임상 메타데이터(나이, 성별, BMI) 및 에포크별 전문가 이벤트 주석(무호흡, 탈산소, 각성, 주기적 호흡)을 통합한다. Sleep Heart Health Study(SHHS) 코호트(n=8,357)를 사용하여, 맥락 정보를 통합함으로써 수면 단계 분류 정확도가 크게 향상됨을 입증했다. 특히, 이벤트 주석이 가장 큰 기여를 했으며, 기능 융합이 동일한 보조 레이블을 예측하는 다중 작업(multi-task) 대안보다 우수했다.

시사점, 한계점

임상적 맥락 정보를 통합하여 수면 단계 분류 정확도 향상 (macro-F1 0.8031, micro-F1 0.9051)
PSG 신호만 사용한 baseline 모델 대비 성능 향상 (macro-F1 0.7745, micro-F1 0.8774)
이벤트 주석 정보가 가장 큰 성능 향상에 기여
PSG 몽타주나 추가 센서 없이 성능 향상
임상적 특징 통합을 통해 성능과 해석 가능성 향상
SHHS 코호트에 국한된 연구이므로, 다른 데이터셋에 대한 일반화는 추가 연구 필요
다중 작업 학습(Multi-task learning)보다 기능 융합 방식이 더 우수함
👍