融合词向量与语言特征,提升阿尔茨海默病早期诊断准确率
A Robust Classification Method using Hybrid Word Embedding for Early Diagnosis of Alzheimer's Disease
- 用Doc2Vec与ELMo混合生成词嵌入,捕捉语义和流畅性
- 通过调参实现91%准确率和97%AUC,优于现有模型
- 模型稳定可靠,适合大规模筛查或医生辅助诊断
阿尔茨海默病(AD)早期检测对患者治疗和医疗负担缓解至关重要。语言能力变化是AD的显著征兆,可用于早期诊断。本文提出一种基于混合词嵌入与超参数微调的鲁棒分类方法,在早期AD检测中达到领先性能。具体而言,构建基于Doc2Vec和ELMo的混合词向量,计算句子困惑度以判断流畅性并捕捉语义上下文;进一步引入语法与语义语言特征增强嵌入表示;将嵌入特征输入逻辑回归模型,并在全管道中精细调整超参数(如正则化强度、学习率、Doc2Vec与ELMo的向量维度)。实验表明,该方法在区分早期AD患者与健康对照组时,分类准确率达91%,受试者工作特征曲线下面积(AUC)为97%。据作者所知,此性能超越现有最佳NLP模型(准确率88%)。通过重复实验验证,模型在随机数据划分下仍表现稳定(准确率标准差0.0403,AUC标准差0.0174),证明其可靠性。该模型可作为大规模筛查工具或医生辅助诊断手段。
原文摘要 · Abstract (English)
Early detection of Alzheimer's Disease (AD) is greatly beneficial to AD patients, leading to early treatments that lessen symptoms and alleviating financial burden of health care. As one of the leading signs of AD, language capability changes can be used for early diagnosis of AD. In this paper, I develop a robust classification method using hybrid word embedding and fine-tuned hyperparameters to achieve state-of-the-art accuracy in the early detection of AD. Specifically, we create a hybrid word embedding based on word vectors from Doc2Vec and ELMo to obtain perplexity scores of the sentences. The scores identify whether a sentence is fluent or not and capture semantic context of the sentences. I enrich the word embedding by adding linguistic features to analyze syntax and semantics. Further, we input an embedded feature vector into logistic regression and fine tune hyperparameters throughout the pipeline. By tuning hyperparameters of the machine learning pipeline (e.g., model regularization parameter, learning rate and vector size of Doc2Vec, and vector size of ELMo), I achieve 91% classification accuracy and an Area Under the Curve (AUC) of 97% in distinguishing early AD from healthy subjects. Based on my knowledge, my model with 91% accuracy and 97% AUC outperforms the best existing NLP model for AD diagnosis with an accuracy of 88% [32]. I study the model stability through repeated experiments and find that the model is stable even though the training data is split randomly (standard deviation of accuracy = 0.0403; standard deviation of AUC = 0.0174). This affirms our proposed method is accurate and stable. This model can be used as a large-scale screening method for AD, as well as a complementary examination for doctors to detect AD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。