brain_state_pred开源 – 预测未来脑状态的Transformer模型
该项目是基于论文《Predicting Human Brain States with Transformer》的代码实现,旨在通过时间序列Transformer模型预测人类脑状态。模型使用功能性磁共振成像(fMRI)数据,训练于人类连接组项目(HCP)的高质量静息态fMRI数据上,能够基于21.6秒的fMRI数据,准确预测未来约5.04秒的脑状态。项目支持单状态预测和时间序列预测,评估指标包括均方误差(MSE)和Spearman相关系数。