A subscription to JoVE is required to view this content. Sign in or start your free trial.

Research Article

Automated Machine Learning Model for Non-Alcoholic Fatty Liver Disease Prediction and External Cohort Validation

175 views

⸱

DOI:

10.3791/70033

⸱

July 3rd, 2026

 ,  ,  ,  ,  ,  ,  , 

* These authors contributed equally

In This Article

Summary

A step-by-step AutoML protocol is presented to screen for nonalcoholic fatty liver disease using NHANES data and an external cohort. The method automates feature prioritization and model selection (GBM), and includes SHAP-based interpretation and external validation for reproducible clinical deployment.

Abstract

Non-alcoholic fatty liver disease (NAFLD) is a common liver disorder associated with obesity, insulin resistance, and metabolic syndrome, often going undiagnosed until advanced stages. Traditional diagnostic methods, including imaging and liver biopsy, have limitations in early detection. There is a need for an efficient, non-invasive tool for early NAFLD screening. Using automated machine learning (AutoML) technology, a diagnostic model for NAFLD was developed leveraging a large dataset from NHANES (n = 2677). Additionally, an independent external validation cohort (n = 200) was employed to assess the external validity and performance of the model. For selection of promising clinical features, A two-stage feature selection method was applied, combining LASSO regression with ChatGPT-4-based intelligent analysis. Subsequently, the AutoML process, which integrated multiple machine learning algorithms, was performed for model training and validation. The model's performance was evaluated using ROC curves, F1 scores, and SHapley additive explanation (SHAP) analysis. The GBM model achieved an AUC of 0.843 in the training set, 0.851 in the testing set, and 0.945 in the external validation set, demonstrating high diagnostic accuracy across different datasets. Key predictors, including BMI, triglycerides, and GGT, were identified as significant contributors to the model’s predictions. SHAP analysis further confirmed the importance of these variables in predicting NAFLD. The AutoML-driven diagnostic model for NAFLD demonstrated significantly improved early-detection performance, offering a reliable, non-invasive, and efficient alternative to conventional diagnostic methods. This method holds great potential for broader clinical application in NAFLD, diminishing dependence on expert knowledge while improving diagnostic precision.

Introduction

Non-alcoholic fatty liver disease (NAFLD) is one of the most prevalent liver diseases globally, currently affecting approximately 35% of the adult population worldwide, and is not attributable to excessive alcohol consumption1. It is characterized by the accumulation of fat in more than 5% of hepatocytes, which can potentially progress to more severe conditions, including inflammation, liver fibrosis, cirrhosis, and ultimately liver failure2,3. Currently, the traditional methods for diagnosing NAFLD primarily rely on abnormal liver biochemical markers and ultrasound ima....

Access restricted. Please log in or start a trial to view this content.

Protocol

Patients and study design

The first dataset of this study was obtained from the National Health and Nutrition Examination Survey (NHANES) database for the 2017–2018 period. This specific timeframe was selected because the survey data included liver ultrasound transient elastography measurements using FibroScan® technology. NHANES employs a stratified, multi-stage probability sampling design and serves as a nationwide, population-based survey conducted biennially. It systematically gathers nationally representative health-related data on the non-institutionalized U.S. population to evaluate the nutritional and he....

Access restricted. Please log in or start a trial to view this content.

Results

Clinical features of NAFLD

In this study, the clinical characteristics of individuals with and without NAFLD were systematically analyzed. A range of methods was employed to clearly illustrate the distribution and differences of these features within the sample population (Table 1). A total of 2,677 individuals were included in the analysis, of whom 718 had NAFLD. The analysis covered a range of clinical indicators, including BMI, TG, GGT, UA, and others (as s.......

Access restricted. Please log in or start a trial to view this content.

Discussion

NAFLD is the commonest cause of chronic liver disease worldwide, and its prevalence rises by about 1% each year26. Roughly 30% of the global population is affected, making NAFLD not only the leading chronic liver disorder but also the fastest‑growing indication for liver transplantation3. Automated machine learning (AutoML) has emerged as a valuable tool in medicine, with one of its main goals being to build predictive models by identifying correlatio.......

Access restricted. Please log in or start a trial to view this content.

Disclosures

The authors declare that they have no conflict of interest. In this study, ChatGPT-4 (OpenAI) was used as an intelligent analytical tool during the feature selection phase to assess the clinical relevance of variables identified by LASSO regression. All AI-generated scores were carefully reviewed by the authors and empirically validated through the subsequent AutoML process. ChatGPT-4 was not used to alter raw data or perform mathematical calculations. The authors take full responsibility for the integrity and accuracy of the final results.

Acknowledgements

This work was supported by the Startup Fund for Scientific Research of Fujian Medical University [2021QH1176]; the Key Laboratory of Clinical Laboratory Diagnosis and Translational Research of Zhejiang Province [2022E10022]; the Zhejiang Provincial Medical and Health Science and Technology Plan Project [2024KY1265]; and the Wenzhou Municipal Basic Public Welfare Research Project [Y2023096].

....

Access restricted. Please log in or start a trial to view this content.

Materials

List of materials used in this article
NameCompanyCatalog NumberComments
Datasets for NAFLD (2017–2018)NHANES database [https://wwwn.cdc.gov/nchs/nhanes/continuousnhanes/default.aspx?BeginYear=2017]N/Aas the training set for model construction
Datasets for NAFLD (2018–2020)The First Affiliated Hospital of Wenzhou Medical University (2018–2020), China. [https://doi.org/10.6084/m9.figshare.32105248.]N/Aas a test set for the external validation of the model
R (version 4.2.3)R Foundation for Statistical ComputingN/Aused in conjunction with other tools to conduct data analysis and generate visual and graphical presentations.
Adobe Illustrator 2025 (version 29.2)Adobe Systems IncorporatedN/AFor image layout and editing
ChatGPT-4OpenAI Inc.N/AFor assigning an importance score to each selected variable
H2O AutoML (3.44.0.3)H2O.aiN/AFor integrating a comprehensive set of machine learning algorithms

Reprints and Permissions

Explore More Articles

NAFLD PredictionFeature SelectionLASSO RegressionSHAP AnalysisDiagnostic ModelEarly NAFLD ScreeningMachine Learning Algorithms