摘要
本发明公开了一种基于预训练和图神经网络的功能脑网络分类方法,包括步骤:1)获取fMRI数据,并进行预处理;2)对fMRI数据进行大脑区域划分并提取特征,以图的形式构建功能脑网络;3)将无标签的的功能脑网络输入节点编码层训练;4)通过节点信息聚合网络训练;5)将步骤3)和4)的输出通过边关系预测网络训练;6)将带标签的功能脑网络数据输入步骤3)训练后的节点编码层训练;7)在步骤4)训练后的节点信息聚合网络中进行训练;8)通过功能脑网络分类模型,进行训练和分类。本发明利用大量无标签脑网络数据,通过对图神经网络的预训练,使完成预训练的网络只需在少量标签数据上训练即可适应下游的功能脑网络分类任务。
- 单位