S2842 A Multi-Stage Machine-Learning Pipeline for Early Detection, Severity Stratification, and Prognostic Forecasting in Liver Disease
Introduction: Liver disease remains a major global health burden, often progressing undetected until advanced stages. We applied supervised and unsupervised machine-learning techniques—including random forest classification, k-means clustering, and long short-term memory networks. Methods: We analyzed the Indian Liver Patient Dataset (583 records) containing age, gender, total and direct bilirubin, alkaline phosphatase, alanine and aspartate aminotransferases, total proteins, albumin, and albumin–globulin ratio. We imputed missing albumin–globulin values with the mean, encoded gender numerically, and created interaction features (age × bilirubin, gender × alkaline phosphatase). We normalized all continuous variables and split the data into 80 percent training and 20 percent testing sets. A random forest classifier distinguished disease-positive from disease-negative cases. To infer severity without predefined labels, we applied k-means clustering (k = 3) on the normalized feature set and mapped each cluster to mild, moderate, or severe stages. We trained a second random forest on these inferred labels to predict severity for new patients. Finally, we generated synthetic 5-visit time-series data—spacing visits 30 days apart and adding Gaussian noise—and trained long short-term memory networks to forecast next-visit bilirubin values and severity stages based on the preceding 3 visits. Results: The disease classifier reached 81.2% accuracy on the held-out test set, with precision and recall both above 0.80 for the liver-disease class. Clustering yielded 3 subgroups of comparable size, each exhibiting distinct biochemical signatures. The severity classifier achieved 99.1% accuracy in assigning mild, moderate, or severe labels. The bilirubin-forecasting LSTM produced a mean absolute error of 1.07 mg/dL on the test set. The severity-forecasting LSTM achieved 96% overall accuracy and an F1-score of 0.92 for the severe category. These results demonstrate that combining machine-learning classification, unsupervised clustering, and sequence modeling can accurately detect liver disease, stratify its severity, and anticipate biomarker and stage progression. Conclusion: The results demonstrate the potential of combining classification, clustering, and sequence modeling to support timely diagnosis, personalized risk stratification, and proactive management in liver disease care.
Authors
- Gautam Maddineni (ORCID: https://orcid.org/0009-0007-9492-2733)
- Sri Harsha Boppana
- Manaswitha Thota (ORCID: https://orcid.org/0009-0002-2680-8772)
- C. David Mintz (ORCID: https://orcid.org/0000-0001-9278-4380)
- Sachin Sravan Kumar Komati
- Sarath Chandra Ponnada
- Sai Lakshmi Prasanna Komati
Publication Details
- Journal
- The American Journal of Gastroenterology
- Published
- 2025-10-01
- DOI
- https://doi.org/10.14309/01.ajg.0001138828.50148.e7
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00