arXiv:2506.13160cs.LGcs.AI2025-06被引 5

用可信预测方法验证模型是否训练过特定数据集。

CertDW: Towards Certified Dataset Ownership Verification via Conformal Prediction

  • 基于置信区间理论设计水印验证机制,确保稳定性。
  • 在像素扰动下仍能准确识别数据集归属,误判率低于5%。
  • 适合需防伪造的版权保护场景,如开源数据集维权。

深度神经网络依赖高质量开源数据集(如ImageNet)取得成功,因此数据集所有权验证(DOV)对保护公共数据集版权至关重要。现有方法隐含假设验证过程是可靠的,即可疑模型直接以验证样本为输入并返回结果。然而该假设在实际中未必成立,面对有意或无意的扰动时性能可能急剧下降。为此,我们提出首个可认证的数据集水印(CertDW)及基于此的认证所有权验证方法,可在特定条件下(如受限像素级扰动)抵御恶意攻击,保证验证可靠性。受置信区间理论启发,我们引入主概率(PP)和水印鲁棒性(WR)两个统计量,评估模型在正常与加水印样本上的预测稳定性。我们证明了PP与WR之间存在可证明的下界关系:若可疑模型的WR显著高于多个无水印数据集训练的良性模型的PP值,且低于阈值的PP数量超过设定值,则判定该模型曾训练于被保护数据集。大量实验在基准数据集上验证了方法的有效性及其对抗潜在自适应攻击的鲁棒性。代码已开源。

原文摘要 · Abstract (English)

Deep neural networks (DNNs) rely heavily on high-quality open-source datasets (e.g., ImageNet) for their success, making dataset ownership verification (DOV) crucial for protecting public dataset copyrights. In this paper, we find existing DOV methods (implicitly) assume that the verification process is faithful, where the suspicious model will directly verify ownership by using the verification samples as input and returning their results. However, this assumption may not necessarily hold in practice and their performance may degrade sharply when subjected to intentional or unintentional perturbations. To address this limitation, we propose the first certified dataset watermark (i.e., CertDW) and CertDW-based certified dataset ownership verification method that ensures reliable verification even under malicious attacks, under certain conditions (e.g., constrained pixel-level perturbation). Specifically, inspired by conformal prediction, we introduce two statistical measures, including principal probability (PP) and watermark robustness (WR), to assess model prediction stability on benign and watermarked samples under noise perturbations. We prove there exists a provable lower bound between PP and WR, enabling ownership verification when a suspicious model's WR value significantly exceeds the PP values of multiple benign models trained on watermark-free datasets. If the number of PP values smaller than WR exceeds a threshold, the suspicious model is regarded as having been trained on the protected dataset. Extensive experiments on benchmark datasets verify the effectiveness of our CertDW method and its resistance to potential adaptive attacks. Our codes are at \href{https://github.com/NcepuQiaoTing/CertDW}{GitHub}.

数据版权水印验证置信区间模型安全

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