BaB-prob: Branch and Bound with Preactivation Splitting for Probabilistic Verification of Neural Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Fangji, Tsiotras, Panagiotis
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915524222386176
author Wang, Fangji
Tsiotras, Panagiotis
author_facet Wang, Fangji
Tsiotras, Panagiotis
contents Branch-and-bound with preactivation splitting has been shown highly effective for deterministic verification of neural networks. In this paper, we extend this framework to the probabilistic setting. We propose BaB-prob that iteratively divides the original problem into subproblems by splitting preactivations and leverages linear bounds computed by linear bound propagation to bound the probability for each subproblem. We prove soundness and completeness of BaB-prob for feedforward-ReLU neural networks. Furthermore, we introduce the notion of uncertainty level and design two efficient strategies for preactivation splitting, yielding BaB-prob-ordered and BaB+BaBSR-prob. We evaluate BaB-prob on untrained networks, MNIST and CIFAR-10 models, respectively, and VNN-COMP 2025 benchmarks. Across these settings, our approach consistently outperforms state-of-the-art approaches in medium- to high-dimensional input problems.
format Preprint
id arxiv_https___arxiv_org_abs_2509_25647
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BaB-prob: Branch and Bound with Preactivation Splitting for Probabilistic Verification of Neural Networks
Wang, Fangji
Tsiotras, Panagiotis
Machine Learning
Artificial Intelligence
Branch-and-bound with preactivation splitting has been shown highly effective for deterministic verification of neural networks. In this paper, we extend this framework to the probabilistic setting. We propose BaB-prob that iteratively divides the original problem into subproblems by splitting preactivations and leverages linear bounds computed by linear bound propagation to bound the probability for each subproblem. We prove soundness and completeness of BaB-prob for feedforward-ReLU neural networks. Furthermore, we introduce the notion of uncertainty level and design two efficient strategies for preactivation splitting, yielding BaB-prob-ordered and BaB+BaBSR-prob. We evaluate BaB-prob on untrained networks, MNIST and CIFAR-10 models, respectively, and VNN-COMP 2025 benchmarks. Across these settings, our approach consistently outperforms state-of-the-art approaches in medium- to high-dimensional input problems.
title BaB-prob: Branch and Bound with Preactivation Splitting for Probabilistic Verification of Neural Networks
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.25647