Joint Data Harmonization and Group Cardinality Constrained Classification

To boost the power of classifiers, studies often increase the size of existing samples through the addition of independently collected data sets. Doing so requires harmonizing the data for demographic and acquisition differences based on a control cohort before performing disease specific classification. The initial harmonization often mitigates group differences negatively impacting classification accuracy. To preserve cohort separation, we propose the first model unifying linear regression for data harmonization with a logistic regression for disease classification. Learning to harmonize data is now an adaptive process taking both disease and control data into account. Solutions within that model are confined by group cardinality to reduce the risk of overfitting (via sparsity), to explicitly account for the impact of disease on the inter-dependency of regions (by grouping them), and to identify disease specific patterns (by enforcing sparsity via the l0-'norm'). We test those solutions in distinguishing HIV-Associated Neurocognitive Disorder from Mild Cognitive Impairment of two independently collected, neuroimage data sets; each contains controls and samples from one disease. Our classifier is impartial to acquisition difference between the data sets while being more accurate in diseases seperation than sequential learning of harmonization and classification parameters, and non-sparsity based logistic regressors.

[1]  T. Ernst,et al.  Impact of apolipoprotein E ε4 and HIV on cognition and brain atrophy: Antagonistic pleiotropy and premature brain aging , 2011, NeuroImage.

[2]  Aurel Popa-Wagner,et al.  HIV-associated neurocognitive disorders , 2014, Journal of molecular psychiatry.

[3]  R. Fisher,et al.  The Logic of Inductive Inference , 1935 .

[4]  R. Bartha,et al.  Ventricular enlargement as a possible measure of Alzheimer's disease progression validated using the Alzheimer's disease neuroimaging initiative database. , 2008, Brain : a journal of neurology.

[5]  Torsten Rohlfing,et al.  Variation in longitudinal trajectories of regional brain volumes of healthy men and women (ages 10 to 85years) measured with atlas-based parcellation of MRI , 2013, NeuroImage.

[6]  Mert R. Sabuncu A Universal and Efficient Method to Compute Maps from Image-Based Prediction Models , 2014, MICCAI.

[7]  Paul M. Thompson,et al.  3D mapping of ventricular and corpus callosum abnormalities in HIV/AIDS , 2006, NeuroImage.

[8]  Magda Tsolaki,et al.  Multisite longitudinal reliability of tract-based spatial statistics in diffusion tensor imaging of healthy elderly subjects , 2014, NeuroImage.

[9]  Yong Zhang,et al.  Sparse Approximation via Penalty Decomposition Methods , 2012, SIAM J. Optim..

[10]  Kilian M. Pohl,et al.  Computing group cardinality constraint solutions for logistic regression problems , 2017, Medical Image Anal..

[11]  Budhachandra S. Khundrakpam,et al.  Predicting symptom severity in autism spectrum disorder based on cortical thickness measures in agglomerative data , 2016, bioRxiv.

[12]  Neda Jahanshad,et al.  Mapping white matter integrity in elderly people with HIV , 2014, Human brain mapping.