arXiv:2409.13546cs.LGcs.AI2024-09被引 1

通过分块随机平滑提升模型鲁棒性,显著扩大认证防御范围。

Certified Adversarial Robustness via Partition-based Randomized Smoothing

  • 将图像分块后分别加噪,提升分类置信度
  • 在ImageNet等数据集上认证准确率明显提高
  • 适合需要强鲁棒性保证的AI安全场景

深度神经网络分类器在实际应用中需具备对抗扰动的鲁棒性。高斯平滑是广泛研究的认证鲁棒性方法,其认证预测半径取决于高斯噪声方差和模型在加噪输入下的置信度。然而,在高维图像数据集上,单纯使用高方差高斯噪声会严重损害图像可视性,导致认证半径较小。本文提出基于像素分块的随机平滑(PPRS)方法,通过分块加噪增强模型置信度,从而提升认证鲁棒半径。实验表明,该方法在标准计算机视觉数据集与主流神经网络架构上均有效提升了图像在加噪下的可视性,并显著提高了认证准确率与预测稳定性。

原文摘要 · Abstract (English)

A reliable application of deep neural network classifiers requires robustness certificates against adversarial perturbations. Gaussian smoothing is a widely analyzed approach to certifying robustness against norm-bounded perturbations, where the certified prediction radius depends on the variance of the Gaussian noise and the confidence level of the neural net's prediction under the additive Gaussian noise. However, in application to high-dimensional image datasets, the certified radius of the plain Gaussian smoothing could be relatively small, since Gaussian noise with high variances can significantly harm the visibility of an image. In this work, we propose the Pixel Partitioning-based Randomized Smoothing (PPRS) methodology to boost the neural net's confidence score and thus the robustness radius of the certified prediction. We demonstrate that the proposed PPRS algorithm improves the visibility of the images under additive Gaussian noise. We discuss the numerical results of applying PPRS to standard computer vision datasets and neural network architectures. Our empirical findings indicate a considerable improvement in the certified accuracy and stability of the prediction model to the additive Gaussian noise in randomized smoothing.

对抗鲁棒性随机平滑认证防御

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