Correlation Constraints for Regression Models: Controlling Bias in Brain Age Prediction.

Journal: Frontiers in psychiatry

Volume: 12

Issue: 

Year of Publication: 

Affiliated Institutions:  School of Computer Science & Informatics, Cardiff University, Cardiff, United Kingdom. Department of Mathematics and Applied Mathematics, University of Cape Town, Cape Town, South Africa. SA MRC Unit on Risk & Resilience in Mental Disorders, Department of Psychiatry and Neuroscience Institute, University of Cape Town, Cape Town, South Africa. Department of Psychiatry, Faculty of Medicine and Health Sciences, Stellenbosch University, Cape Town, South Africa. Department of Clinical Neurosciences, University of Cambridge, Cambridge, United Kingdom.

Abstract summary 

In neuroimaging, the difference between chronological age and predicted brain age, also known as , has been proposed as a pathology marker linked to a range of phenotypes. Brain age delta is estimated using regression, which involves a frequently observed bias due to a negative correlation between chronological age and brain age delta. In brain age prediction models, this correlation can manifest as an overprediction of the age of young brains and an underprediction for elderly ones. We show that this bias can be controlled for by adding correlation constraints to the model training procedure. We develop an analytical solution to this constrained optimization problem for Linear, Ridge, and Kernel Ridge regression. The solution is optimal in the least-squares sense i.e., there is no other model that satisfies the correlation constraints and has a better fit. Analyses on the PAC2019 competition data demonstrate that this approach produces optimal unbiased predictive models with a number of advantages over existing approaches. Finally, we introduce regression toolboxes for Python and MATLAB that implement our algorithm.

Authors & Co-authors:  Treder Matthias S MS Shock Jonathan P JP Stein Dan J DJ du Plessis Stéfan S Seedat Soraya S Tsvetanov Kamen A KA

Study Outcome 

Source Link: Visit source

Statistics
Citations :  Cole JH, Ritchie SJ, Bastin ME, Valdés Hernández MC, Muñoz Maniega S, Royle N, et al. . Brain age predicts mortality. Mol Psychiatry. (2018) 23:1385–92. 10.1038/mp.2017.62
Authors :  6
Identifiers
Doi : 615754
SSN : 1664-0640
Study Population
Male,Female
Mesh Terms
Other Terms
age;brain;correlation;optimization;prediction;regression
Study Design
Cross Sectional Study
Study Approach
Country of Study
Publication Country
Switzerland