Fine-Tuning In-House Large Language Models to Infer Differential Diagnosis from Radiology Reports

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Luoyao, Teotia, Revant, Verdone, Antonio, Cardall, Aidan, Tyagi, Lakshay, Shen, Yiqiu, Chopra, Sumit
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909346745548800
author Chen, Luoyao
Teotia, Revant
Verdone, Antonio
Cardall, Aidan
Tyagi, Lakshay
Shen, Yiqiu
Chopra, Sumit
author_facet Chen, Luoyao
Teotia, Revant
Verdone, Antonio
Cardall, Aidan
Tyagi, Lakshay
Shen, Yiqiu
Chopra, Sumit
contents Radiology reports summarize key findings and differential diagnoses derived from medical imaging examinations. The extraction of differential diagnoses is crucial for downstream tasks, including patient management and treatment planning. However, the unstructured nature of these reports, characterized by diverse linguistic styles and inconsistent formatting, presents significant challenges. Although proprietary large language models (LLMs) such as GPT-4 can effectively retrieve clinical information, their use is limited in practice by high costs and concerns over the privacy of protected health information (PHI). This study introduces a pipeline for developing in-house LLMs tailored to identify differential diagnoses from radiology reports. We first utilize GPT-4 to create 31,056 labeled reports, then fine-tune open source LLM using this dataset. Evaluated on a set of 1,067 reports annotated by clinicians, the proposed model achieves an average F1 score of 92.1\%, which is on par with GPT-4 (90.8\%). Through this study, we provide a methodology for constructing in-house LLMs that: match the performance of GPT, reduce dependence on expensive proprietary models, and enhance the privacy and security of PHI.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09234
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Fine-Tuning In-House Large Language Models to Infer Differential Diagnosis from Radiology Reports
Chen, Luoyao
Teotia, Revant
Verdone, Antonio
Cardall, Aidan
Tyagi, Lakshay
Shen, Yiqiu
Chopra, Sumit
Computation and Language
Radiology reports summarize key findings and differential diagnoses derived from medical imaging examinations. The extraction of differential diagnoses is crucial for downstream tasks, including patient management and treatment planning. However, the unstructured nature of these reports, characterized by diverse linguistic styles and inconsistent formatting, presents significant challenges. Although proprietary large language models (LLMs) such as GPT-4 can effectively retrieve clinical information, their use is limited in practice by high costs and concerns over the privacy of protected health information (PHI). This study introduces a pipeline for developing in-house LLMs tailored to identify differential diagnoses from radiology reports. We first utilize GPT-4 to create 31,056 labeled reports, then fine-tune open source LLM using this dataset. Evaluated on a set of 1,067 reports annotated by clinicians, the proposed model achieves an average F1 score of 92.1\%, which is on par with GPT-4 (90.8\%). Through this study, we provide a methodology for constructing in-house LLMs that: match the performance of GPT, reduce dependence on expensive proprietary models, and enhance the privacy and security of PHI.
title Fine-Tuning In-House Large Language Models to Infer Differential Diagnosis from Radiology Reports
topic Computation and Language
url https://arxiv.org/abs/2410.09234