论文标题
基于MRI的多任务脱钩学习,用于阿尔茨海默氏病检测和MMSE评分预测:多站点验证
MRI-based Multi-task Decoupling Learning for Alzheimer's Disease Detection and MMSE Score Prediction: A Multi-site Validation
论文作者
论文摘要
准确地检测阿尔茨海默氏病(AD)和预测小精神状态检查(MMSE)得分是通过磁共振成像(MRI)在老年人健康方面的重要任务。这两个任务的先前方法中的大多数方法都是基于单任务学习,并且很少考虑它们之间的相关性。由于MMSE得分是AD诊断的重要基础,也可以反映认知障碍的进步,因此一些研究已开始将多任务学习方法应用于这两个任务。但是,如何利用特征相关性仍然是这些方法的挑战性问题。为了全面解决这一挑战,我们提出了一种基于MRI的多任务脱钩学习方法,用于AD检测和MMSE得分预测。首先,提出了一个多任务学习网络来实现AD检测和MMSE得分预测,该预测通过在两个任务的骨架之间添加三个多任务相互作用层来利用特征相关性。每个多任务相互作用层包含两个特征解耦模块和一个特征交互模块。此外,为了增强特征解耦模块选择的特征的任务之间的概括,我们提出了特征一致性损失约束特征解耦模块。最后,为了利用不同组中MMSE得分的特定分布信息,提出了分配损失以进一步提高模型性能。我们在多站点数据集上评估了我们提出的方法。实验结果表明,我们提出的多任务脱钩表示方法可实现良好的性能,优于单件任务学习和其他现有的最新方法。
Accurately detecting Alzheimer's disease (AD) and predicting mini-mental state examination (MMSE) score are important tasks in elderly health by magnetic resonance imaging (MRI). Most of the previous methods on these two tasks are based on single-task learning and rarely consider the correlation between them. Since the MMSE score, which is an important basis for AD diagnosis, can also reflect the progress of cognitive impairment, some studies have begun to apply multi-task learning methods to these two tasks. However, how to exploit feature correlation remains a challenging problem for these methods. To comprehensively address this challenge, we propose a MRI-based multi-task decoupled learning method for AD detection and MMSE score prediction. First, a multi-task learning network is proposed to implement AD detection and MMSE score prediction, which exploits feature correlation by adding three multi-task interaction layers between the backbones of the two tasks. Each multi-task interaction layer contains two feature decoupling modules and one feature interaction module. Furthermore, to enhance the generalization between tasks of the features selected by the feature decoupling module, we propose the feature consistency loss constrained feature decoupling module. Finally, in order to exploit the specific distribution information of MMSE score in different groups, a distribution loss is proposed to further enhance the model performance. We evaluate our proposed method on multi-site datasets. Experimental results show that our proposed multi-task decoupled representation learning method achieves good performance, outperforming single-task learning and other existing state-of-the-art methods.