GNNs Also Deserve Editing, and They Need It More Than Once

Shaochen Zhong, Duy Le, Zirui Liu, Zhimeng Jiang, Andrew Ye, Jiamu Zhang, Jiayi Yuan, Kaixiong Zhou, Zhaozhuo Xu, Jing Ma, Shuai Xu, Vipin Chaudhary, Xia Hu

Research output: Contribution to journalConference articlepeer-review

2 Scopus citations

Abstract

Suppose a self-driving car is crashing into pedestrians, or a chatbot is instructing its users to conduct criminal wrongdoing; the stakeholders of such products will undoubtedly want to patch these catastrophic errors as soon as possible.To address such concerns, Model Editing: the study of efficiently patching model behaviors without significantly altering their general performance, has seen considerable activity, with hundreds of editing techniques developed in various domains such as CV and NLP.However, the graph learning community has objectively fallen behind with only a few Graph Neural Network-compatible - and just one GNN-specific - model editing methods available, where all of which are limited in their practical scope.We argue that the impracticality of these methods lies in their lack of Sequential Editing Robustness: the ability to edit multiple errors sequentially, and therefore fall short in effectiveness, as this setup mirrors how errors are discovered and addressed in the real world.In this paper, we delve into the specific reasons behind the difficulty of editing GNNs in succession and observe the root cause to be model overfitting.We subsequently propose a simple yet effective solution - SEED-GNN - by leveraging overfit-prevention techniques in a GNN-specific context to derive the first and only GNN model editing method that scales practically.Additionally, we formally frame the task paradigm of GNN editing and hope to inspire future research in this crucial but currently overlooked field.Please refer to our GitHub repository for code and checkpoints.

Original languageEnglish
Pages (from-to)61727-61746
Number of pages20
JournalProceedings of Machine Learning Research
Volume235
StatePublished - 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024

Fingerprint

Dive into the research topics of 'GNNs Also Deserve Editing, and They Need It More Than Once'. Together they form a unique fingerprint.

Cite this