Robustness Inspired Graph Backdoor Defense

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Zhiwei, Lin, Minhua, Xu, Junjie, Wu, Zongyu, Dai, Enyan, Wang, Suhang
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910870862299136
author Zhang, Zhiwei
Lin, Minhua
Xu, Junjie
Wu, Zongyu
Dai, Enyan
Wang, Suhang
author_facet Zhang, Zhiwei
Lin, Minhua
Xu, Junjie
Wu, Zongyu
Dai, Enyan
Wang, Suhang
contents Graph Neural Networks (GNNs) have achieved promising results in tasks such as node classification and graph classification. However, recent studies reveal that GNNs are vulnerable to backdoor attacks, posing a significant threat to their real-world adoption. Despite initial efforts to defend against specific graph backdoor attacks, there is no work on defending against various types of backdoor attacks where generated triggers have different properties. Hence, we first empirically verify that prediction variance under edge dropping is a crucial indicator for identifying poisoned nodes. With this observation, we propose using random edge dropping to detect backdoors and theoretically show that it can efficiently distinguish poisoned nodes from clean ones. Furthermore, we introduce a novel robust training strategy to efficiently counteract the impact of the triggers. Extensive experiments on real-world datasets show that our framework can effectively identify poisoned nodes, significantly degrade the attack success rate, and maintain clean accuracy when defending against various types of graph backdoor attacks with different properties.
format Preprint
id arxiv_https___arxiv_org_abs_2406_09836
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Robustness Inspired Graph Backdoor Defense
Zhang, Zhiwei
Lin, Minhua
Xu, Junjie
Wu, Zongyu
Dai, Enyan
Wang, Suhang
Machine Learning
Cryptography and Security
Graph Neural Networks (GNNs) have achieved promising results in tasks such as node classification and graph classification. However, recent studies reveal that GNNs are vulnerable to backdoor attacks, posing a significant threat to their real-world adoption. Despite initial efforts to defend against specific graph backdoor attacks, there is no work on defending against various types of backdoor attacks where generated triggers have different properties. Hence, we first empirically verify that prediction variance under edge dropping is a crucial indicator for identifying poisoned nodes. With this observation, we propose using random edge dropping to detect backdoors and theoretically show that it can efficiently distinguish poisoned nodes from clean ones. Furthermore, we introduce a novel robust training strategy to efficiently counteract the impact of the triggers. Extensive experiments on real-world datasets show that our framework can effectively identify poisoned nodes, significantly degrade the attack success rate, and maintain clean accuracy when defending against various types of graph backdoor attacks with different properties.
title Robustness Inspired Graph Backdoor Defense
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2406.09836