轻量微调方法让小数据也能精准分割海马体,助力阿尔茨海默病早期诊断。
A Novel Hybrid Parameter-Efficient Fine-Tuning Approach for Hippocampus Segmentation and Alzheimer's Disease Diagnosis
- 采用混合串并行结构,只更新少量参数实现高效微调。
- 在少样本下分割精度超越基线,对AD与正常人分类准确率达83.78%。
- 适合资源有限的医疗场景,尤其适用于海马体分析与神经退行性疾病筛查。
深度学习显著推进了医学图像分割,但其成功依赖大量人工标注数据,且需大量计算资源,尤其在三维医学影像任务中。针对标注数据少、计算资源受限的挑战,本文提出一种新型参数高效微调方法HyPS,采用混合串并行架构,仅更新极小部分模型参数,保留预训练模型原始结构的同时增强其对下游任务特征的学习能力。将该方法应用于SwinUNETR模型,在BraTs2021上预训练后,迁移至三个海马体数据集。实验表明,HyPS在少样本场景下表现优于基线方法。基于分割结果,计算了ADNI数据集中受试者的海马体体积,并结合元数据进行疾病分类:区分阿尔茨海默病(AD)与认知正常(CN)个体的准确率为83.78%,区分早期轻度认知障碍(EMCI)与晚期轻度认知障碍(LMCI)的准确率为64.29%。结果表明,HyPS不仅可有效实现海马体分割,还具备辅助阿尔茨海默病检测潜力。代码已公开。
原文摘要 · Abstract (English)
Deep learning methods have significantly advanced medical image segmentation, yet their success hinges on large volumes of manually annotated data, which require specialized expertise for accurate labeling. Additionally, these methods often demand substantial computational resources, particularly for three-dimensional medical imaging tasks. Consequently, applying deep learning techniques for medical image segmentation with limited annotated data and computational resources remains a critical challenge. In this paper, we propose a novel parameter-efficient fine-tuning strategy, termed HyPS, which employs a hybrid parallel and serial architecture. HyPS updates a minimal subset of model parameters, thereby retaining the pre-trained model's original knowledge tructure while enhancing its ability to learn specific features relevant to downstream tasks. We apply this strategy to the state-of-the-art SwinUNETR model for medical image segmentation. Initially, the model is pre-trained on the BraTs2021 dataset, after which the HyPS method is employed to transfer it to three distinct hippocampus datasets.Extensive experiments demonstrate that HyPS outperforms baseline methods, especially in scenarios with limited training samples. Furthermore, based on the segmentation results, we calculated the hippocampal volumes of subjects from the ADNI dataset and combined these with metadata to classify disease types. In distinguishing Alzheimer's disease (AD) from cognitively normal (CN) individuals, as well as early mild cognitive impairment (EMCI) from late mild cognitive impairment (LMCI), HyPS achieved classification accuracies of 83.78% and 64.29%, respectively. These findings indicate that the HyPS method not only facilitates effective hippocampal segmentation using pre-trained models but also holds potential for aiding Alzheimer's disease detection. Our code is publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。