A novel causality analysis of brain tumor subregion-based automated segmentation networks on multi-sequence MRIs

Hao Chen,Fangyuan Du,Xuelian Yang,Xiaoyin Pan,X.Sharon Qi
DOI: https://doi.org/10.1016/j.bspc.2024.106466
IF: 5.1
2024-05-29
Biomedical Signal Processing and Control
Abstract:The 'black-box' nature of Deep Neural Networks (DNNs) limits their application in healthcare and many other prominent fields. The interpretable techniques for DNNs are expected to mitigate the potential issue. Causal analysis refers to methods and techniques to identify the pivotal causal relationships between the input data and the decision-making results. In this study, we developed a novel causal analysis mechanism to explain the internal computational logic of DNNs. We used a Boosting Ensemble Learning process to globally simulate DNNs to create a 'Grey-box' fitting model. By exploiting a degree of transparency that the Grey-box model has, we developed a rule extraction mechanism for collecting a set of causal rules from the fitting model to reveal the calculation logic of a type of operation in the network. The Gradient Boosting Decision Tree algorithm was performed to implement fitting learning for typical DNNs. To demonstrate the effectiveness of the proposed method, we applied the framework to the widely adopted U-Net for the task of glioma auto-segmentation on multi-sequence MRIs as the research object. A set of 79 rules were extracted from the fitting result to capture the logic of segmenting the enhancing tumor subregion by U-Net. Through carefully designed experiments, we validated the significance and utility of the identified causal rules. The causality analysis framework reveals the internal computational logic of complex DNNs for decision-making in target delineation and various clinical applications. Our work holds great potentials and insights for further improvement of DNN model performance and generalizability in medical image analysis and beyond. Source code is available at https://github.com/CodingVoyagers/BMFM_U-Net-fitting-mode .
engineering, biomedical
What problem does this paper attempt to address?