提出DEAL模型提升阿尔茨海默病早期诊断的群体鲁棒性
DEAL: Decoupled Classifier with Adaptive Linear Modulation for Group Robust Early Diagnosis of MCI to AD Conversion
- 分离特征与分类器,用年龄和认知指标自适应调制特征
- 在不同人群间显著提升诊断准确率,尤其改善了特定组别表现
- 适合关注临床群体差异的医疗AI研究者使用
尽管基于深度学习的阿尔茨海默病(AD)诊断在预测轻度认知障碍(MCI)向AD转化方面取得进展,但对诊断结果在不同群体间的鲁棒性仍缺乏系统研究。现有方法常依赖无关特征,导致某些群体性能下降。本文首次系统研究了利用MRI图像进行MCI转AD早期诊断中的群体鲁棒性问题,重点关注按年龄划分的sMCI与pMCI人群之间的准确率差异。实验发现,标准分类器在不同架构下均对特定群体表现不佳,凸显定制化方法的必要性。为此,提出DEAL(Decoupled Classifier with Adaptive Linear Modulation)模型,包含两个核心组件:(1) 对倒数第二层特征进行线性调制,融合可获取的年龄与认知指标表格式特征;(2) 解耦分类器,为各群体提供更个性化的决策边界。通过多架构实验验证,DEAL显著提升了MCI转AD预测的群体鲁棒性。
原文摘要 · Abstract (English)
While deep learning-based Alzheimer's disease (AD) diagnosis has recently made significant advancements, particularly in predicting the conversion of mild cognitive impairment (MCI) to AD based on MRI images, there remains a critical gap in research regarding the group robustness of the diagnosis. Although numerous studies pointed out that deep learning-based classifiers may exhibit poor performance in certain groups by relying on unimportant attributes, this issue has been largely overlooked in the early diagnosis of MCI to AD conversion. In this paper, we present the first comprehensive investigation of the group robustness in the early diagnosis of MCI to AD conversion using MRI images, focusing on disparities in accuracy between groups, specifically sMCI and pMCI individuals divided by age. Our experiments reveal that standard classifiers consistently underperform for certain groups across different architectures, highlighting the need for more tailored approaches. To address this, we propose a novel method, dubbed DEAL (DEcoupled classifier with Adaptive Linear modulation), comprising two key components: (1) a linear modulation of features from the penultimate layer, incorporating easily obtainable age and cognitive indicative tabular features, and (2) a decoupled classifier that provides more tailored decision boundaries for each group, further improving performance. Through extensive experiments and evaluations across different architectures, we demonstrate the efficacy of DEAL in improving the group robustness of the MCI to AD conversion prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。