IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i9p1303-d1382754.html
   My bibliography  Save this article

TPTM-HANN-GA: A Novel Hyperparameter Optimization Framework Integrating the Taguchi Method, an Artificial Neural Network, and a Genetic Algorithm for the Precise Prediction of Cardiovascular Disease Risk

Author

Listed:
  • Chia-Ming Lin

    (Professional Master Program in Artificial Intelligence in Medicine, College of Medicine, Taipei Medical University, Taipei 11031, Taiwan)

  • Yu-Shiang Lin

    (Professional Master Program in Artificial Intelligence in Medicine, College of Medicine, Taipei Medical University, Taipei 11031, Taiwan)

Abstract

The timely and precise prediction of cardiovascular disease (CVD) risk is essential for effective prevention and intervention. This study proposes a novel framework that integrates the two-phase Taguchi method (TPTM), the hyperparameter artificial neural network (HANN), and a genetic algorithm (GA) called TPTM-HANN-GA. This framework efficiently optimizes hyperparameters for an artificial neural network (ANN) model during the training stage, significantly enhancing prediction accuracy for cardiovascular disease (CVD) risk. The proposed TPTM-HANN-GA framework requires far fewer experiments than a traditional grid search, making it highly suitable for application in resource-constrained, low-power computers, and edge artificial intelligence (edge AI) devices. Furthermore, the proposed TPTM-HANN-GA framework successfully identified the optimal configurations for the ANN model’s hyperparameters, resulting in a hidden layer of 4 nodes, a tanh activation function, an SGD optimizer, a learning rate of 0.23425849, a momentum rate of 0.75462782, and seven hidden nodes. This optimized ANN model achieves 74.25% accuracy in predicting the risk of cardiovascular disease, which exceeds the existing state-of-the-art GA-ANN and TSTO-ANN models. The proposed TPTM-HANN-GA framework enables personalized CVD prediction to be efficiently conducted on low-power computers and edge-AI devices, achieving the goal of point-of-care testing (POCT) and empowering individuals to manage their heart health effectively.

Suggested Citation

  • Chia-Ming Lin & Yu-Shiang Lin, 2024. "TPTM-HANN-GA: A Novel Hyperparameter Optimization Framework Integrating the Taguchi Method, an Artificial Neural Network, and a Genetic Algorithm for the Precise Prediction of Cardiovascular Disease R," Mathematics, MDPI, vol. 12(9), pages 1-22, April.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:9:p:1303-:d:1382754
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/9/1303/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/9/1303/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:9:p:1303-:d:1382754. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.