通过优先探索更可能含反例的子问题,大幅加速神经网络验证
Efficient Neural Network Verification via Order Leading Exploration of Branch-and-Bound Trees
- 按反例出现概率排序子问题,优先检查高风险区域
- 在MNIST上最快提速25倍,CIFAR10上达80倍
- 适合需要高效验证神经网络安全性的研究人员
神经网络对对抗扰动的脆弱性催生了形式化验证技术以严格证明其可靠性。当前最先进的分支定界(BaB)方法采用‘分而治之’策略,将问题分解为子问题并调用现有验证器求解。然而,该方法以‘先到先服务’顺序遍历子问题空间,效率低下。为此,本文提出新框架Oliva,根据子问题包含反例的可能性建立优先级顺序,优先探索高风险子问题以快速得出验证结论。即使未发现反例,也仅改变访问顺序,不影响整体性能。Oliva有两种变体:$Oliva^{GR}$采用贪心策略始终优先高风险子问题;$Oliva^{SA}$受模拟退火启发,逐步从探索转向利用,寻找全局最优解。我们在690个涵盖5个模型、使用MNIST和CIFAR10数据集的验证任务上评估,相比现有方法,Oliva在MNIST上最高提速25倍,在CIFAR10上最高提速80倍。
原文摘要 · Abstract (English)
The vulnerability of neural networks to adversarial perturbations has necessitated formal verification techniques that can rigorously certify the quality of neural networks. As the state-of-the-art, branch and bound (BaB) is a "divide-and-conquer" strategy that applies off-the-shelf verifiers to sub-problems for which they perform better. While BaB can identify the sub-problems that are necessary to be split, it explores the space of these sub-problems in a naive "first-come-first-serve" manner, thereby suffering from an issue of inefficiency to reach a verification conclusion. To bridge this gap, we introduce an order over different sub-problems produced by BaB, concerning with their different likelihoods of containing counterexamples. Based on this order, we propose a novel verification framework Oliva that explores the sub-problem space by prioritizing those sub-problems that are more likely to find counterexamples, in order to efficiently reach the conclusion of the verification. Even if no counterexample can be found in any sub-problem, it only changes the order of visiting different sub-problem and so will not lead to a performance degradation. Specifically, Oliva has two variants, including $Oliva^{GR}$, a greedy strategy that always prioritizes the sub-problems that are more likely to find counterexamples, and $Oliva^{SA}$, a balanced strategy inspired by simulated annealing that gradually shifts from exploration to exploitation to locate the globally optimal sub-problems. We experimentally evaluate the performance of Oliva on 690 verification problems spanning over 5 models with datasets MNIST and CIFAR10. Compared to the state-of-the-art approaches, we demonstrate the speedup of Oliva for up to 25X in MNIST, and up to 80X in CIFAR10.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。