Erase Then Rectify: A Training-Free Parameter Editing Approach for Cost-Effective Graph Unlearning

Authors

  • Zhe-Rui Yang Sun Yat-sen University The Hong Kong University of Science and Technology (Guangzhou) Guangdong Key Laboratory of Big Data Analysis and Processing
  • Jindong Han The Hong Kong University of Science and Technology
  • Chang-Dong Wang Sun Yat-sen University Guangdong Key Laboratory of Big Data Analysis and Processing
  • Hao Liu The Hong Kong University of Science and Technology (Guangzhou) The Hong Kong University of Science and Technology

DOI:

https://doi.org/10.1609/aaai.v39i12.33423

Abstract

Graph unlearning, which aims to eliminate the influence of specific nodes, edges, or attributes from a trained Graph Neural Network (GNN), is essential in applications where privacy, bias, or data obsolescence is a concern. However, existing graph unlearning techniques often necessitate additional training on the remaining data, leading to significant computational costs, particularly with large-scale graphs. To address these challenges, we propose a two-stage training-free approach, Erase then Rectify (ETR), designed for efficient and scalable graph unlearning while preserving the model utility. Specifically, we first build a theoretical foundation showing that masking parameters critical for unlearned samples enables effective unlearning. Building on this insight, the Erase stage strategically edits model parameters to eliminate the impact of unlearned samples and their propagated influence on intercorrelated nodes. To further ensure the GNN's utility, the Rectify stage devises a gradient approximation method to estimate the model's gradient on the remaining dataset, which is then used to enhance model performance. Overall, ETR achieves graph unlearning without additional training or full training data access, significantly reducing computational overhead and preserving data privacy. Extensive experiments on seven public datasets demonstrate the consistent superiority of ETR in model utility, unlearning efficiency, and unlearning effectiveness, establishing it as a promising solution for real-world graph unlearning challenges.

Downloads

Published

2025-04-11

How to Cite

Yang, Z.-R., Han, J., Wang, C.-D., & Liu, H. (2025). Erase Then Rectify: A Training-Free Parameter Editing Approach for Cost-Effective Graph Unlearning. Proceedings of the AAAI Conference on Artificial Intelligence, 39(12), 13044–13051. https://doi.org/10.1609/aaai.v39i12.33423

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management II