Abstract:Diffusion models have been recently employed to improve certified robustness through the process of denoising. However, the theoretical understanding of why diffusion models are able to improve the certified robustness is still lacking, preventing from further improvement. In this study, we close this gap by analyzing the fundamental properties of diffusion models and establishing the conditions under which they can enhance certified robustness. This deeper understanding allows us to propose a new method DensePure , designed to improve the certified robustness of a pretrained model (i.e. classifier). Given an (adversarial) input, DensePure consists of multiple runs of denoising via the reverse process of the diffusion model (with different random seeds) to get multiple reversed samples, which are then passed through the classifier, followed by majority voting of inferred labels to make the final prediction. This design of using multiple runs of denoising is informed by our theoretical analysis of the conditional distribution of the reversed sample. Specifically, when the data density of a clean sample is high, its conditional density under the reverse process in a diffusion model is also high; thus sampling from the latter conditional distribution can purify the adversarial example and return the corresponding clean sample with a high probability. By using the highest density point in the conditional distribution as the reversed sample, we identify the robust region of a given instance under the diffusion model’s reverse process. We show that this robust region is a union of multiple convex sets, and is potentially much larger than the robust regions identified in previous works. In practice, DensePure can approximate the label of the high density region in the conditional distribution so that it can enhance certified robustness. We conduct extensive experiments to demonstrate the effectiveness of DensePure by evaluating its certified robustness given a standard model via randomized smoothing. We show that DensePure is consistently better than existing methods on ImageNet, with 7% improvement on average. Project

Certified Distributional Robustness on Smoothed Classifiers

Regularized Training and Tight Certification for Randomized Smoothed Classifier with Provable Robustness

Certified Adversarial Robustness Under the Bounded Support Set.

DiffSmooth: Certifiably Robust Learning via Diffusion Models and Local Smoothing

Certified Adversarial Robustness Within Multiple Perturbation Bounds

Tight Certificates of Adversarial Robustness for Randomly Smoothed Classifiers

Projected Randomized Smoothing for Certified Adversarial Robustness

Your Diffusion Model is Secretly a Certifiably Robust Classifier

Adversarial robustness via noise injection in smoothed models

Towards Bridging the gap between Empirical and Certified Robustness against Adversarial Examples

On Certifying Robustness against Backdoor Attacks via Randomized Smoothing

DRF: Improving Certified Robustness Via Distributional Robustness Framework

Center Smoothing: Certified Robustness for Networks with Structured Outputs

Diffusion Models are Certifiably Robust Classifiers

DensePure: Understanding Diffusion Models for Adversarial Robustness

Certified $\ell_2$ Attribution Robustness via Uniformly Smoothed Attributions

Integrating uncertainty quantification into randomized smoothing based robustness guarantees

On the Certified Robustness for Ensemble Models and Beyond

GSmooth: Certified Robustness against Semantic Transformations via Generalized Randomized Smoothing

Certified Adversarial Robustness of Machine Learning-based Malware Detectors via (De)Randomized Smoothing

Towards Assessment of Randomized Smoothing Mechanisms for Certifying Adversarial Robustness