What are you looking for today?

Danso, Samuel O, Zeng, Zhanhang, Muniz-Terrera, Graciela and Ritchie, Craig W (2021) Developing an explainable machine learning-based personalised dementia risk prediction model: A transfer learning approach with ensemble learning algorithms. Frontiers in big Data, 4. p. 613047. ISSN 2624-909X

Document

Abstract

Abstract

Alzheimer's disease (AD) has its onset many decades before dementia develops, and work is ongoing to characterise individuals at risk of decline on the basis of early detection through biomarker and cognitive testing as well as the presence/absence of identified risk factors. Risk prediction models for AD based on various computational approaches, including machine learning, are being developed with promising results. However, these approaches have been criticised as they are unable to generalise due to over-reliance on one data source, poor internal and external validations, and lack of understanding of prediction models, thereby limiting the clinical utility of these prediction models. We propose a framework that employs a transfer-learning paradigm with ensemble learning algorithms to develop explainable personalised risk prediction models for dementia. Our prediction models, known as source models, are initially trained and tested using a publicly available dataset (n = 84,856, mean age = 69 years) with 14 years of follow-up samples to predict the individual risk of developing dementia. The decision boundaries of the best source model are further updated by using an alternative dataset from a different and much younger population (n = 473, mean age = 52 years) to obtain an additional prediction model known as the target model. We further apply the SHapely Additive exPlanation (SHAP) algorithm to visualise the risk factors responsible for the prediction at both population and individual levels. The best source model achieves a geometric accuracy of 87%, specificity of 99%, and sensitivity of 76%. In comparison to a baseline model, our target model achieves better performance across several performance metrics, within an increase in geometric accuracy of 16.9%, specificity of 2.7%, and sensitivity of 19.1%, an area under the receiver operating curve (AUROC) of 11% and a transfer learning efficacy rate of 20.6%. The strength of our approach is the large sample size used in training the source model, transferring and applying the “knowledge” to another dataset from a different and undiagnosed population for the early detection and prediction of dementia risk, and the ability to visualise the interaction of the risk factors that drive the prediction. This approach has direct clinical utility.

Item Type: Article
Related URLs:
Depositing User: Sam Danso
URI: https://sure.sunderland.ac.uk/id/eprint/17001

Further Details

More Information

Related URLs
Depositing User

Identifiers

Item ID
17001
Identification Number
ISSN
2624-909X

Users with ORCIDS

Catalogue record

Date Deposited
23 Sep 2024 13:57
Last Modified
26 Nov 2025 09:09

Contributors

Author
Samuel O Danso
Author
Zhanhang Zeng
Author
Graciela Muniz-Terrera
Author
Craig W Ritchie

University Divisions

Subjects

Actions (login required)

View Item (Repository Staff Only)

Downloads per month over past year