DEEP FMRI: AN END-TO-END DEEP NETWORK FOR CLASSIFICATION OF FMRI DATA

With recent advancements in machine learning, the research community has made tremendous advances towards the classification of neurological disorders from time-series functional MRI signals. However, existing classification techniques rely on hand-crafted features and classical machine learning models. In this paper, we propose an end-to-end model that utilizes the representation learning capability of deep learning to classify a neurological disorder from fMRI data. The proposed DeepFMRI model is comprised of three networks, namely (1) a feature extractor, (2) a similarity network, and (3) a classification network. The model takes fMRI raw time-series signals as input and outputs the predicted labels; and is trained end-to-end using back-propagation. Experimental results on the publicly available ADHD-200 dataset demonstrate that this innovative model outperforms previous state-of-the-art.