DocScribe: 의료 QA 챗봇

소개

DocScribe는 의료 데이터와 상호 작용하는 방식을 혁신하는 의료 질문 및 답변 챗봇입니다. DocScribe는 일반적인 의료 문의와 환자별 질문 모두에 대해 빠르고 정확한 답변을 제공합니다. 우리의 주요 목표는 의료 보고서의 접근성과 이해력을 높이는 것입니다. DocScribe는 다음을 통해 이를 달성합니다.

더 접근하기 쉬운 형식으로 의료 보고서를 전사합니다.
일반적인 의료 질문에 대한 답변을 제공합니다.
여러 방문에서 얻은 보고서를 요약합니다.
환자 병력에서 관련 정보를 식별합니다.

건축학

DocScribe의 아키텍처는 사용자와 의료 데이터 간의 원활한 상호 작용을 촉진합니다. 여기에는 다음이 포함됩니다.

보고서 업로드 및 추가 상호 작용을 위한 Jupyter Notebook 기반 웹 UI입니다.
의료 기록에서 임베딩 및 인덱스 생성.
질문을 처리하고 답변하기 위해 LangChain 및 HuggingFace 프레임워크를 활용합니다.

DocScribe 아키텍처

데이터 소스

우리 프로젝트에서는 다음과 같은 다양한 데이터 소스를 활용하여 모델을 교육합니다.

의료 성적 증명서: GPT-3.5 모델을 활용하여 의료 성적 증명서에서 4,500개의 QA 프롬프트를 생성했습니다. MT샘플
WikiDoc: WikiDoc 방문
WikiPatient: WikiPatient 방문

샘플 데이터

데이터세트	지침	입력	산출
위키독	이 질문에 솔직하게 대답하세요	폐의 편평 세포 암종에 대한 개요를 제공할 수 있습니까?	폐의 편평 세포 암종은 WHO 조직학적 분류 체계에 따라 유두상 암, 투명 세포 암, 소세포 암, 기저암의 4가지 주요 유형으로 분류될 수 있습니다.
위키환자	이 질문에 솔직하게 대답하세요	알스트롬 증후군이 있는 경우 언제 긴급 진료를 받아야 합니까?	귀하 또는 귀하의 자녀가 갈증 및 배뇨 증가와 같은 당뇨병 증상이 있는 경우 의료 서비스 제공자에게 연락하십시오. 자녀가 정상적으로 보거나 듣지 못한다고 생각되면 즉시 의사의 진료를 받으십시오.
MT 샘플	주어진 의료 기록을 기반으로 LLM 교육을 위한 프롬프트와 답변을 생성합니다.	환자의 수술 전 진단은 무엇이었나요?	환자의 수술 전 진단은 전립선암이었다.

모델링 접근법

우리는 LoRA, PEFT 및 비트샌드바이트로 미세 조정된 Vicuna-13B 모델을 선택했습니다. 우리의 접근 방식은 엄격한 테스트를 통해 검증되었으며 의료 데이터 해석에 있어 유망한 결과를 보여주었습니다.

샘플 결과

결과 및 향후 작업

DocScribe는 의료 문의 처리 및 환자 보고서 요약에 있어 놀라운 능력을 보여주었습니다. 향후 방향에는 의료 자료에 대한 모델 교육 확장, 의료 이미지 분석 통합, 임상 연구에서의 적용 탐색이 포함됩니다.

설치

git clone https://github.com/kmnis/DocScribe.git
cd DocScribe
pip install -r requirements.txt

# Start the jupyter server by running
jupyter notebook

# Open your browser and open http://localhost:8888/inference and open a notebook