CK4Gen: A Knowledge Distillation Framework for Generating High-Utility Synthetic Survival Datasets in Healthcare
On the map: Knowledge distillation → Distillation for medical imaging
Abstract (arXiv)
Access to real clinical data is heavily restricted by privacy regulations, hindering both healthcare research and education. These constraints slow progress in developing new treatments and data-driven healthcare solutions, while also limiting students' access to real-world datasets, leaving them without essential practical skills. High-utility synthetic datasets are therefore critical for advancing research and providing meaningful training material. However, current generative models -- such as Variational Autoencoders (VAEs) and Generative Adversarial Networks (GANs) -- produce surface-level realism at the expense of healthcare utility, blending distinct patient profiles and producing synthetic data of limited practical relevance. To overcome these limitations, we introduce CK4Gen (Cox Knowledge for Generation), a novel framework that leverages knowledge distillation from Cox Proportional Hazards (CoxPH) models to create synthetic survival datasets that preserve key clinical characteristics, including hazard ratios and survival curves. CK4Gen avoids the interpolation issues seen in VAEs and GANs by maintaining distinct patient risk profiles, ensuring realistic and reliable outputs for research and educational use. Validated across four benchmark datasets -- GBSG2, ACTG320, WHAS500, and FLChain -- CK4Gen outperforms competing techniques by better aligning real and synthetic data, enhancing survival model performance in both discrimination and calibration via data augmentation. As CK4Gen is scalable across clinical conditions, and with code to be made publicly available, future researchers can apply it to their own datasets to generate synthetic versions suitable for open sharing.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Distillation for medical imaging, one of 12 directions in Knowledge distillation. First posted on arXiv 22 Oct 2024.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2025: 1; 2026: 2.
3 papers.
- 2025Attention-Based Synthetic Data Generation for Calibration-Enhanced Survival Analysis: A Case Study for Chronic Kidney Disease Using Electronic Health Recordscited by 1
- 2026Synthetic but Not Realistic: The Evaluation Challenge in Generative Modelling for Structured Electronic Medical Recordscited by 0
- 2026From Hazard Functions to Language Space: Cox-Supervised Distillation of Survival Risk into a Large Language Modelcited by 0
- 2026From Hazard Functions to Language Space: Cox-Supervised Distillation of Survival Risk into a Large Language Model8 Jun 2026
- 2026Synthetic but Not Realistic: The Evaluation Challenge in Generative Modelling for Structured Electronic Medical Records8 Jun 2026
- 2025Attention-Based Synthetic Data Generation for Calibration-Enhanced Survival Analysis: A Case Study for Chronic Kidney Disease Using Electronic Health Records8 Mar 2025
Similar, not cited
Close in wording, no citation either way
- 2025Synthetic Survival Data Generation for Heart Failure Prognosis Using Deep Generative Models
- 2025Generating Reliable Synthetic Clinical Trial Data: The Role of Hyperparameter Optimization and Domain Constraints
- 2023Synthetic Health-related Longitudinal Data with Mixed-type Variables Generated using Diffusion Models
- 2024Generative AI for Synthetic Data Across Multiple Medical Modalities: A Systematic Review of Recent Developments and Challenges
- 2025SurvDiff: A Diffusion Model for Generating Synthetic Data in Survival Analysis
- 2024Guided Discrete Diffusion for Electronic Health Record Generation
- 2025An Integrated Approach to AI-Generated Content in e-health
- 2025A Reinforcement Learning Approach to Synthetic Data Generation