Towards Trustworthy Automatic Diagnosis Systems by Emulating Doctors' Reasoning with Deep Reinforcement Learning
Arsène Fansi Tchango, Rishab Goel, Julien Martel, Zhi Wen, Gaétan Marceau-Caron, Joumana Ghosn
Abstract
The automation of the medical evidence acquisition and diagnosis process has recently attracted increasing attention in order to reduce the workload of doctors and democratize access to medical care. However, most works proposed in the machine learning literature focus solely on improving the prediction accuracy of a patient's pathology. We argue that this objective is insufficient to ensure doctors' acceptability of such systems. In their initial interaction with patients, doctors do not only focus on identifying the pathology a patient is suffering from; they instead generate a differential diagnosis (in the form of a short list of plausible diseases) because the medical evidence collected from patients is often insufficient to establish a final diagnosis. Moreover, doctors explicitly explore severe pathologies before potentially ruling them out from the differential, especially in acute care settings. Finally, for doctors to trust a system's recommendations, they need to understand how the gathered evidences led to the predicted diseases. In particular, interactions between a system and a patient need to emulate the reasoning of doctors. We therefore propose to model the evidence acquisition and automatic diagnosis tasks using a deep reinforcement learning framework that considers three essential aspects of a doctor's reasoning, namely generating a differential diagnosis using an exploration-confirmation approach while prioritizing severe pathologies. We propose metrics for evaluating interaction quality based on these three aspects. We show that our approach performs better than existing models while maintaining competitive pathology prediction accuracy. I have recently had stools that were black (like coal)
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 998d33e9-4ee6-4b28-a218-bc01aa5de32eCited by top-tier papers2
- ProtoGate: Prototype-based Neural Networks with Global-to-local Feature Selection for Tabular Biomedical DataXiangjian Jiang, Andrei Margeloiu, Nikola Simidjievski, Mateja JamnikICML 2024 · 23 citations
- MediTOD: An English Dialogue Dataset for Medical History Taking with Comprehensive AnnotationsVishal Vivek Saley, Goonjan Saha, Rocktim Jyoti Das, Dinesh Raghu et al.EMNLP 2024 · 1 citation
Builds on1
Related papers
- Dialogue Based Disease Screening Through Domain Customized Reinforcement LearningZhuo Liu, Yanxuan Li, Xingzhi Sun, Fei Wang et al.KDD 2021 · 8 citations
- MEDDxAgent: A Unified Modular Agent Framework for Explainable Automatic Differential DiagnosisDaniel Philip Rose, Chia-Chien Hung, Marco Lepri, Israa Alqassem et al.ACL 2025 · 14 citations
- MedEyes: Learning Dynamic Visual Focus for Medical Progressive DiagnosisChunzheng Zhu, Yangfang Lin, Shen Chen, Yijun Wang et al.AAAI 2026 · 8 citations
- Timely Clinical Diagnosis through Active Test SelectionSilas Ruhrberg Estévez, Nicolás Astorga, Mihaela van der SchaarNeurIPS 2025 · 4 citations
- Language Agents for Hypothesis-driven Clinical Decision Making with Reinforcement LearningDavid Bani-Harouni, Chantal Pellegrini, Ege Özsoy, Nassir Navab et al.ICLR 2026 · 16 citations
