A human-in-the-loop explanation framework for morphologically transparent AI predictions from whole-slide images
A human-in-the-loop explanation framework for morphologically transparent AI predictions from whole-slide images
Lou, P.; Zhu, Y.; Chia, N.; Kumari, R.; Yang, W.; Wang, Y.; Brenna, N.; Winham, S.; Guo, R.; Goode, E.; Huang, Y.; Han, W.; Feng, T.; Wang, C.
AbstractDeep learning models enable the prediction of clinical endpoints from whole-slide images (WSIs), but many such models function as "black boxes", lacking transparency about whether and which histomorphological patterns drive their predictions, hindering interpretability and clinical adoption. Here we propose a human-in-the-loop explanation framework, MorphoXAI, which provides both local and global interpretability for deep learning models by incorporating human-expert interpretations. At the global level, it reveals the histomorphological patterns on which the model consistently relies to distinguish between classes of WSIs, as well as the patterns associated with confusion between classes. At the local level, it indicates which of these patterns are used in the prediction of an individual WSI and which regions within the slide correspond to such patterns. We validated our method on a deep learning model trained for ovarian tumor histologic subtype prediction. The results show that our framework generates explanations that accurately reflect the histomorphology underlying the model's predictions at both global and local levels. For interpretability and clinical utility in diagnostic contexts, human evaluation results showed that our explanations were easy to interpret, rich in diagnostic features, and directly helpful for diagnostic decision-making, thereby enhancing pathologist-AI collaboration. Our work highlights that unifying global and local explanations and grounding them in expert-interpreted morphology enhances the interpretability and verifiability of deep learning models, thereby facilitating the transparent deployment of such models in clinical practice.