Optimization of diabetes prediction methods based on combinatorial balancing algorithm

Background Diabetes, as a significant disease affecting public health, requires early detection for effective management and intervention. However, imbalanced datasets pose a challenge to accurate diabetes prediction. This imbalance often results in models performing poorly in predicting minority cl...

Full description

Saved in:
Bibliographic Details
Published in:Nutrition & diabetes Vol. 14; no. 1; pp. 63 - 13
Main Authors: Shao, HuiZhi, Liu, Xiang, Zong, DaShuai, Song, QingJun
Format: Journal Article
Language:English
Published: London Nature Publishing Group UK 14.08.2024
Nature Publishing Group
Subjects:
692/699
692/700
Algorithms
Clinical Nutrition
Datasets
Diabetes
Diabetes Mellitus
Epidemiology
Humans
Internal Medicine
Machine Learning
Medicine
Medicine & Public Health
Metabolic Diseases
Optimization
Review
Review Article
ISSN:2044-4052, 2044-4052
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Background Diabetes, as a significant disease affecting public health, requires early detection for effective management and intervention. However, imbalanced datasets pose a challenge to accurate diabetes prediction. This imbalance often results in models performing poorly in predicting minority classes, affecting overall diagnostic performance. Objectives To address this issue, this study employs a combination of Synthetic Minority Over-sampling Technique (SMOTE) and Random Under-Sampling (RUS) for data balancing and uses Optuna for hyperparameter optimization of machine learning models. This approach aims to fill the gap in current research concerning data balancing and model optimization, thereby improving prediction accuracy and computational efficiency. Methods First, the study uses SMOTE and RUS methods to process the imbalanced diabetes dataset, balancing the data distribution. Then, Optuna is utilized to optimize the hyperparameters of the LightGBM model to enhance its performance. During the experiment, the effectiveness of the proposed methods is evaluated by comparing the training results of the dataset before and after balancing. Results The experimental results show that the enhanced LightGBM-Optuna model improves the accuracy from 97.07% to 97.11%, and the precision from 97.17% to 98.99%. The time required for a single search is only 2.5 seconds. These results demonstrate the superiority of the proposed method in handling imbalanced datasets and optimizing model performance. Conclusions The study indicates that combining SMOTE and RUS data balancing algorithms with Optuna for hyperparameter optimization can effectively enhance machine learning models, especially in dealing with imbalanced datasets for diabetes prediction.
Bibliography:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
ObjectType-Review-3
content type line 23
ISSN:2044-4052
2044-4052
DOI:10.1038/s41387-024-00324-z