MedRoute: RL-Based Dynamic Specialist Routing in Multi-Agent Medical Diagnosis
This addresses the need for more precise and adaptable AI-driven medical diagnosis in healthcare, though it appears incremental as it builds on existing multi-agent LMM approaches.
The paper tackles the problem of overly general medical diagnosis by Large Multimodal Models (LMMs) by proposing MedRoute, a dynamic multi-agent framework that emulates clinical workflows with specialist agents, resulting in improved diagnostic accuracy over state-of-the-art baselines.
Medical diagnosis using Large Multimodal Models (LMMs) has gained increasing attention due to capability of these models in providing precise diagnoses. These models generally combine medical questions with visual inputs to generate diagnoses or treatments. However, they are often overly general and unsuitable under the wide range of medical conditions in real-world healthcare. In clinical practice, diagnosis is performed by multiple specialists, each contributing domain-specific expertise. To emulate this process, a potential solution is to deploy a dynamic multi-agent LMM framework, where each agent functions as a medical specialist. Current approaches in this emerging area, typically relying on static or predefined selection of various specialists, cannot be adapted to the changing practical scenario. In this paper, we propose MedRoute, a flexible and dynamic multi-agent framework that comprises of a collaborative system of specialist LMM agents. Furthermore, we add a General Practitioner with an RL-trained router for dynamic specialist selection, and a Moderator that produces the final decision. In this way, our framework closely mirrors real clinical workflows. Extensive evaluations on text and image-based medical datasets demonstrate improved diagnostic accuracy, outperforming the state-of-the-art baselines. Our work lays a strong foundation for future research. Code and models are available at https://github.com/UCF-CRCV/MedRoute/.