Integrating Clinical Insights via Hierarchical Inference to Predict Conditions in Bilaterally Symmetric Organs
Substantial progress has been made in developing deep-learning models for clinical diagnosis. While excelling in diagnostics, the broader clinical decision-making process also involves establishing optimal follow-up intervals (TCU), crucial for prognosis and timely treatment. To fully support clinical practice, it is imperative that deep learning models contribute to both initial diagnosis and TCU prediction. However, relying on separate monolithic models is computationally demanding and lacks interpretability, hindering clinician trust. Our proposed bilateral model, emphasizing ophthalmological cases, offers both initial diagnoses and follow-up predictions, enhancing interpretability and trust in clinical applications as clinicians are more likely to trust recommendations, knowing the diagnosis used is correct. Inspired by clinical practice, the model integrates hierarchical inference and self-supervised learning techniques to enhance predictive accuracy and interpretability. Consisting of a sparse autoencoder, diagnosis classifier, and TCU classifier, the model leverages insights from clinicians and observations of ophthalmological datasets to capture salient features and facilitate robust learning. By employing shared weights for encoding and diagnosing each organ, the model optimizes efficiency and doubles the effective dataset size. Experimental results on an ophthalmological dataset demonstrate superior performance compared to baseline models, with the hierarchical inference structure providing valuable insights into the model's decision-making process. The bilateral model not only enhances predictive modeling for conditions affecting bilaterally symmetrical organs but also empowers clinicians with interpretable outputs crucial for informed clinical decision-making, thereby advancing clinical practice and improving patient care.
Authors
- Li Rong Wang (ORCID: https://orcid.org/0000-0003-3546-9354)
- Kelvin Zhenghao Li (ORCID: https://orcid.org/0000-0002-9993-5118)
- Vivien Cherng-Hui Yip (ORCID: https://orcid.org/0000-0003-2479-8515)
- Xiuyi Fan (ORCID: https://orcid.org/0000-0003-1223-9986)
- Si Yin Charlene Chia
Institutions
- Nanyang Technological University (SG)
- Tan Tock Seng Hospital (SG)
Publication Details
- Journal
- IEEE Journal of Biomedical and Health Informatics
- Published
- 2025-04-01
- DOI
- https://doi.org/10.1109/jbhi.2025.3556717
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00