Source-linked AI summary
Adversarial Attack on Hierarchical Graph Pooling Neural Networks
Haoteng Tang, Guixiang Ma, Yurong Chen, Lei Guo, Wei Wang, Bo Zeng, Liang Zhan
TL;DR
Robustness research on GNNs had focused mainly on node-level tasks, with little study of graph classification. This paper attacks hierarchical pooling models using a surrogate that perturbs preserved nodes; the samples transfer across graph-classification models, and robust training improves defense against them.
Problem
Most prior GNN robustness studies addressed node-level tasks, while robustness of graph-classification models remained little studied.
Source: §I · pp. 1–2Method
A convolution-and-pooling surrogate generates attacks by perturbing preserved nodes to make hierarchical pooling select different nodes.
Source: §I · p. 2, 1) Attack Loss Design · p. 5, 3) Gradient-Based Pooling Attack… · p. 5Results
35.49% reduction in surrogate-model classification accuracy on DD; the adversarial samples also transfer to different HGP models.
Source: B. Surrogate Model Attack · p. 7, 2) Transferability of Adversarial… · p. 7Takeaways & Limitations
Robustly retrained graph-classification models are reported to better defend against the generated adversarial samples.
Source: §I · p. 2The study evaluates only one type of GCN-based model; robustness of other graph-classification methods remains unverified.
Source: §VII · p. 9
Abstract
from arXiv · showhide
Recent years have witnessed the emergence and development of graph neural networks (GNNs), which have been shown as a powerful approach for graph representation learning in many tasks, such as node classification and graph classification. The research on the robustness of these models has also started to attract attentions in the machine learning field. However, most of the existing work in this area focus on the GNNs for node-level tasks, while little work has been done to study the robustness of the GNNs for the graph classification task. In this paper, we aim to explore the vulnerability of the Hierarchical Graph Pooling (HGP) Neural Networks, which are advanced GNNs that perform very well in the graph classification in terms of prediction accuracy. We propose an adversarial attack framework for this task. Specifically, we design a surrogate model that consists of convolutional and pooling operators to generate adversarial samples to fool the hierarchical GNN-based graph classification models. We set the preserved nodes by the pooling operator as our attack targets, and then we perturb the attack targets slightly to fool the pooling operator in hierarchical GNNs so that they will select the wrong nodes to preserve. We show the adversarial samples generated from multiple datasets by our surrogate model have enough transferability to attack current state-of-art graph classification models. Furthermore, we conduct the robust train on the target models and demonstrate that the retrained graph classification models are able to better defend against the attack from the adversarial samples. To the best of our knowledge, this is the first work on the adversarial attack against hierarchical GNN-based graph classification models.
I. INTRODUCTION
The paper addresses limited study of robustness for graph-classification GNNs by attacking hierarchical pooling models. Its surrogate attack targets preserved nodes, and the authors report transferability across graph-classification models and improved defense after robust training.
I. INTRODUCTION
- Most prior GNN attack work addressed node-level tasks, while robustness of graph-classification models had received little study.The introduction identifies hierarchical GNNs as especially relevant to graph classification.Source: §I · pp. 1–2
- The authors identify effective target selection and costly loss-gradient computation as challenges for designing surrogate attacks.The attack framework uses a pooling-score-based loss to avoid computing gradients of the entire network loss for each candidate edge.Source: §I · p. 2
- The paper presents its framework as the first adversarial attack against hierarchical GCN-based graph-classification models.It focuses on HGP models as instances of hierarchical GCNs.Source: §I · p. 2
- The attack uses a convolution-and-pooling surrogate to perturb preserved nodes so the target pooling operator selects the wrong nodes.The method targets nodes selected for preservation and uses a pooling-score-based loss to reduce computation when identifying edge changes.Source: §I · p. 2
- The generated adversarial samples transfer to state-of-the-art graph-classification models, and robustly retrained models defend against the samples.The evaluation covers multiple datasets; the introduction reports both transferability and defense after robust training.Source: §I · p. 2
II. RELATED WORK
Prior graph attacks largely addressed node or edge tasks, with few graph-level studies and concerns about generalization and reinforcement-learning efficiency. The paper then sets out attributed-graph notation and the hierarchical graph pooling model used in its study.
II. RELATED WORK
- Graph-level attacks were rare; prior graph-classification work used reinforcement learning but was evaluated only on synthetic data, and RL efficiency was a concern.The paper chooses evasion attacks, which generate adversarial samples against a trained model with fixed parameters.Source: §II · pp. 2–3
A. Graph Notation
- The supervised setting represents each attributed graph as G = (A, H), with an adjacency matrix, node-feature matrix, and graph class label.Graphs may have different node counts, and node order is treated as irrelevant to classification and attack.Source: A. Graph Notation · p. 3
B. Hierarchical Graph Pooling (HGP) Model
- HGP models combine convolution and pooling: pooling preserves a subset of nodes and aggregates discarded-node information into them.This hierarchical aggregation is presented as a way to capture graph structure beyond flat propagation through nodes and edges.Source: B. Hierarchical Graph Pooling (HGP)… · p. 3
- The model's convolution follows the previous GCN formulation, using the adjacency and degree matrices at each pooling level and trainable layer weights.The adjacency changes across layers because pooling discards nodes.Source: B. Hierarchical Graph Pooling (HGP)… · p. 3
- Pooling ranks nodes with a score function and preserves high-scoring nodes; the model's final linear layers produce class probabilities.The paper uses a unified pooling operator P to express a pooling layer.Source: B. Hierarchical Graph Pooling (HGP)… · p. 3
C. Adversarial Attacks on Graph
The graph evasion attack is framed as a bilevel optimization: train model parameters on original graphs, then optimize adversarial graphs against the fixed trained model.
C. Adversarial Attacks on Graph
- The evasion attack alternates a lower-level model-training objective with an upper-level adversarial objective evaluated using the trained parameters.The model loss is minimized during training; after training, the fixed model's attack loss is maximized over perturbed graphs.Source: C. Adversarial Attacks on Graph · p. 3
- The adversarial graph is constrained by perturbation restrictions relative to the original graph.The restriction is denoted δ(·).Source: C. Adversarial Attacks on Graph · p. 3
IV. METHODOLOGY
The methodology addresses the difficulty of differentiating through nonlinear hierarchical pooling by using a trainable pooling surrogate. It generates adversarial test samples, evaluates their transfer to target HGP models, and uses adversarial training to improve target-model robustness.
IV. METHODOLOGY
- Nonlinear pooling makes gradients of the attack loss with respect to graph inputs difficult to compute, motivating a new solution.Existing attack losses based on or equal to the model loss make the upper-level attack objective difficult to optimize.Source: §IV · pp. 3–4
A. Workflow of Adversarial Attack
- Figure 1 distinguishes adversarial-sample generation, workflow, and neural-network propagation with red, green, and black arrows, respectively.Source: Figure 1 · p. 4
- The workflow trains a surrogate on original data, generates adversarial test samples, tests them on the surrogate and target HGP models, then retrains target models with adversarial training samples.The target-model attack tests transferability; the paper says robust training is used to improve target-model robustness.Source: A. Workflow of Adversarial Attack · p. 4
B. Surrogate Model
- Because many existing pooling layers are not trainable, the authors use a parameterized self-attention-inspired pooling surrogate to facilitate gradient computation.The surrogate has one convolution layer, one pooling layer, and two fully connected output layers.Source: B. Surrogate Model · p. 4
- In the surrogate, convolution encodes node features before pooling scores nodes, preserves the top K, and sends pooled features to linear layers for graph classification.The node score function is parameterized, and the select function discards unselected nodes.Source: B. Surrogate Model · p. 4
C. Generating Adversarial Samples
The attack targets HGP's node-score-based pooling: it seeks to lower scores of preserved nodes relative to discarded nodes, using a gradient-based loss designed to avoid differentiating through the full nonlinear model loss.
C. Generating Adversarial Samples
- The authors design a new attack loss to avoid computing gradients through the complicated nonlinear model loss.They describe directly using Lmodel as tedious for gradient-ascent optimization of the upper-level objective.Source: C. Generating Adversarial Samples · pp. 4–5
- The attack loss targets preserved nodes by seeking to lower their scores below those of discarded nodes, so pooling selects different nodes and captures different hierarchical information.The paper links this wrong-node selection to wrong prediction outputs.Source: 1) Attack Loss Design · p. 5
- Graph-edge changes are restricted by edit-distance and second-order DELTACON0 budgets, while node-feature changes are restricted by an l1-norm budget.The budgets are denoted △1, △2, and △3, respectively.Source: 2) Unnoticeable Perturbation Restriction · p. 5
- After training the surrogate and fixing its parameters, the attack computes loss gradients for the selected top K nodes and perturbs their features and edges.For each selected node, edge gradients are computed across possible neighbors; the largest absolute edge gradient determines an add or delete operation, while the node feature is updated using its gradient.Source: 3) Gradient-Based Pooling Attack… · p. 5
- The edge-update procedure skips an add or delete when the edge already has the requested state, moving to the next-largest gradient until a valid change is found or the perturbation restriction is reached.Source: 3) Gradient-Based Pooling Attack… · p. 5
D. Computation of Attack Gradient
The gradient-computation section formulates derivatives of the attack objective with respect to a selected node's features and incident adjacency entries. Its proofs identify the terms relevant to each derivative.
D. Computation of Attack Gradient
- For a selected node, the computation considers the attack-loss gradient with respect to its feature vector and each adjacency entry in its row.The section defines the selected node as the attack target and identifies the node feature vector as the corresponding row of H.Source: D. Computation of Attack Gradient · pp. 5–6
- For the feature-gradient derivation, the original-graph score term is constant, leaving only the second term to differentiate.Source: D. Computation of Attack Gradient · p. 6
- The section introduces an assist degree matrix and its elementwise square as part of its matrix definitions.Source: D. Computation of Attack Gradient · p. 6
- For the adjacency-gradient derivation, only the second term, −S(A,H)i, depends on the gradient computation.Source: D. Computation of Attack Gradient · p. 6
V. EXPERIMENTS
The experiments cover datasets, surrogate-model attacks, transferability, attack power, feature-versus-edge attacks, and robustness defenses.
- The experiments are organized around dataset descriptions and adversarial-sample generation and surrogate-model attacks.Source: §V · p. 6
- Transferability and attack-power analysis examine the generated adversarial samples.Source: §V · p. 6
- The final parts compare feature and edge attacks and examine model robustness and defense.Source: §V · p. 6
A. Dataset
The study uses six graph datasets covering protein structures, chemical molecules and bonds, and biological compounds; their statistics are summarized in Table I.
- Six datasets from the Benchmark Data Sets for Graph Kernels cover protein structures, chemical molecular structures, chemical bonds, and biological compounds.Source: A. Dataset · p. 6
- DD graphs represent protein crystal structures, with labels indicating whether the protein is an enzyme.Source: A. Dataset · p. 6
- Table I summarizes the dataset statistics.Source: Table I · p. 7
B. Surrogate Model Attack
The surrogate model is trained and evaluated in an inductive setting, then used to generate adversarial test samples; under 5% node perturbation, attacks substantially reduce classification accuracy across six datasets.
- In the inductive setting, each dataset is split into 80% training, 10% validation, and 10% testing data.Source: B. Surrogate Model Attack · p. 7
- Adversarial test samples are generated from a fixed trained model under perturbation restrictions intended to keep similarity above 80%.The setting uses Δ1 = 0.05, Δ2 = 0.25, ε = 0.0001, and K = 50%.Source: B. Surrogate Model Attack · p. 7
- Classification accuracy fell by 35.49% on DD, 33.86% on Mutagenicity, 9.46% on ER MD, 10.91% on DHFR, 27.69% on AIDS, and 9.10% on BZR after attack.The comparison is with original training results and uses 5% node perturbation.Source: B. Surrogate Model Attack · p. 7, Table II · p. 7
C. Transferability of the Adversarial Samples
The experiments assess transfer to two target HGP models, attack strength as edge perturbations increase, and comparisons among feature, edge, and combined attacks.
C. Transferability of the Adversarial Samples
- Adversarial samples generated by the surrogate transfer to HGP-SL and SAG, with the paper attributing this to perturbations of nodes supporting graph hierarchy.The samples can encode different latent hierarchical structure despite looking similar to the originals.Source: 2) Transferability of Adversarial… · p. 7, Table III · p. 7
D. Attack Power Analysis
- The attack-strength analysis limits perturbations to 25% of edges, except for ER MD and DD, where all selected top-K nodes are attacked.The processes are conducted under the unnoticeable restriction.Source: D. Attack Power Analysis · p. 7
- Attack power rises rapidly near 5% edge perturbation, then changes little as additional edges are perturbed.The authors attribute the pattern to early perturbation of important edges that hold hierarchical structure; Figure 2 plots edge perturbation percentage against classification accuracy for SAG and HGP-SL.Source: D. Attack Power Analysis · p. 7, Figure 2 · p. 8
E. Feature Attack v.s. Edge Attack
- The feature-versus-edge experiment uses AIDS and BZR, which include specific node features, and applies a feature-perturbation restriction of Δ3 = 0.05.The authors first attack features with original edges, then combine feature and edge attacks.Source: 1) Experiment Setting · p. 8
- On AIDS and BZR, edge attacks are more powerful than node-feature attacks, while combined attacks are most powerful across datasets and target models.Figure 3 compares node-feature, edge, and combined attacks; its axes are edge perturbation percentage and classification error rate.Source: 2) Feature Attack v.s. Edges Attack · p. 8, Figure 3 · p. 8
F. Robust Train
On DD and Mutagenicity, retraining improved performance on adversarial tests with little performance cost on mixed samples. The evaluation reports significant structural differences between original and adversarial graphs, while the conclusion identifies limits in model scope and perturbation measurement.
F. Robust Train
- Robust-trained models improved performance on adversarial samples with little performance cost on mixed samples.Training mixed original and adversarial data was evaluated on adversarial test data.Source: 2) Robust Train Analysis · p. 9
VI. EVALUATION
- The evaluation compares original and adversarial node-degree distributions using mean KL divergence, where lower values indicate smaller differences.The attack is evaluated in terms of perturbation and its effect on graph hierarchy.Source: §VI · p. 9
- All p-values for GRC and community-louvain comparisons were below 0.05, indicating significant hierarchical-structure differences between original and adversarial samples.Figure 5 illustrates node-degree distributions and hierarchical structure for an example DD graph.Source: §VI · p. 9, Figure 5 · p. 9
VII. CONCLUSION
- The study evaluates only one type of GCN-based method, leaving other graph-classification approaches such as graph kernels and topology methods unverified.The authors identify broader attack strategies for other graph-classification models as future work.Source: §VII · p. 9
- The authors identify the lack of a uniform criterion for measuring unnoticeable graph perturbations and discrete optimization for sample generation as future-work challenges.They call for a more convincing perturbation criterion and a better algorithm for the discrete optimization problem.Source: §VII · p. 10
Related papers
This paper cites
- Going Deeper with Convolutions2014
- Semi-Supervised Classification with Graph Convolutional Networks2016
- Neural Message Passing for Quantum Chemistry2017
- Fast unfolding of communities in large networks2008
- Simplifying Graph Convolutional Networks2019
- Deep Convolutional Networks on Graph-Structured Data2015