Exact Verification of Graph Neural Networks with Incremental Constraint Solving
About
Graph neural networks (GNNs) are increasingly employed in high-stakes applications, such as fraud detection or healthcare, but are susceptible to adversarial attacks. A number of techniques have been proposed to provide adversarial robustness guarantees, but support for commonly used aggregation functions in message-passing GNNs is lacking. In this paper, we develop an exact (sound and complete) verification method for GNNs to compute guarantees against attribute and structural perturbations that involve edge addition or deletion, subject to budget constraints. Our method employs constraint solving with bound tightening, and iteratively solves a sequence of relaxed constraint satisfaction problems while relying on incremental solving capabilities of solvers to improve efficiency. We implement GNNev, a versatile exact verifier for message-passing neural networks, which supports three aggregation functions, sum, max and mean, with the latter two considered here for the first time. Extensive experimental evaluation of GNNev on real-world fraud datasets (Amazon and Yelp) and biochemical datasets (MUTAG and ENZYMES) demonstrates its usability and effectiveness, as well as superior performance for node classification and competitiveness on graph classification compared to existing exact verification tools on sum-aggregated GNNs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification Robustness Certification | Texas | Solved Instances Count (All)655 | 45 | |
| Node Classification Robustness Certification | Cornell | Solved Count (All Instances)675 | 36 | |
| Node Classification Robustness Certification | Wisconsin | Time (s) (All Instances)13.86 | 36 | |
| Node Classification Robustness Certification | Cora | Solved Instances Count (All)8.12e+3 | 30 | |
| Node Classification Robustness Certification | Citeseer | Count Solved (All Instances)3.17e+3 | 18 | |
| Node Classification Robustness Verification | CiteSeer Robust instances Delta=10 | Solved Count2.60e+3 | 9 | |
| Node Classification Robustness Verification | Texas Robust instances Delta=10 | Solved Count112 | 9 | |
| Node Classification Robustness | Wisconsin Δ = 5 | #Solved (All)205 | 9 | |
| Node Classification Robustness Verification | Cora Delta=10 (All instances) | Solved Count2.00e+3 | 9 | |
| Node Classification Robustness Verification | Cornell Delta=10 (All instances) | Solved Instances171 | 9 |