ISSN 1000-1239 CN 11-1777/TP

Journal of Computer Research and Development ›› 2020, Vol. 57 ›› Issue (7): 1449-1459.doi: 10.7544/issn1000-1239.2020.20190186

Previous Articles     Next Articles

Brain Networks Classification Based on an Adaptive Multi-Task Convolutional Neural Networks

Xing Xinying, Ji Junzhong, Yao Yao   

  1. (College of Computer, Faculty of Information Technology, Beijing University of Technology, Beijing 100124)
  • Online:2020-07-01
  • Supported by: 
    This work was supported by the National Natural Science Foundation of China (61672065).

Abstract: Brain networks classification is an important subject in brain science. In recent years, brain networks classification based on convolutional neural networks has become a hot topic. However, it is still difficult to accurately classify brain network data with high dimension and small sample size. Due to the close relationship between different clinical phenotypes and brain networks of different populations, it is highly possible to provide auxiliary information for the brain networks classification. Therefore, we propose a new brain networks classification method based on an adaptive multi-task convolutional neural network in this paper. Firstly, the clinical phenotype predictions are introduced as different auxiliary tasks and the shared representation mechanism of multi-task convolutional neural networks is used to provide general and useful information for brain networks classification. Then, in order to reduce the experimental cost and the error caused by the manual operation, a new adaptive method is proposed to substitute for manual adjustments of the weight of every task in the multi-task learning. The experimental results on the autism brain imaging data exchange I (ABIDE I) dataset show that the multi-task convolutional neural networks which introduce clinical phenotype predictions can achieve better classification results. Moreover, the adaptive multi-task learning method can further improve the performance of brain networks classification.

Key words: brain networks classification, convolutional neural network, multi-task learning, clinical phenotype, adaptive method

CLC Number: