MDAgents: An Adaptive Collaboration of LLMs for Medical Decision-Making

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kim, Yubin, Park, Chanwoo, Jeong, Hyewon, Chan, Yik Siu, Xu, Xuhai, McDuff, Daniel, Lee, Hyeonhoon, Ghassemi, Marzyeh, Breazeal, Cynthia, Park, Hae Won
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916460283035648
author Kim, Yubin
Park, Chanwoo
Jeong, Hyewon
Chan, Yik Siu
Xu, Xuhai
McDuff, Daniel
Lee, Hyeonhoon
Ghassemi, Marzyeh
Breazeal, Cynthia
Park, Hae Won
author_facet Kim, Yubin
Park, Chanwoo
Jeong, Hyewon
Chan, Yik Siu
Xu, Xuhai
McDuff, Daniel
Lee, Hyeonhoon
Ghassemi, Marzyeh
Breazeal, Cynthia
Park, Hae Won
contents Foundation models are becoming valuable tools in medicine. Yet despite their promise, the best way to leverage Large Language Models (LLMs) in complex medical tasks remains an open question. We introduce a novel multi-agent framework, named Medical Decision-making Agents (MDAgents) that helps address this gap by automatically assigning a collaboration structure to a team of LLMs. The assigned solo or group collaboration structure is tailored to the medical task at hand, emulating real-world medical decision-making processes adapted to tasks of varying complexities. We evaluate our framework and baseline methods using state-of-the-art LLMs across a suite of real-world medical knowledge and medical diagnosis benchmarks, including a comparison of LLMs' medical complexity classification against human physicians. MDAgents achieved the best performance in seven out of ten benchmarks on tasks requiring an understanding of medical knowledge and multi-modal reasoning, showing a significant improvement of up to 4.2% (p < 0.05) compared to previous methods' best performances. Ablation studies reveal that MDAgents effectively determines medical complexity to optimize for efficiency and accuracy across diverse medical tasks. Notably, the combination of moderator review and external medical knowledge in group collaboration resulted in an average accuracy improvement of 11.8%. Our code can be found at https://github.com/mitmedialab/MDAgents.
format Preprint
id arxiv_https___arxiv_org_abs_2404_15155
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MDAgents: An Adaptive Collaboration of LLMs for Medical Decision-Making
Kim, Yubin
Park, Chanwoo
Jeong, Hyewon
Chan, Yik Siu
Xu, Xuhai
McDuff, Daniel
Lee, Hyeonhoon
Ghassemi, Marzyeh
Breazeal, Cynthia
Park, Hae Won
Computation and Language
Artificial Intelligence
Machine Learning
Foundation models are becoming valuable tools in medicine. Yet despite their promise, the best way to leverage Large Language Models (LLMs) in complex medical tasks remains an open question. We introduce a novel multi-agent framework, named Medical Decision-making Agents (MDAgents) that helps address this gap by automatically assigning a collaboration structure to a team of LLMs. The assigned solo or group collaboration structure is tailored to the medical task at hand, emulating real-world medical decision-making processes adapted to tasks of varying complexities. We evaluate our framework and baseline methods using state-of-the-art LLMs across a suite of real-world medical knowledge and medical diagnosis benchmarks, including a comparison of LLMs' medical complexity classification against human physicians. MDAgents achieved the best performance in seven out of ten benchmarks on tasks requiring an understanding of medical knowledge and multi-modal reasoning, showing a significant improvement of up to 4.2% (p < 0.05) compared to previous methods' best performances. Ablation studies reveal that MDAgents effectively determines medical complexity to optimize for efficiency and accuracy across diverse medical tasks. Notably, the combination of moderator review and external medical knowledge in group collaboration resulted in an average accuracy improvement of 11.8%. Our code can be found at https://github.com/mitmedialab/MDAgents.
title MDAgents: An Adaptive Collaboration of LLMs for Medical Decision-Making
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2404.15155