arXiv:2410.14207cs.LG2024-10被引 2

用灵活加权机制提升阿尔茨海默病诊断模型抗噪与不平衡能力

Flexi-Fuzz least squares SVM for Alzheimer's diagnosis: Tackling noise, outliers, and class imbalance

  • 设计新型柔性隶属度机制,动态分配样本权重
  • 在多个数据集上准确率超基线模型,噪声下仍保持稳定
  • 适合医疗影像分类中存在异常值和类别不均衡的场景

阿尔茨海默病(AD)是导致痴呆的主要神经退行性疾病,以认知功能渐进性下降和记忆丧失为特征,其进展表现为大脑皮层萎缩,且不可逆。已有多种机器学习方法用于早期诊断,但常受噪声、离群点和类别不平衡问题影响。本文提出一种新型鲁棒灵活隶属度方案Flexi-Fuzz,融合灵活加权机制、类别概率与不平衡比率。该加权机制在中心邻近区域赋予最大权重,并在阈值外逐步降低,确保边界样本仍具影响力;类别概率用于抑制噪声样本干扰,不平衡比率缓解类别失衡。将该方案嵌入最小二乘支持向量机(LSSVM)框架,构建出柔韧鲁棒的Flexi-Fuzz-LSSVM模型。采用均值与中位数两种方式确定类别中心,形成两个变体:Flexi-Fuzz-LSSVM-I 和 Flexi-Fuzz-LSSVM-II。在标准UCI与KEEL数据集(含/不含标签噪声)及阿尔茨海默病神经影像计划(ADNI)数据集上进行验证。实验结果表明,该模型显著优于基线模型。

原文摘要 · Abstract (English)

Alzheimer's disease (AD) is a leading neurodegenerative condition and the primary cause of dementia, characterized by progressive cognitive decline and memory loss. Its progression, marked by shrinkage in the cerebral cortex, is irreversible. Numerous machine learning algorithms have been proposed for the early diagnosis of AD. However, they often struggle with the issues of noise, outliers, and class imbalance. To tackle the aforementioned limitations, in this article, we introduce a novel, robust, and flexible membership scheme called Flexi-Fuzz. This scheme integrates a novel flexible weighting mechanism, class probability, and imbalance ratio. The proposed flexible weighting mechanism assigns the maximum weight to samples within a specific proximity to the center, with a gradual decrease in weight beyond a certain threshold. This approach ensures that samples near the class boundary still receive significant weight, maintaining their influence in the classification process. Class probability is used to mitigate the impact of noisy samples, while the imbalance ratio addresses class imbalance. Leveraging this, we incorporate the proposed Flexi-Fuzz membership scheme into the least squares support vector machines (LSSVM) framework, resulting in a robust and flexible model termed Flexi-Fuzz-LSSVM. We determine the class-center using two methods: the conventional mean approach and an innovative median approach, leading to two model variants, Flexi-Fuzz-LSSVM-I and Flexi-Fuzz-LSSVM-II. To validate the effectiveness of the proposed Flexi-Fuzz-LSSVM models, we evaluated them on benchmark UCI and KEEL datasets, both with and without label noise. Additionally, we tested the models on the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset for AD diagnosis. Experimental results demonstrate the superiority of the Flexi-Fuzz-LSSVM models over baseline models.

阿尔茨海默病支持向量机分类优化医学诊断

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