Guarding Graph Neural Networks Against Backdoor Attacks-A Training Loss Dynamics Approach
摘要
This paper investigates the problem of defending deep graph neural networks against backdoor attacks, a critical challenge in secure AI systems. With the widespread use of pre-trained models and open-source datasets, data contamination poses a significant threat to deep learning models. To address this issue, we propose a Data-Cleaning-Based Defense algorithm (DCBD) that monitors the dynamic difference of training loss between clean data and poisoned data during GNN training to effectively remove poisoned data. Notably, DCBD identifies poisoned data at an early stage in the training process, terminating learning on poisoned inputs. Our algorithm does not require prior knowledge of the proportion of contaminated data, making it more practical for real-world applications. Experimental results show that DCBD achieves high data cleaning accuracy and outperforms baseline algorithms in both attack success rate reduction and downstream tasks’ performance.