Should We Pay This Much for Robustness? Efficient Proxy Certificates with Marginal Guarantees
Organizations: CISPA Helmholtz Center for Information Security · University of Cologne
Abstract
Modern classifiers remain sensitive to small input perturbations, while strong model-agnostic robustness certificates such as randomized smoothing are often too expensive for deployment, requiring an extensive number (e.g. 2000) of model forward passes per input. We propose a framework that delegates certification to a cheap proxy function while statistically controlling its deviation from the original sound certificate. Given an exchangeable unlabeled calibration set, we tune the proxy via robust conformal risk control to guarantee a bounded marginal (rather than pointwise) false-approval rate: the probability, over future inputs, that the proxy accepts an input whose prediction is flipped due to perturbation. The resulting proxy certificate can be used either as a standalone marginal certificate or as a fallback mechanism that invokes the expensive certificate only when the proxy rejects. We provide both single-sample and sample-heavy calibration recipes. Empirically, as a standalone certificate our method reduces the test-time certification cost to a single model forward while retaining much of the certified performance of randomized smoothing at small radii, and as a fallback it recovers the certified ratio of randomized smoothing at a more modest speedup. This enables cheap yet effective robustness guarantees for larger architectures such as vision transformers and language models.
Figures & tables
| Method | Variant | Guarantee | Cert. Ratio | Cert. Accuracy | Error Rate | Forwards / image | ||
| Cal | Cert.-only (Speedup) | End-to-end (Speedup) | ||||||
| Rand. Smoothing | — | Pointwise | 70.06% | 56.08% | — | — | 1950 | 2000 |
| Verification | Hu et al. (2023) | Pointwise | — | 35.0% | — | — | 1 | 1 |
| ProxC | Standalone | Marginal | 56.11% | 48.53% | 2.29% | 2000 | 1 | 51 |
| Fallback | Marginal | 72.35% | 57.3% | 2.29% | 2000 | 779 | 829 | |
| RS (Pointwise) | ProxC (Marginal 5% Risk) | Ref. CA | |||||
| Model | CR | CA | CR | CA | Risk | ||
| ResNet-50 | — | ||||||
| — | |||||||
| — | |||||||
| — | |||||||
| — | — | ||||||
| Base (Cohen) | ProxC-H Proxy CR (FA) | ||||||
| CR | CA | ||||||
| ( ) | ( ) | ( ) | ( ) | ( ) | |||
| ( ) | ( ) | ( ) | ( ) | ( ) | |||
| ( ) | ( ) | ( ) | ( ) | ( ) | |||
| — | — | ( ) | ( ) | ( ) | |||
| Method | Metric | ||||||
| Backbone | CR | % | % | % | % | % | % |
| CA | % | % | % | % | % | % | |
| ProxC | CR | % | % | % | % | % | % |
| CA | % | % | % | % | % | % | |
| Risk | – | % | % | % | % | % | % |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Prediction rule | Randomness | Averaged over | Guarantee |
| RS | Majority vote, small | MC-corrected | — (pointwise) | Pointwise soundness |
| ProxC | Same as RS | One fresh smoothing noise at proxy evaluation | Clean input, calibration set, internal noise | Marginal false-approval control (not pointwise) |
| ProxC-H | Same as ProxC | One fresh noise draw at test | Same as ProxC, plus calibration MC | Same marginal guarantee, usually less conservative |
| ProxC-A | One-sample noise-augmented inference; approves with the same draw | One fresh draw affects both prediction and approval | Clean input, calibration set, noises | Controls the joint event {randomized prediction toggles} {proxy approves}; not a certificate for a fixed deterministic prediction |
| ProxC | ProxC-A | |||||
| CR | CA | Risk | CR | CA | Risk | |
| RS (backbone) | ProxC-H | ProxC | ||||||
| CR | CA | CR | CA | Risk | CR | CA | Risk | |
| RS (backbone) | ProxC ( ) | ProxC ( ) | ProxC ( ) | ||||||||
| CR | CA | CR | CA | Risk | CR | CA | Risk | CR | CA | Risk | |
| Variant | Label | Certification-only | End-to-end |
| Randomized smoothing (backbone) | 50 | 1950 | 2000 |
| ProxC/ ProxC-H (standalone) | 50 | 1 | 51 |
| ProxC-A | shared | 1 | 1 |
| ProxC/ ProxC-H (fallback) | 50 | certification-only |
| Certification-only | End-to-end | ||||
| Reject rate | Forwards | Speedup | Forwards | Speedup | |
| 0.00 | 24.2% | 473 | 4.12 | 523 | 3.82 |
| 0.06 | 32.1% | 627 | 3.11 | 677 | 2.95 |
| 0.12 | 38.6% | 753 | 2.59 | 803 | 2.49 |
| 0.14 | 39.9% | 779 | 2.50 | 829 | 2.41 |
| 0.18 | 45.7% | 893 | 2.18 | 943 | 2.12 |
| Group | #points | Proxy risk | Proxy cert. ratio | |
| 0.12 | top 100 | 4,577 | 0.0 | 0.5724 |
| 0.12 | worst 100 | 10,654 | 0.0428 | 0.6555 |
| 0.25 | top 100 | 7,023 | 0.0 | 0.5090 |
| 0.25 | worst 100 | 8,025 | 0.0320 | 0.5499 |
| 0.50 | top 100 | 12,229 | 0.0 | 0.2728 |
| 0.50 | worst 100 | 2,613 | 0.0168 | 0.3750 |
| ( ) | ( ) | |||||
| Class | Risk | CR | FA/appr | Risk | CR | FA/appr |
| airplane | 3.27 | 59.9 | 5.5 | 2.25 | 44.9 | 5.0 |
| automobile | 3.37 | 67.7 | 5.0 | 2.59 | 54.7 | 4.7 |
| bird | 2.73 | 32.3 | 8.5 | 1.18 | 16.7 | 7.1 |
| cat | 2.75 | 23.0 | 12.0 | 0.82 | 7.9 | 10.4 |
| deer | 3.16 | 25.4 | 12.4 | 1.09 | 11.1 | 9.8 |
| , RS CR | , RS CR | , RS CR | , RS CR | |||||
| Risk | CR | Risk | CR | Risk | CR | Risk | CR | |
| 25 | infeasible | infeasible | infeasible | |||||
| 50 | infeasible | infeasible | ||||||
| 100 | infeasible | |||||||
| 200 | infeasible | |||||||
Certified Robustness
Certified robustness develops formal guarantees that model predictions remain reliable under specified input or system perturbations.
See all 94 papers