arXiv:2508.11732cs.LGcs.AI2025-08

用类脑强化学习自动优化脑连接模型,提升精神疾病分类准确率

BRIEF: BRain-Inspired network connection search with Extensive temporal feature Fusion enhances disease classification

  • 基于类脑强化学习动态搜索最优网络连接结构
  • 在精神疾病分类中达91.5%与78.4%的AUC,优于21种模型
  • 适合关注脑影像分析与可解释性建模的研究者

基于功能磁共振成像的精神疾病分类现有模型存在网络结构依赖经验设计、特征融合方式单一(多为简单拼接)的问题。受人类大脑通过学习更新神经连接机制启发,本文提出一种新型脑启发特征融合框架BRIEF,通过改进的神经网络连接搜索策略与基于Transformer的多特征融合模块,实现网络架构自动优化。具体地,从fMRI数据中提取四类时间特征:时序信号(TCs)、静态/动态功能连接(FNC/dFNC)及多尺度熵(MsDE),构建四个编码器。每个编码器采用改进的Q-learning算法,将连接搜索建模为马尔可夫决策过程,动态优化以提取高级特征向量。所有特征向量通过Transformer融合,利用稳定与时变连接及跨脑区多尺度依赖关系完成最终分类,并嵌入注意力模块增强可解释性。在区分精神分裂症(SZ, n=1100)与自闭症谱系障碍(ASD, n=1550)患者与健康对照的对比实验中,BRIEF相较21种先进模型提升2.2%至12.1%,分别达到91.5%±0.6%和78.4%±0.5%的AUC。该研究首次将类脑强化学习引入fMRI疾病分类,展现出精准神经影像生物标志物识别的巨大潜力。

原文摘要 · Abstract (English)

Existing deep learning models for functional MRI-based classification have limitations in network architecture determination (relying on experience) and feature space fusion (mostly simple concatenation, lacking mutual learning). Inspired by the human brain's mechanism of updating neural connections through learning and decision-making, we proposed a novel BRain-Inspired feature Fusion (BRIEF) framework, which is able to optimize network architecture automatically by incorporating an improved neural network connection search (NCS) strategy and a Transformer-based multi-feature fusion module. Specifically, we first extracted 4 types of fMRI temporal representations, i.e., time series (TCs), static/dynamic functional connection (FNC/dFNC), and multi-scale dispersion entropy (MsDE), to construct four encoders. Within each encoder, we employed a modified Q-learning to dynamically optimize the NCS to extract high-level feature vectors, where the NCS is formulated as a Markov Decision Process. Then, all feature vectors were fused via a Transformer, leveraging both stable/time-varying connections and multi-scale dependencies across different brain regions to achieve the final classification. Additionally, an attention module was embedded to improve interpretability. The classification performance of our proposed BRIEF was compared with 21 state-of-the-art models by discriminating two mental disorders from healthy controls: schizophrenia (SZ, n=1100) and autism spectrum disorder (ASD, n=1550). BRIEF demonstrated significant improvements of 2.2% to 12.1% compared to 21 algorithms, reaching an AUC of 91.5% - 0.6% for SZ and 78.4% - 0.5% for ASD, respectively. This is the first attempt to incorporate a brain-inspired, reinforcement learning strategy to optimize fMRI-based mental disorder classification, showing significant potential for identifying precise neuroimaging biomarkers.

脑启发fMRI分类强化学习

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