Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal

Objective: Functional magnetic resonance imaging data pose significant challenges due to their inherently noisy and complex nature, making traditional statistical models less effective in capturing predictive features. While deep learning models offer superior performance through their non-linear ca...

Full description

Saved in:
Bibliographic Details
Main Authors: Zafar Iqbal, Md. Mahfuzur Rahman, Usman Mahmood, Qasim Zia, Zening Fu, Vince D. Calhoun, Sergey Plis
Format: Article
Language:English
Published: MDPI AG 2025-01-01
Series:Brain Sciences
Subjects:
Online Access:https://www.mdpi.com/2076-3425/15/1/60
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1832588929022296064
author Zafar Iqbal
Md. Mahfuzur Rahman
Usman Mahmood
Qasim Zia
Zening Fu
Vince D. Calhoun
Sergey Plis
author_facet Zafar Iqbal
Md. Mahfuzur Rahman
Usman Mahmood
Qasim Zia
Zening Fu
Vince D. Calhoun
Sergey Plis
author_sort Zafar Iqbal
collection DOAJ
description Objective: Functional magnetic resonance imaging data pose significant challenges due to their inherently noisy and complex nature, making traditional statistical models less effective in capturing predictive features. While deep learning models offer superior performance through their non-linear capabilities, they often lack transparency, reducing trust in their predictions. This study introduces the Time Reversal (TR) pretraining method to address these challenges. TR aims to learn temporal dependencies in data, leveraging large datasets for pretraining and applying this knowledge to improve schizophrenia classification on smaller datasets. Methods: We pretrained an LSTM-based model with attention using the TR approach, focusing on learning the direction of time in fMRI data, achieving over 98 % accuracy on HCP and UK Biobank datasets. For downstream schizophrenia classification, TR-pretrained weights were transferred to models evaluated on FBIRN, COBRE, and B-SNIP datasets. Saliency maps were generated using Integrated Gradients (IG) to provide post hoc explanations for pretraining, while Earth Mover’s Distance (EMD) quantified the temporal dynamics of salient features in the downstream tasks. Results: TR pretraining significantly improved schizophrenia classification performance across all datasets: median AUC scores increased from 0.7958 to 0.8359 (FBIRN), 0.6825 to 0.7778 (COBRE), and 0.6341 to 0.7224 (B-SNIP). The saliency maps revealed more concentrated and biologically meaningful salient features along the time axis, aligning with the episodic nature of schizophrenia. TR consistently outperformed baseline pretraining methods, including OCP and PCL, in terms of AUC, balanced accuracy, and robustness. Conclusions: This study demonstrates the dual benefits of the TR method: enhanced predictive performance and improved interpretability. By aligning model predictions with meaningful temporal patterns in brain activity, TR bridges the gap between deep learning and clinical relevance. These findings emphasize the potential of explainable AI tools for aiding clinicians in diagnostics and treatment planning, especially in conditions characterized by disrupted temporal dynamics.
format Article
id doaj-art-bb3dd665ae444f2e8ca6e8ee8f8f1071
institution Kabale University
issn 2076-3425
language English
publishDate 2025-01-01
publisher MDPI AG
record_format Article
series Brain Sciences
spelling doaj-art-bb3dd665ae444f2e8ca6e8ee8f8f10712025-01-24T13:25:50ZengMDPI AGBrain Sciences2076-34252025-01-011516010.3390/brainsci15010060Explainable Self-Supervised Dynamic Neuroimaging Using Time ReversalZafar Iqbal0Md. Mahfuzur Rahman1Usman Mahmood2Qasim Zia3Zening Fu4Vince D. Calhoun5Sergey Plis6Department of Computer Science, Georgia State University, Atlanta, GA 30302, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USACenter for Translational Research in Neuroimaging and Data Science (TReNDS), Atlanta, GA 30303, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USAObjective: Functional magnetic resonance imaging data pose significant challenges due to their inherently noisy and complex nature, making traditional statistical models less effective in capturing predictive features. While deep learning models offer superior performance through their non-linear capabilities, they often lack transparency, reducing trust in their predictions. This study introduces the Time Reversal (TR) pretraining method to address these challenges. TR aims to learn temporal dependencies in data, leveraging large datasets for pretraining and applying this knowledge to improve schizophrenia classification on smaller datasets. Methods: We pretrained an LSTM-based model with attention using the TR approach, focusing on learning the direction of time in fMRI data, achieving over 98 % accuracy on HCP and UK Biobank datasets. For downstream schizophrenia classification, TR-pretrained weights were transferred to models evaluated on FBIRN, COBRE, and B-SNIP datasets. Saliency maps were generated using Integrated Gradients (IG) to provide post hoc explanations for pretraining, while Earth Mover’s Distance (EMD) quantified the temporal dynamics of salient features in the downstream tasks. Results: TR pretraining significantly improved schizophrenia classification performance across all datasets: median AUC scores increased from 0.7958 to 0.8359 (FBIRN), 0.6825 to 0.7778 (COBRE), and 0.6341 to 0.7224 (B-SNIP). The saliency maps revealed more concentrated and biologically meaningful salient features along the time axis, aligning with the episodic nature of schizophrenia. TR consistently outperformed baseline pretraining methods, including OCP and PCL, in terms of AUC, balanced accuracy, and robustness. Conclusions: This study demonstrates the dual benefits of the TR method: enhanced predictive performance and improved interpretability. By aligning model predictions with meaningful temporal patterns in brain activity, TR bridges the gap between deep learning and clinical relevance. These findings emphasize the potential of explainable AI tools for aiding clinicians in diagnostics and treatment planning, especially in conditions characterized by disrupted temporal dynamics.https://www.mdpi.com/2076-3425/15/1/60interpretabilityexplainabilityschizophreniafMRIpretrainingself-supervised
spellingShingle Zafar Iqbal
Md. Mahfuzur Rahman
Usman Mahmood
Qasim Zia
Zening Fu
Vince D. Calhoun
Sergey Plis
Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
Brain Sciences
interpretability
explainability
schizophrenia
fMRI
pretraining
self-supervised
title Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
title_full Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
title_fullStr Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
title_full_unstemmed Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
title_short Explainable Self-Supervised Dynamic Neuroimaging Using Time Reversal
title_sort explainable self supervised dynamic neuroimaging using time reversal
topic interpretability
explainability
schizophrenia
fMRI
pretraining
self-supervised
url https://www.mdpi.com/2076-3425/15/1/60
work_keys_str_mv AT zafariqbal explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT mdmahfuzurrahman explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT usmanmahmood explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT qasimzia explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT zeningfu explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT vincedcalhoun explainableselfsuperviseddynamicneuroimagingusingtimereversal
AT sergeyplis explainableselfsuperviseddynamicneuroimagingusingtimereversal