Warning: Undefined variable $main_title in /www/wwwroot/msbd/wp-content/themes/onenav/sidebar-sitestop.php on line 26
AI交流(进群备注:)

该项目是基于论文《Predicting Human Brain States with Transformer》的代码实现,旨在通过时间序列Transformer模型预测人类脑状态。模型使用功能性磁共振成像(fMRI)数据,训练于人类连接组项目(HCP)的高质量静息态fMRI数据上,能够基于21.6秒的fMRI数据,准确预测未来约5.04秒的脑状态。项目支持单状态预测和时间序列预测,评估指标包括均方误差(MSE)和Spearman相关系数。
brain_state_pred的特点:
- 1. 使用4个编码层和4个解码层的时间序列Transformer,每个层有8个注意力头。
- 2. 训练数据来自HCP,包含893个受试者。
- 3. 输入窗口为30个时间点(21.6秒),预测下一个时间点。
- 4. 支持单状态预测和时间序列预测,评估指标包括MSE和Spearman相关系数。
brain_state_pred的功能:
- 1. 克隆GitHub仓库并安装依赖,按照README文件配置环境。
- 2. 运行训练脚本,使用HCP数据训练Transformer模型。
- 3. 运行预测脚本,对新数据进行预测并评估模型性能。
- 4. 用于神经科学研究,探索脑状态预测和功能连接组分析。
相关导航
暂无评论...