Improving Stroke Prediction on Imbalanced Clinical Data Using Ctgan and Tvae: a Synthetic Data Approach
Year: 2025
Type: conference-paper
Abstract: Synthetic data (SD) have been evaluated and adopted in different domains and areas, especially in health. To conduct this study, we chose tabular data on stroke prediction, available in [3]. The dataset contains 11 clinical features, including the last column, positive = 1 and negative = 0 for stroke. We chose this dataset because of its imbalance, which will be a perfect fit for implementing the generating techniques to know how well the real data resemble these. For generating SD, we use two techniques known as conditional tabular GAN (CTGAN) and tabular variational autoencoder (TVAE), which have different numbers of epochs and batch sizes. We further evaluated the results with three Machine Learning (ML) models as a benchmark with real data. The results highlight data generated with CTGAN (epochs=1500, batch size=500) perform better with an accuracy score of 0.995 on random forest (RF) and Support Vector Machine (SVM).
Authors: Liaquat, Majid; id_orcid 0000-0002-3772-2299, Nugent, CD, Cleland, I, Khan, Naveed; id_orcid 0000-0002-9301-5855
Institution: University of Ulster
Language: English
Subfield: Artificial Intelligence
Field: Computer Science
Domain: Physical Sciences
Open Access status: green
OpenAlex ID: W7038453754
FWCI:
Cites:
Cited by:
Related to: