Research map / Graph learning / GNN architectures
Erase Then Rectify: A Training-Free Parameter Editing Approach for Cost-Effective Graph Unlearning
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes ETR, a training-free two-stage graph unlearning method that edits GNN parameters identified via Fisher information (Erase) and then takes one gradient step using an approximated remaining-data gradient (Rectify).
Problem
Graph unlearning removes the influence of nodes, edges, or features from a trained GNN while preserving utility, but existing methods typically require additional training on the remaining data, which is costly on large graphs.
Approach
Identify parameters critical to the unlearned set and its k-hop neighborhood using diagonal Fisher information, scale them down via a neighborhood-aware editing rule (Erase), then approximate the remaining-data gradient from stored full-data gradients and the induced subgraph of the unlearned nodes, and apply one gradient-descent step (Rectify), with no retraining.
Key results
- On node unlearning with GCN on PubMed, ETR achieves F1 89.48±0.34 vs. 89.74±0.27 for Retrain, at 0.02 s vs. 39.1 s runtime.
- ETR reports on average 4583.9x less time and 4.2x less memory than retraining from scratch (abstract claim).
- On ogbn-arxiv with GCN, ETR reaches F1 66.4 vs. 67.5 for Retrain at 0.08 s vs. 152 s, and it is the only method among those listed that completes ogbn-products with GCN at 73.8 F1 (Retrain 73.6).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Use ETR instead of MEGU | GCN node unlearning · PubMed, CiteSeer, Cora, CS, Physics | F1: 88.26±0.47 -> 89.48±0.34; 76.04±0.93 -> 77.09±0.60; 82.62±1.05 -> 83.54±0.92; 92.39±0.51 -> 92.72±0.37; 95.40±0.20 -> 95 | Helped |
| Use ETR instead of the strongest baseline | GAT node unlearning · PubMed, CiteSeer, Cora, CS, Physics | F1: 88.07±0.22 -> 88.32±0.22; 76.07±0.79 -> 76.31±1.59; 82.84±0.72 -> 79.70±0.85; 92.79±0.36 -> 92.45±0.13; 95.84±0.16 -> 96 | Mixed |
| Use ETR instead of MEGU | GCN node unlearning · ogbn-arxiv, ogbn-products | F1: 65.6 -> 66.4; 71.3 -> 73.8 | Helped |
| Replace parameter editing with masking (ablation) | ETR · PubMed, CiteSeer, Cora, CS | F1: 88.91 -> 89.48; 77.00 -> 77.09; 82.51 -> 83.54; 92.44 -> 92.72 | Hurt |
| Replace parameter editing with masking (ablation) | ETR · Physics | F1: 95.71 -> 95.80 | Helped |
| Omit modeling message-propagation influence (ablation) | ETR · PubMed, Cora, Physics | F1: 89.30 -> 89.48; 82.44 -> 83.54; 95.70 -> 95.71 | Hurt |
| Omit modeling message-propagation influence (ablation) | ETR · CiteSeer, CS | F1: 76.91 -> 77.09; 92.53 -> 92.72 | Helped |
| Increase erase parameter m from 0 to the reported range (ablation) | ETR Erase strategy · PubMed, CiteSeer, Cora | F1 | Helped |
| Increase rectify step size λ from 0 (ablation) | ETR Rectify strategy · PubMed, CiteSeer, Cora | F1 | Helped |
Surprising findings
- ETR's model utility sometimes exceeds the Retrain baseline (e.g., Cora with GCN in edge unlearning and ogbn-products), attributed to the Rectify step.
Limitations
- Gradient on the remaining data is only approximated, with reported relative differences up to 0.36 on CS.
- Memory overhead advantage is limited on small datasets, where model parameters dominate.
- Hyperparameters (m, lambda, a, b) require tuning per dataset; the adaptive top-m% selection is a heuristic.
Experimental setup
Datasets: PubMed CiteSeer Cora CS Physics ogbn-arxiv ogbn-products
Benchmarks: PubMed CiteSeer Cora CS Physics ogbn-arxiv ogbn-products
Metrics: F1 score running time memory overhead parameter distance to Retrain attack-based unlearning utility
Baselines: Retrain GraphEraser-BEKM GraphEraser-BLPA GIF GUIDE-SR GUIDE-Fast GNNDelete MEGU
Builds on: SSD GIF MEGU GraphEraser
Closest papers by meaning
- Community-Centric Graph Unlearning (AAAI 2025 · similarity 0.92)
- Towards Effective and General Graph Unlearning via Mutual Evolution (AAAI 2024 · similarity 0.92)
- A Cognac Shot To Forget Bad Memories: Corrective Unlearning for Graph Neural Networks (ICML 2025 · similarity 0.90)
- Scalable and Certifiable Graph Unlearning: Overcoming the Approximation Error Barrier (ICLR 2025 · similarity 0.89)
- Efficient Unlearning for Spatio-temporal Graph (Student Abstract) (AAAI 2025 · similarity 0.87)
- Gradient Rewiring for Editable Graph Neural Network Training (NeurIPS 2024 · similarity 0.86)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
