arXiv:2410.16872cs.LG2024-10被引 3

用生存分析知识蒸馏生成真实可用的医疗合成数据

CK4Gen: A Knowledge Distillation Framework for Generating High-Utility Synthetic Survival Datasets in Healthcare

  • 从Cox模型提取生存规律,指导合成数据生成
  • 在4个临床数据集上提升生存预测性能,优于现有方法
  • 适合需要隐私保护数据的研究者与医学生使用

真实临床数据因隐私法规难以获取,制约了医疗研究与教学。现有生成模型如VAE、GAN虽能生成表面逼真的数据,但混淆患者风险分层,实用性不足。本文提出CK4Gen框架,通过知识蒸馏自Cox比例风险模型,生成保留关键临床特征(如风险比、生存曲线)的合成生存数据。该方法避免了VAE和GAN中患者群体混杂的问题,确保风险分层清晰。在GBSG2、ACTG320、WHAS500和FLChain四个基准数据集上验证,合成数据在判别力和校准度上均优于对比方法,且可扩展至多种临床场景。代码将公开,支持研究人员生成可共享的合成数据。

原文摘要 · Abstract (English)

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.

医疗合成数据知识蒸馏生存分析隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。