用双流蒸馏方法训练隐私保护的学生模型,避免泄露原始数据。
Learning Privacy-Preserving Student Networks via Discriminative-Generative Distillation
- 通过判别与生成双流蒸馏,从私有数据中提取知识并传递给学生网络。
- 仅少量私有数据用于标签查询,且通过差分隐私聚合降低泄露风险。
- 适用于需保护数据隐私的场景,如医疗、金融等敏感领域应用。
深度模型虽能从大量标注数据中学习丰富知识,但在实际部署中可能引发隐私泄露风险。本文提出一种判别-生成蒸馏方法,实现高效的隐私保护建模。核心思路是将模型作为桥梁,通过两阶段流程:首先在私有数据上训练基准分类器及多个互不重叠子集上的教师集成;其次固定分类器为判别器,以无数据方式训练生成器,生成大量合成数据,并用于训练变分自编码器(VAE)。其中部分合成数据经教师集成查询标签(采用差分隐私聚合),其余数据嵌入已训练的VAE进行重构。最后通过半监督学习同时完成两项任务:基于少量私有标签合成数据的知识蒸馏,以及基于大量重构数据对的切线-正交对抗正则化增强知识。该方法统一控制了对私有数据的查询成本,缓解了精度下降问题,最终获得隐私保护的学生模型。大量实验与分析验证了该方法的有效性。
原文摘要 · Abstract (English)
While deep models have proved successful in learning rich knowledge from massive well-annotated data, they may pose a privacy leakage risk in practical deployment. It is necessary to find an effective trade-off between high utility and strong privacy. In this work, we propose a discriminative-generative distillation approach to learn privacy-preserving deep models. Our key idea is taking models as bridge to distill knowledge from private data and then transfer it to learn a student network via two streams. First, discriminative stream trains a baseline classifier on private data and an ensemble of teachers on multiple disjoint private subsets, respectively. Then, generative stream takes the classifier as a fixed discriminator and trains a generator in a data-free manner. After that, the generator is used to generate massive synthetic data which are further applied to train a variational autoencoder (VAE). Among these synthetic data, a few of them are fed into the teacher ensemble to query labels via differentially private aggregation, while most of them are embedded to the trained VAE for reconstructing synthetic data. Finally, a semi-supervised student learning is performed to simultaneously handle two tasks: knowledge transfer from the teachers with distillation on few privately labeled synthetic data, and knowledge enhancement with tangent-normal adversarial regularization on many triples of reconstructed synthetic data. In this way, our approach can control query cost over private data and mitigate accuracy degradation in a unified manner, leading to a privacy-preserving student model. Extensive experiments and analysis clearly show the effectiveness of the proposed approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。