arXiv:2503.12822cs.LGstat.ML2025-03KDD被引 2

用私有梯度优化选择微调权重,提升隐私保护下的模型性能。

An Optimization Framework for Differentially Private Sparse Fine-Tuning

  • 基于私有梯度信息动态选择可训练权重,非固定或依赖公开权重。
  • 在多个视觉模型和数据集上,准确率优于全模型私有微调及现有稀疏方法。
  • 适合需要高隐私保障且追求高性能的迁移学习场景。

差分隐私随机梯度下降(DP-SGD)被视为在差分隐私约束下训练和微调神经网络的金标准。随着高质量预训练模型检查点(如视觉和语言模型)的普及,微调已成为主流策略。然而,尽管在私有迁移学习中对DP-SGD的理解与应用取得进展,仍存在显著挑战——尤其是使用DP-SGD微调的模型与非私有模型之间的性能差距。稀疏微调(仅更新部分权重)成为全模型微调的替代方案;近期研究表明,在私有数据上仅微调少量模型参数并固定其余参数,可获得更好性能。本文提出一种新的差分隐私下神经网络稀疏微调优化框架。现有工作常采用固定可训练权重(如仅更新最后一层),或依赖公开模型权重来选择待修改参数,此类选择方式仍非最优。相比之下,本文方法基于优化思想,利用私有梯度信息进行权重选择,同时采用现成的隐私会计技术。在多个计算机视觉模型和数据集上的数值实验表明,该方法在预测准确率上优于全模型私有微调及现有私有稀疏微调方法。

原文摘要 · Abstract (English)

Differentially private stochastic gradient descent (DP-SGD) is broadly considered to be the gold standard for training and fine-tuning neural networks under differential privacy (DP). With the increasing availability of high-quality pre-trained model checkpoints (e.g., vision and language models), fine-tuning has become a popular strategy. However, despite recent progress in understanding and applying DP-SGD for private transfer learning tasks, significant challenges remain -- most notably, the performance gap between models fine-tuned with DP-SGD and their non-private counterparts. Sparse fine-tuning on private data has emerged as an alternative to full-model fine-tuning; recent work has shown that privately fine-tuning only a small subset of model weights and keeping the rest of the weights fixed can lead to better performance. In this work, we propose a new approach for sparse fine-tuning of neural networks under DP. Existing work on private sparse finetuning often used fixed choice of trainable weights (e.g., updating only the last layer), or relied on public model's weights to choose the subset of weights to modify. Such choice of weights remains suboptimal. In contrast, we explore an optimization-based approach, where our selection method makes use of the private gradient information, while using off the shelf privacy accounting techniques. Our numerical experiments on several computer vision models and datasets show that our selection method leads to better prediction accuracy, compared to full-model private fine-tuning or existing private sparse fine-tuning approaches.

隐私计算微调优化稀疏训练差分隐私

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