Skip to main content

ORIGINAL RESEARCH article

Front. Aging Neurosci.
Sec. Parkinson’s Disease and Aging-related Movement Disorders
Volume 16 - 2024 | doi: 10.3389/fnagi.2024.1396737

Prediction of Mild Cognitive Impairment in Parkinson's Disease Using an Optimized Stacking Ensemble Model

Provisionally accepted
  • 1 Qingdao University of Science and Technology, Qingdao, China
  • 2 Qilu Hospital of Shandong University (Qingdao), Qingdao, Shandong, China

The final, formatted version of the article will be published soon.

    Parkinson's disease (PD) is a neurodegenerative disorder prevalent among the elderly, often leading to cognitive impairment in severe cases. However, its early stages may lack significant non-motor symptoms, resulting in a high misdiagnosis rate. To promote early treatment and targeted intervention for mild cognitive impairment (MCI) in PD patients, this study proposes a Stacking ensemble model with Optuna hyperparameter optimization. The model features a two-layer learning framework, comprising a basic classifier learning layer and a metamodel learning layer, aimed at improving the accuracy of predicting MCI in PD patients. Firstly, we designed a feature selection method based on multimodal data, identifying seven risk factors closely associated with cognitive impairment. Secondly, we constructed the Stacking ensemble model using five different learning methods to build the base classifier learning layer, with Extreme Gradient Boosting (XGBoost) serving as the metamodel. Additionally, the key parameters of XGBoost were autonomously optimized using the Optuna hyperparameter optimization method to enhance the predictive accuracy of the proposed Stacking ensemble model. Experimental results demonstrate that the proposed Optuna hyperparameter-optimized Stacking ensemble method exhibits high accuracy and effectiveness, with an accuracy of 0.9200, precision of 0.8768, recall of 0.9495, and an F1 score of 0.91315.

    Keywords: Stacking ensemble, machine learning, Parkinson's disease, Mild Cognitive Impairment, Optuna hyperparameter optimization

    Received: 07 Mar 2024; Accepted: 16 Oct 2024.

    Copyright: © 2024 Yin, Ba, Ren and LI. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Shaoyuan LI, Qingdao University of Science and Technology, Qingdao, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.