Spatial Entropy based Partitioning for Spatiotemporal Graph Unlearning
Authors: Qiming Guo, Wenbo Sun, Ye Wang, Wenlu Wang
Organizations: Texas A&M University - Corpus Christi, Corpus Christi, TX, USA · Delft University of Technology, Netherlands · Biogen, Cambridge, MA, USA
Spatiotemporal graphs underpin applications such as traffic forecasting, weather forecasting, and healthcare monitoring. Privacy regulations such as the GDPR and the CCPA require the complete removal of unauthorized data from trained models, but achieving this on a spatiotemporal graph is difficult: because information propagates globally through both spatial and temporal message passing, fully erasing a node's influence forces costly full-graph retraining. ST-graph unlearning requires both exactness and efficiency. We propose IsleNet, which uses spatial-entropy-guided partitioning to create balanced, locally coherent subgraphs and reconnects them with lightweight virtual edges. Upon an unlearning request, only the affected subgraph encoder and virtual-edge layer are retrained, ensuring exact removal with low cost. Experiments on four real-world benchmarks show that IsleNet attains up to 94% of full-graph accuracy while reducing unlearning time by up to an order of magnitude. Our code is publicly available at https://github.com/wenlu-lab/STGraphUnlearning.
Figures & tables
Figure 1.1: The challenge of unlearning on spatiotemporal graphs. (a) In a complete ST-graph, walks propagate node influence globally (e.g., v4 reaches v5 via Walk C; v3 lies on Walks A and B). (b) Target nodes are marked for unlearning ( UN ), blocking every walk through them. (c) After removal, Walk A no longer reaches v1 or v3 , Walk B no longer reaches v2 , and uninvolved v4 loses all outgoing paths, a structural cascade. (d) Retraining the fragmented graph alters the remaining node features ( v1,v2,v3,v5 ) and fades the walk halos that encode global context, while costing as much as full training. ST-graph unlearning must therefore be exact (deleted influence fully removed) and efficient (fast without sacrificing post-unlearning accuracy).
Figure 3.1: Spatial entropy in IsleNet. Top: Spatial entropy Hs (Eq. 3.2 ) quantifies how a subgraph’s nodes are distributed across space. (a) Low: nodes concentrate in narrow regions. (b) High: nodes spread uniformly. Bottom: IsleNet uses spatial entropy to verify partition quality: blue and red halos mark the two subgraphs, and the dashed line separates them. (c) A degenerate partition: one subgraph shrinks to a small corner, so IsleNet detects the imbalance and redistributes nodes. (d) A balanced partition: both subgraphs cover comparable spatial extents.
Figure 3.2: Framework of IsleNet. (a) The original graph. (b) Split into smaller subgraphs, each with its own encoder. (c) Virtual edges (green) reconnect subgraphs; on a deletion request, edges touching the target are marked for removal (RM). (d) Only the affected subgraph is retrained and virtual edges are rebuilt; other subgraphs stay untouched. How it works: Spatial entropy guides the split so no subgraph degenerates. Each encoder trains in isolation, and virtual edges stitch subgraphs back together. To unlearn, IsleNet retrains only the affected encoder and rebuilds the virtual edges from scratch. Untouched subgraphs keep their weights, making unlearning fast and exact.
Dataset
Model
Scratch-100%
Baseline Methods
IsleNet
SISA
STEPS
GraphEraser
GraphRevoker
RWW
STGCN
0.020 ± 0.001
0.035 ± 0.007
0.082 ± 0.003
0.179 ± 0.060
0.177 ± 0.000
0.024 ± 0.001
ST-GAT
0.022 ± 0.002
0.035 ± 0.013
0.075 ± 0.004
0.179 ± 0.059
0.177 ± 0.001
0.025 ± 0.001
ST-GATV2
0.022 ± 0.002
0.036 ± 0.008
0.085 ± 0.008
0.179 ± 0.059
0.177 ± 0.001
0.024 ± 0.001
ST-SAGE
0.022 ± 0.003
0.036 ± 0.010
0.081 ± 0.008
0.179 ± 0.059
0.178 ± 0.000
0.025 ± 0.001
PeMS08
STGCN
28.751 ± 0.117
34.271 ± 0.527
82.404 ± 9.043
58.994 ± 1.663
88.685 ± 5.865
30.532 ± 0.045
Table 4.1: Prediction Performance of Different Methods Before Unlearning (0% Unlearning).
Dataset
Model
Scratch-90%
Baseline Methods
IsleNet
SISA
STEPS
GraphEraser
GraphRevoker
RWW
STGCN
0.023 ± 0.001
0.036 ± 0.007
0.095 ± 0.022
0.188 ± 0.067
0.178 ± 0.006
0.025 ± 0.002
ST-GAT
0.023 ± 0.001
0.038 ± 0.006
0.097 ± 0.025
0.188 ± 0.080
0.178 ± 0.005
0.026 ± 0.003
ST-GATV2
0.024 ± 0.002
0.035 ± 0.003
0.090 ± 0.023
0.188 ± 0.081
0.177 ± 0.005
0.026 ± 0.002
ST-SAGE
0.023 ± 0.002
0.037 ± 0.011
0.092 ± 0.023
0.188 ± 0.085
0.178 ± 0.005
0.026 ± 0.003
PeMS08
STGCN
30.810 ± 0.147
34.332 ± 0.515
99.807 ± 12.190
61.315 ± 4.643
97.568 ± 3.789
31.564 ± 0.121
Table 4.2: Prediction Performance of Different Methods After Unlearning (10% Unlearning).
Method
Bounded Sim (%)
Ratio Score (%)
SISA
86.22
89.51
STEPS
50.81
67.15
GraphEraser
58.26
71.35
GraphRevoker
52.39
65.16
IsleNet
95.35
95.65
Table 4.3: Average similarity to Scratch models (higher is better). Computed across 4 datasets, 4 backbones, and 2 unlearning rates (0% and 10%).
Spatio-temporal graphs are widely used in modeling complex dynamic processes such as temporal forecasting, molecular dynamics, and healthcare monitoring. Recently, stringent privacy regulations such as GDPR and CCPA have introduced significant new challenges for existing spatio-temporal graph models, requiring complete unlearning of unauthorized data. Since each node in a spatio-temporal graph diffuses information globally across both spatial and temporal dimensions, existing unlearning methods primarily designed for static graphs and localized data removal cannot efficiently erase a single node without incurring costs nearly equivalent to full model retraining. To address this, we propose CallosumNet, a spatio-temporal graph unlearning framework biologically inspired by the corpus callosum structure. CallosumNet makes two key technical contributions: (1) it reconstructs subgraphs using biologically-inspired virtual edges; and (2) it restores interlinked spatio-temporal dependencies among subgraphs via a lightweight meta-graph integration layer. Empirical results on four diverse real-world datasets show that CallosumNet achieves complete unlearning while maintaining accuracy very close to the gold model. The code is publicly available at https://github.com/wenlu-lab/STGraphUnlearning.
Qiming Guo, Wenbo Sun, Chen Pan +2
Texas A&M University - Corpus Christi Corpus Christi, Texas, USA · Delft University of Technology Delft, Netherlands · University of Texas at San Antonio San Antonio, Texas, USA +1
Graph unlearning removes the influence of designated training data from a trained graph model without retraining from scratch. However, existing methods suffer a sharp drop in model utility under large deletion ratios (mass deletion), a phenomenon we refer to as catastrophic unlearning. We find that a key cause is the uniform treatment of all deleted samples, which is particularly damaging in graph learning: structural dependencies cause different nodes to play vastly different roles in the learned model, yet existing methods apply the same forgetting operation to the entire forget set. Based on this insight, we propose CUNO, a curriculum-based graph unlearning framework that removes the forget set progressively, ordering samples by their estimated unlearning difficulty across multiple stages. CUNO further employs a distribution-level negative preference optimization (NPO) objective at each curriculum stage that steers the model away from its original behavior on the current forget subset while preserving retained performance. Our theoretical analysis shows that the curriculum design is most beneficial when the forget set spans a wide range of unlearning difficulty, a condition naturally satisfied under mass deletion. Comprehensive experiments confirm that CUNO consistently mitigates catastrophic unlearning: at 20% deletion, it retains 74% of the original utility compared to 26-53% for existing methods, and maintains more than half the original utility even at 50% deletion. Our code is publicly available at https://anonymous.4open.science/r/cuno-D4FF.
Graph neural networks (GNNs) are widely used for learning from graph-structured data in domains such as social networks, recommender systems, and financial platforms. To comply with privacy regulations like the GDPR, CCPA, and PIPEDA, approximate graph unlearning, which aims to remove the influence of specific data points from trained models without full retraining, has become an increasingly important component of trustworthy graph learning. However, approximate unlearning often incurs subtle performance degradation, which may incur negative and unintended side effects. In this work, we show that such degradations can be amplified into adversarial attacks. We introduce the notion of unlearning corruption attacks, where an adversary injects carefully chosen nodes into the training graph and later requests their deletion. Because deletion requests are legally mandated and cannot be denied, this attack surface is both unavoidable and stealthy: the model performs normally during training, but accuracy collapses only after unlearning is applied. Technically, we formulate this attack as a bi-level optimization problem: to overcome the challenges of black-box unlearning and label scarcity, we approximate the unlearning process via gradient-based updates and employ a surrogate model to generate pseudo-labels for the optimization. Extensive experiments across benchmarks and unlearning algorithms demonstrate that small, carefully designed unlearning requests can induce significant accuracy degradation, raising urgent concerns about the robustness of GNN unlearning under real-world regulatory demands. The source code will be released upon paper acceptance.
Jiahao Zhang, Yilong Wang, Suhang Wang
The Pennsylvania State University University Park, PA, USA