Realistic Synthetic Electronic Health Record Data Generation for Cardiovascular Risk Prediction
TL;DR
A Large Language Model-guided retrieval-aware framework for generating realistic synthetic EHR data on a large scale to train ML models to predict cardiovascular risks accurately and achieves high performance in predicting CVD.
Abstract
Background: Increasing access to Electronic Health Records (EHRs) has enabled the development of Machine Learning (ML) models to predict early cardiovascular disease (CVD) risk. Nevertheless, real EHR data is usually limited in availability and distribution because of confidentiality issues, legal limitations, and insufficient access. Synthetic data generation has become a promising approach to overcome such difficulties. Objectives: This study proposes a Large Language Model (LLM)-guided retrieval-aware framework for generating realistic synthetic EHR data on a large scale to train ML models to predict cardiovascular risks accurately. Methods: The framework uses a Tabular Denoising Diffusion Probabilistic Model to learn the underlying distribution of the original dataset and generate an initial synthetic dataset. To improve the clinical plausibility of the generated data, a knowledge-guided refinement module with LLaMA 2 13B combined with Retrieval-Augmented Generation (RAG) is introduced. The LLM analyzes statistical trends in real and synthetic data while retrieving relevant medical information to detect and correct clinically implausible correlations. Results: Experimental findings show that the optimized synthetic dataset preserves important statistical features of the original data and achieves high performance in predicting CVD. Conclusions: Thus, the framework offers a scalable, privacy-conservative method for generating realistic synthetic healthcare data suitable for medical research.