AI 기반 한국어 의료자연어 처리 기술 개발

고대안암병원 주형준 교수팀

고려대학교 안암병원은 주형준 교수팀(순환기내과 주형준 교수, 김종호 교수, 언어학과 송상헌 교수, 한양대학교 김영민 교수, 가톨릭대학교 김유중 교수)이 인공지능으로 한국어 의료 자연어 처리기술을 개발했다고 26일 밝혔다. 이번에 개발한 기술은 몇 문장만 입력하면 적절한 진료과를 추천해주는 인공지능 기반의 기술이다.
 
연구팀은 올해 초 2만6986개의 의학용어를 정리하고, 한국어 일반 자연어처리 모델인 KR-BERT를 기반으로 의학관련 지식(논문 및 기사 등) 통해 약 600만개의 문장과 약 1억1600만개의 단어를 학습시켜 의료분야에 특화한 자연어처리 모델인 KM(Korean Medical)-BERT를 개발했다.
 
KM-BERT는 기존 일반 자연어처리 모델인 M-BERT, KR-BERT에 비해 월등한 의료분야 한국어 자연어처리 성능을 보여 국제학술지에 소개된 바 있다.
 
이번 연구는 KM-BERT에 대한 후속 연구로서 질문자가 몇 문장을 입력하면, 인공지능이 질문자의 문장에서 핵심단어를 찾고 가장 적합한 진료과 3곳을 추천한다. 예를 들어, “중앙 갈비뼈 통증 숨 들이마실 때랑 기침할 때 여기가 너무 아파요ㅠ 왜 그런거죠..?”라고 입력하면, 인공지능이 학습한 내용을 기반으로 33.9%의 예측확률로 호흡기내과, 31.6%로 흉부외과, 9.8%로 가정의학과를 추천한다.
 
주형준 교수는 “양질의 데이터셋을 통해 완성도 높은 언어모델을 구축할 수 있었다”고 전한 뒤 “이번 언어모델 개발과 고도화는 KM-BERT를 활용해 실제로 적용 가능한 의료서비스 개발이 가능하다는 것을 보여줬다는 점에서 의미가 있으며, 앞으로 의료 및 헬스케어 분야에 여러 가지로 활용 가능할 것으로 기대한다”고 강조했다.
 
이번 연구 'Predicting medical specialty from text based on a domain-specific pre-trained BERT'는 국제학술지 'International Journal of Medical Informatics' 최신호에 게재됐다.