DISC: Dynamic Feature Selection for Cost-Sensitive Medical Diagnosis
Yusheng Li, Xincen Duan, Beili Wang, Wei Guo, Han-Jia Ye
Abstract
Accurate medical diagnosis often relies on both textual self-reported symptoms and structured medical examination results of patients. However, these examinations vary significantly in cost—measured in time, money, or patient discomfort---creating a challenging trade-off between diagnostic accuracy and resource efficiency. To address this issue, we propose a dynamic diagnostic framework that incrementally selects medical examinations based on individual characteristics of each patient. Starting with textual self-reported symptoms and basic demographic, the system determines follow-up examinations step-by-step, improving accuracy while minimizing additional costs. Specifically, we introduce Dynamic feature selection with Instance-Specific Cost sensitivity (DISC). DISC treats each examination as a feature and learns to acquire them sequentially to optimize predictive performance under personalized cost constraints. To support richer clinical understanding, we further develop a multimodal framework that integrates unstructured self-reported symptom text with structured medical examination data. We conduct experiments on 680,000 patients with 43 million medical examination records, demonstrating that DISC high diagnostic accuracy even when accounting for examination costs. Our work provides substantial momentum for the advancement of AI in healthcare, offering both methodological and practical foundations that can significantly accelerate the deployment of intelligent, cost-aware diagnostic systems in real-world clinical settings.
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.
Builds on9
- Learning to Maximize Mutual Information for Dynamic Feature SelectionIan Connick Covert, Wei Qiu, Mingyu Lu, Nayoon Kim et al.ICML 2023 · 67 citations
- Active Feature Acquisition with Generative Surrogate ModelsYang Li, Junier OlivaICML 2021 · 52 citations
- Amortized Proximal OptimizationJuhan Bae, Paul Vicol, Jeff Z. HaoChen, Roger B. GrosseNeurIPS 2022 · 15 citations
- DiFA: Differentiable Feature AcquisitionAritra Ghosh, Andrew S. LanAAAI 2023 · 11 citations
- Acquisition Conditioned Oracle for Nongreedy Active Feature AcquisitionMichael Valancius, Max Lennon, Junier OlivaICML 2024 · 7 citations
Related papers
- Timely Clinical Diagnosis through Active Test SelectionSilas Ruhrberg Estévez, Nicolás Astorga, Mihaela van der SchaarNeurIPS 2025 · 4 citations
- Active feature acquisition via explainability-driven rankingOsman Berke Güney, Ketan Suhaas Saichandran, Karim Elzokm, Ziming Zhang et al.ICML 2025
- Salus: Strategic Diagnostic Testing for Complex Diagnosis via Multi-Agent Reinforcement LearningShuohao Gao, Xuanzhong Chen, Lingxiao Luo, Zilin Ding et al.ICML 2026
- PatientVLM Meets DocVLM: Pre-Consultation Dialogue Between Vision-Language Models for Efficient DiagnosisK. Lokesh, Abhirama Subramanyam Penamakuri, Uday Agarwal, Apoorva Challa et al.AAAI 2026
- Dialogue Based Disease Screening Through Domain Customized Reinforcement LearningZhuo Liu, Yanxuan Li, Xingzhi Sun, Fei Wang et al.KDD 2021 · 8 citations
