Machine learning based hierarchical classification of frontotemporal dementia and Alzheimer's disease

Abstract

Background: In a clinical setting, an individual subject classification model rather than a group analysis would be more informative. Specifically, the subtlety of cortical atrophy in some frontotemporal dementia (FTD) patients and overlapping patterns of atrophy among three FTD clinical syndromes including behavioral variant FTD (bvFTD), non-fluent/agrammatic variant primary progressive aphasia (nfvPPA), and semantic variant PPA (svPPA) give rise to the need for classification models at the individual level. In this study, we aimed to classify each individual subject into one of the diagnostic categories in a hierarchical manner by employing a machine learning-based classification method. Methods: We recruited 143 patients with FTD, 50 patients with Alzheimer's disease (AD) dementia, and 146 cognitively normal subjects. All subjects underwent a three-dimensional volumetric brain magnetic resonance imaging (MRI) scan, and cortical thickness was measured using FreeSurfer. We applied the Laplace Beltrami operator to reduce noise in the cortical thickness data and to reduce the dimension of the feature vector. Classifiers were constructed by applying both principal component analysis and linear discriminant analysis to the cortical thickness data. For the hierarchical classification, we trained four classifiers using different pairs of groups: Step 1 - CN vs. FTD + AD, Step 2 - FTD vs. AD, Step 3 - bvFTD vs. PPA, Step 4 - svPPA vs. nfvPPA. To evaluate the classification performance for each step, we used al 0-fold cross-validation approach, performed 1000 times for reliability. Results: The classification accuracy of the entire hierarchical classification tree was 75.8%, which was higher than that of the non-hierarchical classifier (73.0%). The classification accuracies of steps 1-4 were 86.1%, 90.8%, 86.9%, and 92.1%, respectively. Changes in the right frontotemporal area were critical for discriminating behavioral variant FTD from PPA. The left frontal lobe discriminated nfvPPA from svPPA, while the bilateral anterior temporal regions were critical for identifying svPPA. Conclusions: In the present study, our automated classifier successfully classified FTD clinical subtypes with good to excellent accuracy. Our classifier may help clinicians diagnose FTD subtypes with subtle cortical atrophy and facilitate appropriate specific interventions.