一句话:提出 MpBP,通过多路径扩展经典界传播,高效验证神经网络鲁棒性。
神经网络在自动驾驶等安全关键场景中的鲁棒性需要保证。本文提出 MpBP,一个基于 PyTorch 的神经网络鲁棒性验证工具。其核心思想是在经典界传播方法(前向、后向及前向+后向)中引入多传播路径,以提升验证效果。MpBP 利用 PyTorch 和 GPU 实现高效并行验证,并提供友好的使用接口。作者在 MNIST、CIFAR-10 和 Tiny ImageNet 上训练的网络进行评估,结果表明 MpBP 的效果优于 LiRPA 和 GPUPoly 两个先进界传播工具,效率与 LiRPA 相当且显著高于 GPUPoly。代码和视频演示已公开。
Robustness of neural networks need be guaranteed in many safety-critical scenarios, such as autonomous driving and cyber-physical controlling. In this paper, we present MpBP, a tool for verifying the robustness of neural networks. MpBP is inspired by classical bound propagation methods for neural network verification, and aims to improve the effectiveness by exploiting the notion of propagation paths. Specifically, MpBP extends classical bound propagation methods, including forward bound propagation, backward bound propagation, and forward+backward bound propagation, with multiple propagation paths. MpBP is based on the widely-used PyTorch machine learning framework, hence providing efficient parallel verification on GPUs and user-friendly usage. We evaluate MpBP on neural networks trained on standard datasets MNIST, CIFAR-10 and Tiny ImageNet. The results demonstrate the effectiveness advantage of MpBP beyond two state-of-the-art bound propagation tools LiRPA and GPUPoly, with comparable efficiency to LiRPA and significantly higher efficiency than GPUPoly. A video demonstration that showcases the main features of MpBP can be found at https://youtu.be/3KyPMuPpfR8. Source code is available at https://github.com/formes20/MpBP and https://doi.org/10.5281/zenodo.7029261.