用强化学习让大模型生成更真实、更安全的医疗数据。
DISCO-TAB: A Hierarchical Reinforcement Learning Framework for Privacy-Preserving Synthesis of Complex Clinical Data
- 分层反馈机制+自动约束发现,精准控制生成逻辑
- 合成数据临床可用性提升38.2%,隐私保护强于主流方法
- 适合需要高保真医疗数据的科研与临床系统开发
构建稳健的临床决策支持系统常受限于高质量、隐私保护型生物医学数据的稀缺。尽管生成式大语言模型(LLMs)在合成数据方面前景广阔,但往往难以捕捉电子健康记录(EHR)中复杂的非线性依赖关系和严重的类别不平衡,导致生成的数据虽统计上合理却临床无效。为此,我们提出DISCO-TAB(DIScriminator-guided COntrol for TABular synthesis),一种将微调后的LLM与多目标判别器系统结合的框架,通过强化学习优化。不同于以往依赖标量反馈的方法,DISCO-TAB在词、句子、特征和行四个粒度上评估生成结果,并融合自动约束发现与逆频率奖励设计,自主保持潜在医学逻辑并缓解少数类崩溃问题。我们在多种基准上验证该框架,涵盖高维、小样本医疗数据集(如心衰、帕金森病)。结果表明,分层反馈带来顶尖性能,下游临床分类器效用相较GAN与扩散模型基线最高提升38.2%,同时保证极高的统计保真度(JSD < 0.01)和对成员推断攻击的强抵抗力。本工作为敏感医疗应用建立了可信赖、保用性的合成表格数据新标准。
原文摘要 · Abstract (English)
The development of robust clinical decision support systems is frequently impeded by the scarcity of high-fidelity, privacy-preserving biomedical data. While Generative Large Language Models (LLMs) offer a promising avenue for synthetic data generation, they often struggle to capture the complex, non-linear dependencies and severe class imbalances inherent in Electronic Health Records (EHR), leading to statistically plausible but clinically invalid records. To bridge this gap, we introduce DISCO-TAB (DIScriminator-guided COntrol for TABular synthesis), a novel framework that orchestrates a fine-tuned LLM with a multi-objective discriminator system optimized via Reinforcement Learning. Unlike prior methods relying on scalar feedback, DISCO-TAB evaluates synthesis at four granularities, token, sentence, feature, and row, while integrating Automated Constraint Discovery and Inverse-Frequency Reward Shaping to autonomously preserve latent medical logic and resolve minority-class collapse. We rigorously validate our framework across diverse benchmarks, including high-dimensional, small-sample medical datasets (e.g., Heart Failure, Parkinson's). Our results demonstrate that hierarchical feedback yields state-of-the-art performance, achieving up to 38.2% improvement in downstream clinical classifier utility compared to GAN and Diffusion baselines, while ensuring exceptional statistical fidelity (JSD < 0.01) and robust resistance to membership inference attacks. This work establishes a new standard for generating trustworthy, utility-preserving synthetic tabular data for sensitive healthcare applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。