用时序扩散模型生成脑影像时间序列,提升抑郁症诊断准确率
fMRI-Diffusion: Generating fMRI Time Series Via a Temporal Transformer Diffusion Model for Major Depressive Disorder Diagnosis

- 用时序Transformer在扩散模型中生成区域时间序列,保留动态信息
- 在多个数据集上提升诊断准确率,最高增3.7个百分点
- 适合小样本医疗数据增强,尤其对抑郁症研究有实用价值
从功能性磁共振成像(fMRI)诊断重度抑郁症(MDD)依赖大量标注数据,但临床中数据稀缺。现有增强方法仅合成功能连接(FC)矩阵,将fMRI压缩为静态成对关系,丢失时间信息。本文提出fMRI-Diffusion框架,直接生成兴趣区(ROI)层面的fMRI时间序列,而非FC矩阵。采用时序Transformer作为去噪网络,将每个时间点视为一个令牌,通过自注意力捕捉时间依赖性。引入监督预训练策略,在扩散训练前注入任务相关表示;合成的时间序列再用于计算FC矩阵进行分类。在REST-meta-MDD数据集上的实验表明,使用合成时间序列增强训练数据,可稳定提升十种分类器、六种脑图谱分区和三个采集站点的诊断准确率。相比五种近期基于FC的合成方法,本方法最高提升3.7个百分点。消融实验验证了Transformer去噪器和预训练策略的有效性。所有条件下分布保真度均低于0.06,表明真实与合成分布高度一致。结果表明,在计算FC前生成时间序列,能有效保留被丢弃的时序信息,为小样本下的MDD诊断提供可行方案。
原文摘要 · Abstract (English)
Diagnosing Major Depressive Disorder (MDD) from functional magnetic resonance imaging (fMRI) using functional connectivity (FC) analysis requires large amounts of labeled data that are scarce in clinical settings. Existing augmentation methods synthesize FC matrices, which compress fMRI recordings into static pairwise summaries and discard temporal information. We propose fMRI-Diffusion, a framework that synthesizes region-of-interest (ROI)-level fMRI time series rather than FC matrices. A Temporal Transformer serves as the denoising network within a denoising diffusion probabilistic model, treating each time point as a token to capture temporal dependencies through self-attention. A supervised pretraining strategy initializes the Transformer with task-relevant representations before diffusion training, and FC matrices are derived from the synthesized time series for classification. Experiments on the REST-meta-MDD dataset show that augmenting training data with synthetic time series consistently improves diagnostic accuracy across ten classifiers, six parcellation atlases, and three acquisition sites. The method outperforms five recent FC-based synthesis approaches, with accuracy gains of up to 3.7 percentage points over the strongest baseline. Ablation studies confirm the contributions of both the Transformer-based denoiser and the pretraining strategy. Distributional fidelity metrics remain below 0.06 across all conditions, indicating close agreement between real and synthetic distributions. These findings suggest that synthesizing fMRI time series before FC computation preserves temporal information lost in matrix-level augmentation and provides a practical strategy for MDD diagnosis under limited data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。