A Certified Robustness Framework for Backdoor Poisoning Attacks using -Divergences
Organizations: Northeastern University
Abstract
Machine learning systems trained on untrusted data are vulnerable to data and backdoor poisoning attacks, in which an adversary corrupts training and test samples to manipulate predictions at inference time. Randomized smoothing yields provable defenses against such attacks, but existing applications to train-time robustness are limited to specific threat models and smoothing mechanisms. We propose a general framework that allows expressing complex adversary constraints on the number of corrupted training examples, feature and label perturbation magnitudes, and test-time corruptions. Our framework randomizes inputs by combining data subsampling with feature, label and test-sample noising. We develop a composable certification procedure based on -divergences that handles arbitrary combinations of threat models and noising strategies, and recovers several prior certification regimes exactly as special cases. We demonstrate our framework experimentally by certifying, for the first time, robustness under general -constrained poisoning adversaries under bagging with additive Laplace, Gaussian, and Uniform noises.
Figures & tables
| Noise | Adjacency | ||
|---|---|---|---|
| MNIST | MNIST-17 | CIFAR10 | CIFAR10-02 | |||||||||||||
| (%) | ||||||||||||||||
| Bagging | 82.3 | 56.8 | 31.9 | 12.0 | 99.1 | 98.0 | 0.0 | 0.0 | 32.5 | 14.7 | 5.3 | 1.1 | 78.3 | 64.4 | 40.2 | 2.4 |
| Ours, | 81.2 | 50.0 | 23.5 | 5.1 | 98.7 | 96.6 | 48.6 | 0.0 | 22.3 | 9.8 | 3.3 | 0.6 | 66.9 | 48.5 | 26.6 | 0.6 |
| Ours, | 78.7 | 44.8 | 17.9 | 3.7 | 98.8 | 93.0 | 35.3 | 5.3 | 16.7 | 6.4 | 1.7 | 0.1 | 64.2 | 45.4 | 27.5 | 3.4 |
| MNIST | MNIST-17 | CIFAR10 | CIFAR10-02 | |||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| (%) | ||||||||||||||||
| RAB, | 74.1 | 0.0 | 0.0 | 0.0 | 95.9 | 0.0 | 0.0 | 0.0 | 18.4 | 0.0 | 0.0 | 0.0 | 63.5 | 0.0 | 0.0 | 0.0 |
| RAB, | 66.5 | 0.0 | 0.0 | 0.0 | 94.7 | 0.0 | 0.0 | 0.0 | 19.3 | 0.0 | 0.0 | 0.0 | 67.5 | 0.0 | 0.0 | 0.0 |
| Ours, | 48.6 | 42.2 | 20.5 | 5.0 | 92.9 | 88.0 | 76.6 | 41.8 | 6.8 | 5.6 | 3.8 | 2.5 | 40.9 | 34.4 | 27.5 | 20.2 |
| Ours, | 41.9 | 36.6 | 17.8 | 6.2 | 91.3 | 86.1 | 76.1 | 45.4 | 8.7 | 7.9 | 6.5 | 5.3 | 54.0 | 51.4 | 48.7 | 45.8 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| -Divergence | Regime | Generator | Integral form |
|---|---|---|---|
| Total variation | |||
| Hockey-stick | |||
| Kullback–Leibler | |||
| Squared Hellinger | |||
| Hellinger | |||
| Pearson |
| Divergence | Tensorization | Notes | |
|---|---|---|---|
| Hellinger integral | |||
| Rényi | |||
| Kullback–Leibler | case above | ||
| Squared Hellinger | case above | ||
| Pearson | case above | ||
| Neyman | Item B.2(ii) w/ |
| Work | Threat vector | Mechanism | Certificate |
|---|---|---|---|
| Jia et al. [2021] | bagging, none, none | identical ( Theorem E.1 ) | |
| Rosenfeld et al. [2020] | full set, none, flipping | tight a | |
| Zhang et al. [2022] | bagging, flipping, flipping | identical ( Theorem E.2 ) | |
| Weber et al. [2023] | aggregate on | full set, Gaussian, none | identical b |
| Joint feature & label | Backdoor | |||||
|---|---|---|---|---|---|---|
| Dataset | ||||||
| MNIST | 30 | 0.25 | 0.5 | 300 | 0.75 | 1.0 |
| MNIST-17 | 30 | 0.5 | 1.0 | 100 | 0.75 | 1.0 |
| CIFAR10 | 200 | 0.2 | 0.35 | 1000 | 0.5 | 0.75 |
| CIFAR10-02 | 25 | 0.35 | 0.5 | 100 | 0.35 | 0.5 |
Certified Robustness
Certified robustness develops formal guarantees that model predictions remain reliable under specified input or system perturbations.
See all 86 papers