arXiv:2511.03806cs.LG2025-11

用大模型估算敏感特征,提升隐私保护下的模型效果

FusionDP: Foundation Model-Assisted Differentially Private Learning for Partially Sensitive Features

  • 用大模型根据非敏感特征推测敏感特征值,避免直接使用真实敏感数据
  • 在生理数据和病历文本任务中,性能优于传统差分隐私方法
  • 适合需部分隐私保护的医疗等敏感数据场景

保护训练数据隐私是隐私保护机器学习的关键。但在实际中,往往只需对部分特征进行隐私保护。例如,在ICU数据中,年龄、性别等人口统计学特征因易被重新识别而风险更高,而原始检验结果则相对不敏感。传统DP-SGD对单个样本的所有特征统一施加隐私保护,导致噪声过大、模型性能显著下降。我们提出FusionDP,一种两阶段框架,以提升特征级差分隐私下的模型效用。首先,利用大基础模型根据非敏感特征推断敏感特征值,将其作为外部先验,在训练时不访问真实敏感值即可获得高质量估计;其次,引入改进的DP-SGD算法,同时在原始特征与推断特征上训练模型,严格保障原始敏感特征的隐私。我们在两个模态上评估:PhysioNet上的脓毒症预测任务和MIMIC-III上的临床笔记分类任务。对比多个隐私保护基线,结果表明FusionDP显著提升模型性能,同时维持严格的特征级隐私,证明了大模型驱动的插补在多种模态下能有效改善隐私-效用权衡。

原文摘要 · Abstract (English)

Ensuring the privacy of sensitive training data is crucial in privacy-preserving machine learning. However, in practical scenarios, privacy protection may be required for only a subset of features. For instance, in ICU data, demographic attributes like age and gender pose higher privacy risks due to their re-identification potential, whereas raw lab results are generally less sensitive. Traditional DP-SGD enforces privacy protection on all features in one sample, leading to excessive noise injection and significant utility degradation. We propose FusionDP, a two-step framework that enhances model utility under feature-level differential privacy. First, FusionDP leverages large foundation models to impute sensitive features given non-sensitive features, treating them as external priors that provide high-quality estimates of sensitive attributes without accessing the true values during model training. Second, we introduce a modified DP-SGD algorithm that trains models on both original and imputed features while formally preserving the privacy of the original sensitive features. We evaluate FusionDP on two modalities: a sepsis prediction task on tabular data from PhysioNet and a clinical note classification task from MIMIC-III. By comparing against privacy-preserving baselines, our results show that FusionDP significantly improves model performance while maintaining rigorous feature-level privacy, demonstrating the potential of foundation model-driven imputation to enhance the privacy-utility trade-off for various modalities.

差分隐私医疗AI大模型特征级隐私

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