A Novel Approach to Dementia Prediction Leveraging Recursive Feature Elimination and Decision Tree

Abstract

Early prediction of dementia and disease progression remains challenging. This study presents a novel machine learning framework for dementia diagnosis by integrating multimodal neuroimaging biomarkers and inexpensive, readily available clinical factors. Fractional anisotropy (FA) measurements in diffusion tensor imaging (DTI) provide microstructural insights into white matter integrity disturbances in dementia. However, acquiring DTI is costly and time-consuming. We applied Recursive Feature Elimination (RFE) to identify predictors from structural measures of the 9 Brain Atrophy and Lesion Index (BALI) factors and 42 Clinical Lifestyle for Brain Health (LIBRA) factors to estimate fractional anisotropy (FA) in DTI. The 10 most effective BALI/LIBRA features selected by RFE were used to train an interpretable decision tree model to predict dementia severity from DTI. A decision tree model based on biomarkers selected by Recursive Feature Elimination (RFE) achieved an accuracy of 96.25% in predicting dementia in an independent test set. This integrated framework pioneers the prediction of white matter microstructural changes from available structural/clinical factors using machine learning. By avoiding DTI acquisition, our approach provides a practical and objective tool to enhance dementia screening and progress monitoring. Identification of key predictive markers of BALI/LIBRA will also provide insights into lifestyle-related disease mechanisms, neurodegeneration, and white matter dysfunction.

Multimodal dementia identification using lifestyle and brain lesions, a machine learning approach

“…60 Axial diffusion tensor imaging (DTI) metrics, such as fractional anisotropy (FA), are derived from the eigenvalues and eigenvectors of the diffusion tensor, D. These scalar and vector values provide valuable insight into the underlying tissue microstructure. 61 Fractional anisotropy (FA) is obtained from DTI data, which characterizes the degree of anisotropic diffusion of water in biological tissues. In the context of brain imaging, FA provides valuable insights into the orientation and coherence of white matter tracts, reflecting the extent to which water molecule diffusion is restricted along the orientation of fiber bundles.…”