English

AgentMD: Empowering Language Agents for Risk Prediction with Large-Scale Clinical Tool Learning

Computation and Language 2024-02-21 v1 Artificial Intelligence

Abstract

Clinical calculators play a vital role in healthcare by offering accurate evidence-based predictions for various purposes such as prognosis. Nevertheless, their widespread utilization is frequently hindered by usability challenges, poor dissemination, and restricted functionality. Augmenting large language models with extensive collections of clinical calculators presents an opportunity to overcome these obstacles and improve workflow efficiency, but the scalability of the manual curation process poses a significant challenge. In response, we introduce AgentMD, a novel language agent capable of curating and applying clinical calculators across various clinical contexts. Using the published literature, AgentMD has automatically curated a collection of 2,164 diverse clinical calculators with executable functions and structured documentation, collectively named RiskCalcs. Manual evaluations show that RiskCalcs tools achieve an accuracy of over 80% on three quality metrics. At inference time, AgentMD can automatically select and apply the relevant RiskCalcs tools given any patient description. On the newly established RiskQA benchmark, AgentMD significantly outperforms chain-of-thought prompting with GPT-4 (87.7% vs. 40.9% in accuracy). Additionally, we also applied AgentMD to real-world clinical notes for analyzing both population-level and risk-level patient characteristics. In summary, our study illustrates the utility of language agents augmented with clinical calculators for healthcare analytics and patient care.

Keywords

Cite

@article{arxiv.2402.13225,
  title  = {AgentMD: Empowering Language Agents for Risk Prediction with Large-Scale Clinical Tool Learning},
  author = {Qiao Jin and Zhizheng Wang and Yifan Yang and Qingqing Zhu and Donald Wright and Thomas Huang and W John Wilbur and Zhe He and Andrew Taylor and Qingyu Chen and Zhiyong Lu},
  journal= {arXiv preprint arXiv:2402.13225},
  year   = {2024}
}

Comments

Work in progress

R2 v1 2026-06-28T14:54:51.738Z