Predicting Human Brain States with Transformer

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Sun, Yifei, Cabezas, Mariano, Lee, Jiah, Wang, Chenyu, Zhang, Wei, Calamante, Fernando, Lv, Jinglei
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910765037912064
author Sun, Yifei
Cabezas, Mariano
Lee, Jiah
Wang, Chenyu
Zhang, Wei
Calamante, Fernando
Lv, Jinglei
author_facet Sun, Yifei
Cabezas, Mariano
Lee, Jiah
Wang, Chenyu
Zhang, Wei
Calamante, Fernando
Lv, Jinglei
contents The human brain is a complex and highly dynamic system, and our current knowledge of its functional mechanism is still very limited. Fortunately, with functional magnetic resonance imaging (fMRI), we can observe blood oxygen level-dependent (BOLD) changes, reflecting neural activity, to infer brain states and dynamics. In this paper, we ask the question of whether the brain states rep-resented by the regional brain fMRI can be predicted. Due to the success of self-attention and the transformer architecture in sequential auto-regression problems (e.g., language modelling or music generation), we explore the possi-bility of the use of transformers to predict human brain resting states based on the large-scale high-quality fMRI data from the human connectome project (HCP). Current results have shown that our model can accurately predict the brain states up to 5.04s with the previous 21.6s. Furthermore, even though the prediction error accumulates for the prediction of a longer time period, the gen-erated fMRI brain states reflect the architecture of functional connectome. These promising initial results demonstrate the possibility of developing gen-erative models for fMRI data using self-attention that learns the functional or-ganization of the human brain. Our code is available at: https://github.com/syf0122/brain_state_pred.
format Preprint
id arxiv_https___arxiv_org_abs_2412_19814
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Predicting Human Brain States with Transformer
Sun, Yifei
Cabezas, Mariano
Lee, Jiah
Wang, Chenyu
Zhang, Wei
Calamante, Fernando
Lv, Jinglei
Neurons and Cognition
Artificial Intelligence
Machine Learning
The human brain is a complex and highly dynamic system, and our current knowledge of its functional mechanism is still very limited. Fortunately, with functional magnetic resonance imaging (fMRI), we can observe blood oxygen level-dependent (BOLD) changes, reflecting neural activity, to infer brain states and dynamics. In this paper, we ask the question of whether the brain states rep-resented by the regional brain fMRI can be predicted. Due to the success of self-attention and the transformer architecture in sequential auto-regression problems (e.g., language modelling or music generation), we explore the possi-bility of the use of transformers to predict human brain resting states based on the large-scale high-quality fMRI data from the human connectome project (HCP). Current results have shown that our model can accurately predict the brain states up to 5.04s with the previous 21.6s. Furthermore, even though the prediction error accumulates for the prediction of a longer time period, the gen-erated fMRI brain states reflect the architecture of functional connectome. These promising initial results demonstrate the possibility of developing gen-erative models for fMRI data using self-attention that learns the functional or-ganization of the human brain. Our code is available at: https://github.com/syf0122/brain_state_pred.
title Predicting Human Brain States with Transformer
topic Neurons and Cognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2412.19814