arXiv:2409.12304cs.CV2024-09被引 11

用自监督预训练提升脑影像模型,随机掩码区域比时间点效果更好。

Self-Supervised Pre-training Tasks for an fMRI Time-series Transformer in Autism Detection

  • 直接处理时间序列fMRI,不依赖功能连接计算。
  • 区域掩码使AUC提升10.8%,准确率提升9.3%。
  • 适合小样本的自闭症分类研究者参考。

自闭症谱系障碍(ASD)是一种神经发育障碍,症状多样且严重程度不一,诊断与治疗困难。功能性磁共振成像(fMRI)广泛用于研究ASD中的脑活动,机器学习方法也常用于分析静息态fMRI(rs-fMRI)数据。然而,基于Transformer的模型在该领域的应用仍较少。鉴于Transformer在捕捉序列数据长程依赖方面的优势,本文提出一种直接分析时间序列fMRI数据的Transformer自监督框架,无需计算功能连接。为缓解小样本下的过拟合问题并提升性能,设计了重建被随机掩码的时间序列数据的自监督预训练任务,并探究不同掩码策略的影响。随后在两个公开数据集上进行微调,使用五折交叉验证,评估不同训练数据量下的表现。实验表明,随机掩码整个脑区(ROI)的预训练策略优于随机掩码时间点,相比从零训练的Transformer模型,在不同数据量下平均提高AUC 10.8%、受试者准确率9.3%。代码已开源。

原文摘要 · Abstract (English)

Autism Spectrum Disorder (ASD) is a neurodevelopmental condition that encompasses a wide variety of symptoms and degrees of impairment, which makes the diagnosis and treatment challenging. Functional magnetic resonance imaging (fMRI) has been extensively used to study brain activity in ASD, and machine learning methods have been applied to analyze resting state fMRI (rs-fMRI) data. However, fewer studies have explored the recent transformer-based models on rs-fMRI data. Given the superiority of transformer models in capturing long-range dependencies in sequence data, we have developed a transformer-based self-supervised framework that directly analyzes time-series fMRI data without computing functional connectivity. To address over-fitting in small datasets and enhance the model performance, we propose self-supervised pre-training tasks to reconstruct the randomly masked fMRI time-series data, investigating the effects of various masking strategies. We then finetune the model for the ASD classification task and evaluate it using two public datasets and five-fold cross-validation with different amounts of training data. The experiments show that randomly masking entire ROIs gives better model performance than randomly masking time points in the pre-training step, resulting in an average improvement of 10.8% for AUC and 9.3% for subject accuracy compared with the transformer model trained from scratch across different levels of training data availability. Our code is available on GitHub.

自闭症检测fMRITransformer自监督学习

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