arXiv:2410.00946eess.IVcs.LG2024-10

用图谱特征加权样本,让神经影像预测模型更懂不同人群差异。

Spectral Graph Sample Weighting for Interpretable Sub-cohort Analysis in Predictive Models for Neuroimaging

  • 基于图谱特征的样本权重,平滑反映个体差异
  • 在NCANDA和ADNI数据上提升模型可解释性
  • 适合关注群体异质性的临床研究者

脑部疾病常包含多种机制、发展轨迹或严重程度子类型,这些异质性往往与性别等人口学因素或遗传等疾病相关因素有关。因此,机器学习模型在不同受试者间的预测能力存在差异。为建模这种异质性,我们为每个训练样本分配一个依赖于因子的权重,调节其对整体损失函数的贡献。为此,我们提出将样本权重建模为捕捉受试者间因子相似性的谱群体图的特征基的线性组合。由此,学习到的权重在图上平滑变化,突出高/低可预测性子队列。该方法在两个任务上进行评估:一是基于NCANDA数据集中的影像与神经心理学指标预测青少年期重度饮酒启动;二是基于ADNI数据集中的影像与人口统计学指标区分痴呆与轻度认知障碍。相比现有样本加权方案,本方法提升了可解释性,并识别出具有显著特征差异及模型准确率不同的子队列。

原文摘要 · Abstract (English)

Recent advancements in medicine have confirmed that brain disorders often comprise multiple subtypes of mechanisms, developmental trajectories, or severity levels. Such heterogeneity is often associated with demographic aspects (e.g., sex) or disease-related contributors (e.g., genetics). Thus, the predictive power of machine learning models used for symptom prediction varies across subjects based on such factors. To model this heterogeneity, one can assign each training sample a factor-dependent weight, which modulates the subject's contribution to the overall objective loss function. To this end, we propose to model the subject weights as a linear combination of the eigenbases of a spectral population graph that captures the similarity of factors across subjects. In doing so, the learned weights smoothly vary across the graph, highlighting sub-cohorts with high and low predictability. Our proposed sample weighting scheme is evaluated on two tasks. First, we predict initiation of heavy alcohol drinking in young adulthood from imaging and neuropsychological measures from the National Consortium on Alcohol and NeuroDevelopment in Adolescence (NCANDA). Next, we detect Dementia vs. Mild Cognitive Impairment (MCI) using imaging and demographic measurements in subjects from the Alzheimer's Disease Neuroimaging Initiative (ADNI). Compared to existing sample weighting schemes, our sample weights improve interpretability and highlight sub-cohorts with distinct characteristics and varying model accuracy.

神经影像子队列分析可解释性图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。