所有AI工具AI学习网站AI开发框架AI开源项目

brain_state_pred开源 – 预测未来脑状态的Transformer模型

该项目是基于论文《Predicting Human Brain States with Transformer》的代码实现,旨在通过时间序列Transformer模型预测人类脑状态。模型使用功能性磁共振成像(fMRI)数据,训练...

标签:

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. 用于神经科学研究,探索脑状态预测和功能连接组分析。

相关导航

暂无评论

暂无评论...