DocScribe: Chatbot QA Medis

Perkenalan

DocScribe adalah chatbot tanya jawab medis yang merevolusi cara kita berinteraksi dengan data medis. DocScribe memberikan tanggapan yang cepat dan akurat terhadap pertanyaan medis umum dan pertanyaan spesifik pasien. Tujuan utama kami adalah untuk meningkatkan aksesibilitas dan pemahaman laporan medis. DocScribe mencapai hal ini melalui:

Transkripsi laporan medis ke dalam format yang lebih mudah diakses.
Memberikan jawaban atas pertanyaan medis umum.
Meringkas laporan dari beberapa kunjungan.
Mengidentifikasi informasi yang relevan dari riwayat pasien.

Arsitektur

Arsitektur DocScribe memfasilitasi interaksi yang lancar antara pengguna dan data medis. Ini mencakup:

UI Web berbasis Notebook Jupyter untuk mengunggah laporan dan interaksi lebih lanjut.
Pembuatan embeddings dan indeks dari transkrip medis.
Pemanfaatan kerangka LangChain dan HuggingFace untuk memproses dan menjawab pertanyaan.

Arsitektur DocScribe

Sumber Data

Proyek kami memanfaatkan beragam sumber data untuk melatih model kami, termasuk:

Transkrip Medis: Kami memanfaatkan model GPT-3.5 untuk menghasilkan 4,5 ribu perintah QA dari transkrip medis. Sampel MT
WikiDoc: Kunjungi WikiDoc
Pasien Wiki: Kunjungi WikiPatient

Contoh Data

Kumpulan data	Petunjuk	Masukan	Keluaran
Wikidoc	Jawablah pertanyaan ini dengan sejujurnya	Bisakah Anda memberikan gambaran tentang karsinoma sel skuamosa paru?	Karsinoma sel skuamosa paru-paru dapat diklasifikasikan menurut sistem klasifikasi histologis WHO menjadi 4 jenis utama: papiler, sel jernih, sel kecil, dan basaloid.
Pasien Wiki	Jawablah pertanyaan ini dengan sejujurnya	Kapan saya harus mencari perawatan medis darurat ketika saya menderita sindrom Alstrom?	Hubungi penyedia layanan kesehatan Anda jika Anda atau anak Anda memiliki gejala diabetes seperti peningkatan rasa haus dan buang air kecil. Segera cari pertolongan medis jika Anda merasa anak Anda tidak dapat melihat atau mendengar dengan normal.
Sampel MT	Berdasarkan transkrip medis yang diberikan menghasilkan prompt dan jawaban untuk melatih LLM	Apa diagnosis pasien sebelum operasi?	Diagnosis pra operasi pasien adalah kanker prostat.

Pendekatan Pemodelan

Kami memilih model Vicuna-13B, yang disempurnakan dengan LoRA, PEFT, dan bitsandbytes. Pendekatan kami telah divalidasi melalui pengujian yang ketat dan telah menunjukkan hasil yang menjanjikan dalam menafsirkan data medis.

Contoh Hasil

Hasil & Pekerjaan Masa Depan

DocScribe telah menunjukkan kemahiran luar biasa dalam memproses pertanyaan medis dan merangkum laporan pasien. Arah masa depan termasuk memperluas pelatihan model pada korpus medis, menggabungkan analisis citra medis, dan mengeksplorasi penerapannya dalam penelitian klinis.

Instalasi

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