Machine learning on subcortical brain features: A study of sample size efficiency for neurodegenerative disease classification
Im, Y.; Kang, M. J. Y.; Gutman, B. A.; Thomopoulos, S. I.; Thompson, P. M.; Ching, C. R. K.; for the Alzheimers Disease Neuroimaging Initiative,
Show abstract
Subcortical brain alterations are a key feature of dementia disease progression. Machine learning (ML) has been applied widely to MRI-based brain features in dementia, where performance depends on the model choice, training data size, and input feature characteristics. Most studies compare ML models using a single training sample size. Here, we evaluate the sample-size efficiency of ML models based on subcortical gross volume and vertex-wise shape features for dementia stage classification using 2,511 samples in the Alzheimer's Disease Neuroimaging Initiative (ADNI). Learning curves were generated for dementia vs. cognitively normal controls (CN), dementia vs. mild cognitive impairment (MCI), and MCI vs. CN across increasing training sample sizes. Classification performance improved when increasing sample size for all models, with late- fusion models consistently achieving the highest performance, and Logit-TVL1 outperforming the other shape-based models. Learning curve analysis showed that classification performance was driven by the training sample size and the magnitude of anatomical group differences, and can be used to optimize future model selection tasks in dementia and other brain disorders.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- ReMiND: Recovery of Missing Neuroimaging using Diffusion Models with Application to Alzheimer’s Disease 95%
- Automated quality control of T1-weighted brain MRI scans for clinical research: methods comparison and design of a quality prediction classifier 94%
- NODDI-derived measures of microstructural integrity in medial temporal lobe white matter pathways are associated with Alzheimer's disease pathology and cognitive outcomes 93%
Similar papers in this journal
Similar papers in this journal
- Improved segmentation of the intracranial and ventricular volumes in populations with cerebrovascular lesions and atrophy using 3D CNNs 95%
- AnNoBrainer, an Automated Annotation of Mouse Brain Images using Deep Learning 93%
- Regularized Bagged Canonical Component Analysis for Multiclass Learning in Brain Imaging 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.