提出可通用且可扩展的视觉自监督模型成员推理方法,突破黑盒攻击限制。
A Unified and Scalable Membership Inference Method for Visual Self-supervised Encoder via Part-aware Capability
- 基于物体局部响应特性,通过裁剪图像局部区域探测模型输出差异。
- 在多种训练协议与结构上实现高精度攻击,验证方法普适性。
- 适合研究隐私泄露与防御的学者,尤其关注自监督学习安全者。
自监督学习虽能有效利用大量无标签数据,但其隐私风险日益突出。本文在更贴近实际的黑盒场景下开展视觉自监督模型的成员推理攻击,即攻击者对训练方法和细节完全未知。针对不同自监督范式(如掩码图像建模、对比学习)及复杂训练细节,提出统一的成员推理方法 PartCrop。该方法基于模型共享的局部感知能力,利用图像中物体局部区域的更强响应特征进行查询。我们在三种主流图像数据集上对多种架构的自监督模型进行了广泛攻击测试,结果表明 PartCrop 具有优异的有效性与泛化能力。为进一步防御,评估了早停与差分隐私,并提出针对性方法——缩小区间裁剪尺度。实验显示所有防御策略均有效。此外,通过原型测试与小规模数据集验证后,我们进一步量化分析了数据与模型规模扩展对攻击的影响,提出改进版 PartCrop-v2,引入两项结构优化实现可扩展性。代码已开源:https://github.com/JiePKU/PartCrop。
原文摘要 · Abstract (English)
Self-supervised learning shows promise in harnessing extensive unlabeled data, but it also confronts significant privacy concerns, especially in vision. In this paper, we perform membership inference on visual self-supervised models in a more realistic setting: self-supervised training method and details are unknown for an adversary when attacking as he usually faces a black-box system in practice. In this setting, considering that self-supervised model could be trained by completely different self-supervised paradigms, e.g., masked image modeling and contrastive learning, with complex training details, we propose a unified membership inference method called PartCrop. It is motivated by the shared part-aware capability among models and stronger part response on the training data. Specifically, PartCrop crops parts of objects in an image to query responses within the image in representation space. We conduct extensive attacks on self-supervised models with different training protocols and structures using three widely used image datasets. The results verify the effectiveness and generalization of PartCrop. Moreover, to defend against PartCrop, we evaluate two common approaches, i.e., early stop and differential privacy, and propose a tailored method called shrinking crop scale range. The defense experiments indicate that all of them are effective. Finally, besides prototype testing on toy visual encoders and small-scale image datasets, we quantitatively study the impacts of scaling from both data and model aspects in a realistic scenario and propose a scalable PartCrop-v2 by introducing two structural improvements to PartCrop. Our code is at https://github.com/JiePKU/PartCrop.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。