Infection Diagnosis of Hydrocephalus CT images

A Domain-enriched Attention Learning Approach

Journal of Neural Engineering 2023

ABSTRACT

Objective: Hydrocephalus is the leading indication for pediatric neurosurgical care worldwide. Identification of postinfectious hydrocephalus (PIH) verses non-postinfectious hydrocephalus, as well as the pathogen involved in PIH is crucial for developing an appropriate treatment plan. Accurate identification requires clinical diagnosis by neuroscientists and microbiological analysis, which are time-consuming and expensive. In this study, we develop a domain enriched AI method for computerized tomography (CT)-based infection diagnosis in hydrocephalic imagery. State-of-the-art (SOTA) convolutional neural network (CNN) approaches form an attractive neural engineering solution for addressing this problem as pathogen-specific features need discovery. Yet black-box deep networks often need unrealistic abundant training data and are not easily interpreted. Approach. In this paper, a novel brain attention regularizer is proposed, which encourages the CNN to put more focus inside brain regions in its feature extraction and decision making. Our approach is then extended to a hybrid 2D/3D network that mines inter-slice information. A new strategy of regularization is also designed for enabling collaboration between 2D and 3D branches. Main results. Our proposed method achieves SOTA results on a CURE Children's Hospital of Uganda dataset with an accuracy of 95.8% in hydrocephalus classification and 84% in pathogen classification. Statistical analysis is performed to demonstrate that our proposed methods obtain significant improvements over the existing SOTA alternatives. Significance. Such attention regularized learning has particularly pronounced benefits in regimes where training data may be limited, thereby enhancing generalizability. To the best of our knowledge, our findings are unique among early efforts in interpretable AI-based models for classification of hydrocephalus and underlying pathogen using CT scans.

Code

Code and building blocks of BAR-Net and CBAR-Net can be found here.

Paper can be found here.

BAR-Net

Loss function for BAR-Net:

where A is class activation map and M is brain mask.

CBAR-Net

Motivation: Since the 2D branch’s input is a subset of the 3D branch’s input, it is expected that the 2D branch’s activated region is contained within the 3D branch’s activated region. Similarly, the 3D branch’s non-activated region is expected 3D to be contained within the 2D branch’s non-activated region.

Collaborative Brain Attention Regularization To incorporate this information into the classification model, we proposed Collaborative Brain Attention Regularization and corresponding CBAR-Net.

Loss function for CBAR-Net:

where A_2D, A_3D are class activation maps of 2D and 3D branches.

Results

CAMs

Related Publications

  1. Yu, Mingzhao, et al. "Infection diagnosis in hydrocephalus CT images: a domain enriched attention learning approach." Journal of Neural Engineering (2023). [JNE]

Email
ipal.psu@gmail.com

Address
104 Electrical Engineering East,
University Park, PA 16802, USA

Lab Phone:
814-863-7810
814-867-4564