Brain age prediction method based on deep convolutional generative adversarial network / 医疗卫生装备

ABSTRACT

Objective To propose a brain age prediction method based on deep convolutional generative adversarial networks(DCGAN)for objective assessment of brain health status.Methods The DCGAN model was extended from 2D to 3D and improved by integrating the concept of residual block to enhance the ability for feature extraction.The classifiers were pre-trained with unsupervised adversarial learning and fine-tuned with migration learning to eliminate the overfitting of 3D convolutional neural network(CNN)due to small sample size.To verify the effectiveness of the improved model,comparison analyses based on UK Biobank(UKB)database were carried out between the improved model and least absolute shrinkage and selection operator(LASSO)model,machine learning model,3D CNN model and graph convolutional network model by using mean absolute error(MAE)as the evaluation metric.Results The model proposed gained advantages over LASSO model,machine learning model,3D CNN model and graph convolutional network model in predicting brain age with a MAE error of 2.896 years.Conclusion The method proposed behaves well for large-scale datasets,which can predict brain age accurately and assess brain health status objectively.