Tang Zijia, Zhang Tonglin, Song Qianqian, Su Jing, Yang Baijian
Guangdong Experimental High School, Guangzhou Guangdong China.
Department of Statistics, Purdue University, West Lafayette, IN, USA.
ACM BCB. 2023 Sep;2023. doi: 10.1145/3584371.3613000. Epub 2023 Oct 4.
The irreversible and progressive atrophy by Alzheimer's Disease resulted in continuous decline in thinking and behavioral skills. To date, CNN classifiers were widely applied to assist the early diagnosis of AD and its associated abnormal structures. However, most existing black-box CNN classifiers relied heavily on the limited MRI scans, and used little domain knowledge from the previous clinical findings. In this study, we proposed a framework, named as , to consider the previous domain knowledge as a , and open the black-box in the prediction process. The input domain knowledge guides the neural network to learn representative features and introduced intepretability for further analysis. used a Transformer-like fusion module to iteratively calculate the correlation of the features between image features and PI features, and project the features into a latent space for classification. The module served as a verification to highlight the significant features on the input images. was suitable for neuro-imaging tasks and we demonstrated its application in Alzheimer's Disease using structural MRI scans from ADNI dataset. During the experiments, we employed the abnormal brain structures such as the Hippocampus as the , trained the model with the data from 1.5T scanners and tested from 3T scanners. The F1-score showed that was more robust in transferring to a new dataset, with approximatedly 2% drop (from 0.9471 to 0.9231), while the baseline CNN methods had a 29% drop (from 0.8679 to 0.6154). The performance of was relied on the selection of the domain knowledge as the . Our best model was trained under the guidance of 12 selected ROIs, major in the structures of and . In summary, PINet considered the domain knowledge as the to train the CNN model, and the selected introduced both interpretability and generalization ability to the black box CNN classifiers.
阿尔茨海默病导致的不可逆性进行性萎缩致使思维和行为能力持续下降。迄今为止,卷积神经网络(CNN)分类器被广泛应用于辅助阿尔茨海默病的早期诊断及其相关异常结构的检测。然而,大多数现有的黑箱CNN分类器严重依赖有限的磁共振成像(MRI)扫描,且很少利用先前临床发现中的领域知识。在本研究中,我们提出了一个名为PINet的框架,将先前的领域知识视为先验信息,并在预测过程中打开黑箱。输入的领域知识引导神经网络学习代表性特征,并为进一步分析引入可解释性。PINet使用类似Transformer的融合模块迭代计算图像特征和先验信息(PI)特征之间的相关性,并将特征投影到潜在空间进行分类。该融合模块作为一种验证手段,突出输入图像上的显著特征。PINet适用于神经成像任务,我们使用阿尔茨海默病神经成像计划(ADNI)数据集的结构MRI扫描展示了其在阿尔茨海默病中的应用。在实验过程中,我们将海马体等异常脑结构作为先验信息,使用来自1.5T扫描仪的数据训练模型,并在3T扫描仪上进行测试。F1分数表明,PINet在转移到新数据集时更稳健,下降幅度约为2%(从0.9471降至0.9231),而基线CNN方法下降了29%(从0.8679降至0.6154)。PINet的性能依赖于作为先验信息的领域知识的选择。我们的最佳模型是在12个选定的感兴趣区域(ROI)的指导下训练的,主要集中在海马体和内嗅皮质的结构上。总之,PINet将领域知识视为先验信息来训练CNN模型,所选的先验信息为黑箱CNN分类器引入了可解释性和泛化能力。