The current version is preliminary. It's worth noting that certain terminology used within the code does not precisely match the terminology used in our paper. For example, in the code, "BrainMAE" corresponds to the "fMRIAutoEncoder" class, and the "TSE module" corresponds to the "MaskedTransientStateNet." However, the results presented in our paper are reproducible using the provided code.

To enhance clarity and consistency, we are actively working on reorganizing the code. Once completed, we will provide an updated version through an anonymous URL. The updated code will align well with the terminology defined in our submitted paper.