English

Mitigating the Curse of Dimensionality for Certified Robustness via Dual Randomized Smoothing

Computer Vision and Pattern Recognition 2024-06-18 v4 Machine Learning

Abstract

Randomized Smoothing (RS) has been proven a promising method for endowing an arbitrary image classifier with certified robustness. However, the substantial uncertainty inherent in the high-dimensional isotropic Gaussian noise imposes the curse of dimensionality on RS. Specifically, the upper bound of 2{\ell_2} certified robustness radius provided by RS exhibits a diminishing trend with the expansion of the input dimension dd, proportionally decreasing at a rate of 1/d1/\sqrt{d}. This paper explores the feasibility of providing 2{\ell_2} certified robustness for high-dimensional input through the utilization of dual smoothing in the lower-dimensional space. The proposed Dual Randomized Smoothing (DRS) down-samples the input image into two sub-images and smooths the two sub-images in lower dimensions. Theoretically, we prove that DRS guarantees a tight 2{\ell_2} certified robustness radius for the original input and reveal that DRS attains a superior upper bound on the 2{\ell_2} robustness radius, which decreases proportionally at a rate of (1/m+1/n)(1/\sqrt m + 1/\sqrt n ) with m+n=dm+n=d. Extensive experiments demonstrate the generalizability and effectiveness of DRS, which exhibits a notable capability to integrate with established methodologies, yielding substantial improvements in both accuracy and 2{\ell_2} certified robustness baselines of RS on the CIFAR-10 and ImageNet datasets. Code is available at https://github.com/xiasong0501/DRS.

Keywords

Cite

@article{arxiv.2404.09586,
  title  = {Mitigating the Curse of Dimensionality for Certified Robustness via Dual Randomized Smoothing},
  author = {Song Xia and Yi Yu and Xudong Jiang and Henghui Ding},
  journal= {arXiv preprint arXiv:2404.09586},
  year   = {2024}
}

Comments

This paper has been accepted to the International Conference on Learning Representations (ICLR), 2024