An individual-level weighted artificial neural network method to improve the systematic bias in BrainAGE analysis

BrainAGE is a commonly used machine learning technique to measure the accelerated/delayed development pattern of human brain structure/function with neuropsychiatric disorders. However, recent studies have shown a systematic bias ("regression toward mean" effect) in the BrainAGE method, which indicates that the prediction error is not uniformly distributed across Chronological Ages: for the older individuals, the Brain Ages would be under-estimated but would be over-estimated for the younger individuals. In the present study, we propose an individual-level weighted artificial neural network method and apply it to simulation datasets (containing 5000 simulated subjects) and a real dataset (containing 135 subjects). Results show that compared with traditional machine learning methods, the individual-level weighted strategy can significantly reduce the "regression toward mean" effect, while the prediction performance can achieve the comparable level with traditional machine learning methods. Further analysis indicates that the sigmoid active function for artificial neural network shows better performance than the relu active function. The present study provides a novel strategy to reduce the "regression toward mean" effect of BrainAGE analysis, which is helpful to improve accuracy in exploring the atypical brain structure/function development pattern of neuropsychiatric disorders.