Infection diagnosis in hydrocephalus CT images: a domain enriched attention learning approach.
Mingzhao YuMallory R PetersonVenkateswararao CherukuriChristine HehnlyEdith Mbabazi-KabachelorRonnie MulondoBrian Nsubuga KaayaJames R BroachSteven J SchiffVishal MongaPublished in: Journal of neural engineering (2023)
Hydrocephalus is the leading indication for pediatric neurosurgical care worldwide. Identification of postinfectious hydrocephalus (PIH) verses non-postinfectious hydrocephalus (NPIH), 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 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. In this paper, a novel brain attention regularizer (BAR) 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. 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. . 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 hydrocehpalus and underlying pathogen using CT scans.
Keyphrases
- convolutional neural network
- deep learning
- artificial intelligence
- subarachnoid hemorrhage
- cerebrospinal fluid
- computed tomography
- image quality
- dual energy
- machine learning
- big data
- contrast enhanced
- working memory
- cerebral ischemia
- healthcare
- candida albicans
- decision making
- brain injury
- quality improvement
- positron emission tomography
- electronic health record
- white matter
- palliative care
- high resolution
- resting state
- small molecule
- magnetic resonance
- mass spectrometry
- magnetic resonance imaging
- binding protein
- pet ct
- emergency department
- replacement therapy