Annual Conference on Human and Language Technology (한국정보과학회 언어공학연구회:학술대회논문집(한글 및 한국어 정보처리))
- 2021.10a
- /
- Pages.558-563
- /
- 2021
- /
- 2005-3053(pISSN)
BERT-based Two-Stage Classification Models for Alzheimer's Disease and Schizophrenia Diagnosis
BERT 기반 2단계 분류 모델을 이용한 알츠하이머병 치매와 조현병 진단
- Jung, Min-Kyo (Jeonbuk National University) ;
-
Na, Seung-Hoon
(Jeonbuk National University)
;
- Kim, Ko Woon (Jeonbuk National University Hospital) ;
- Shin, Byong-Soo (Jeonbuk National University Hospital) ;
- Chung, Young-Chul (Jeonbuk National University Hospital)
- Published : 2021.10.14
Abstract
알츠하이머병 치매와 조현병 진단을 위한 2단계 분류 모델을 제안한다. 정상군과 환자군의 발화에 나타난 페어 언어 모델 간의 Perplexity 차이에 기반한 분류와 기존 단일 BERT 모델의 미세조정(fine-tuning)을 이용한 분류의 통합을 시도하였다. Perplexity 기반의 분류 성능이 알츠하이머병, 조현병 모두 우수한 결과를 보임을 확인 하였고, 조현병 분류 모델의 성능이 소폭 증가하였다. 향후 설명 가능한 인공지능 기법을 적용에 따른 성능 향상을 기대할 수 있었다.