LLMs-based Few-Shot Disease Predictions using EHR: A Novel Approach Combining Predictive Agent Reasoning and Critical Agent Instruction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cui, Hejie, Shen, Zhuocheng, Zhang, Jieyu, Shao, Hui, Qin, Lianhui, Ho, Joyce C., Yang, Carl
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909147798175744
author Cui, Hejie
Shen, Zhuocheng
Zhang, Jieyu
Shao, Hui
Qin, Lianhui
Ho, Joyce C.
Yang, Carl
author_facet Cui, Hejie
Shen, Zhuocheng
Zhang, Jieyu
Shao, Hui
Qin, Lianhui
Ho, Joyce C.
Yang, Carl
contents Electronic health records (EHRs) contain valuable patient data for health-related prediction tasks, such as disease prediction. Traditional approaches rely on supervised learning methods that require large labeled datasets, which can be expensive and challenging to obtain. In this study, we investigate the feasibility of applying Large Language Models (LLMs) to convert structured patient visit data (e.g., diagnoses, labs, prescriptions) into natural language narratives. We evaluate the zero-shot and few-shot performance of LLMs using various EHR-prediction-oriented prompting strategies. Furthermore, we propose a novel approach that utilizes LLM agents with different roles: a predictor agent that makes predictions and generates reasoning processes and a critic agent that analyzes incorrect predictions and provides guidance for improving the reasoning of the predictor agent. Our results demonstrate that with the proposed approach, LLMs can achieve decent few-shot performance compared to traditional supervised learning methods in EHR-based disease predictions, suggesting its potential for health-oriented applications.
format Preprint
id arxiv_https___arxiv_org_abs_2403_15464
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LLMs-based Few-Shot Disease Predictions using EHR: A Novel Approach Combining Predictive Agent Reasoning and Critical Agent Instruction
Cui, Hejie
Shen, Zhuocheng
Zhang, Jieyu
Shao, Hui
Qin, Lianhui
Ho, Joyce C.
Yang, Carl
Computation and Language
Artificial Intelligence
Machine Learning
Multiagent Systems
J.3; I.2.7
Electronic health records (EHRs) contain valuable patient data for health-related prediction tasks, such as disease prediction. Traditional approaches rely on supervised learning methods that require large labeled datasets, which can be expensive and challenging to obtain. In this study, we investigate the feasibility of applying Large Language Models (LLMs) to convert structured patient visit data (e.g., diagnoses, labs, prescriptions) into natural language narratives. We evaluate the zero-shot and few-shot performance of LLMs using various EHR-prediction-oriented prompting strategies. Furthermore, we propose a novel approach that utilizes LLM agents with different roles: a predictor agent that makes predictions and generates reasoning processes and a critic agent that analyzes incorrect predictions and provides guidance for improving the reasoning of the predictor agent. Our results demonstrate that with the proposed approach, LLMs can achieve decent few-shot performance compared to traditional supervised learning methods in EHR-based disease predictions, suggesting its potential for health-oriented applications.
title LLMs-based Few-Shot Disease Predictions using EHR: A Novel Approach Combining Predictive Agent Reasoning and Critical Agent Instruction
topic Computation and Language
Artificial Intelligence
Machine Learning
Multiagent Systems
J.3; I.2.7
url https://arxiv.org/abs/2403.15464