Unlearning on Spatio-Temporal Graphs through Subgraph Virtual Edge Reconstruction
Authors: Qiming Guo, Wenbo Sun, Chen Pan, Ye Wang, Wenlu Wang
Organizations: 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 · Biogen Cambridge, Massachusetts, USA
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.
Figures & tables
Figure 1. Unlearning on an ST-graph: (a) full graph; (b) consent-revoked nodes marked; (c) record deletion leaves residual model influence; (d) retraining purges it but fragments the graph and distorts remaining features ( v1 – v3 ). Four panels showing a spatio-temporal graph before deletion, with nodes marked for removal, after raw record deletion with residual influence, and after full retraining with fragmented structure and altered node features.
Figure 2. CallosumNet system construction. The original graph (a) is transformed into multiple enhanced local subgraphs (d) through ESC, and then the GGB method adds ganglion nodes and identifies key nodes to construct the meta-graph. Nine-step pipeline diagram showing how the original graph is partitioned into enhanced subgraphs via ESC, how ganglion and key nodes form the meta-graph and global integration slot via GGB, and how sub-models are trained, frozen, integrated, and deployed.
Figure 3. CallosumNet unlearning process Six-step diagram of the unlearning process: identifying nodes to unlearn, removing them from the host subgraph, reconstructing the subgraph with virtual nodes, edges, and K-Ring, replacing the frozen sub-model, resetting and updating the global integration slot, and redeploying CallosumNet.
Dataset
Rate
SISA
STEPs
GE
GR
Ours
RWW
0%
+65.3%
+276.4%
+734.0%
+725.8%
+0.0%
10%
+57.1%
+302.4%
+708.9%
+664.8%
− 3.2%
PeMS08
0%
+19.1%
+183.9%
+107.5%
+216.1%
+1.1%
10%
+13.7%
+211.6%
+99.0%
+211.2%
+1.0%
Weather
0%
+9.6%
+55.1%
+47.5%
+66.8%
+4.2%
10%
+10.6%
+55.3%
+49.3%
+66.1%
+5.1%
Table 1. Average relative MAE degradation (%) versus the gold model, averaged over four backbones (STGCN ( Yu et al., 2018 ) , ST-GAT ( Veličković et al., 2018 ) , ST-GATV2 ( Brody et al., 2022 ) , ST-SAGE ( Hamilton et al., 2017 ) ). Negative values indicate that the method outperforms the gold model.
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.
Qiming Guo, Wenbo Sun, Ye Wang +1
Texas A&M University - Corpus Christi, Corpus Christi, TX, USA · Delft University of Technology, Netherlands · Biogen, Cambridge, MA, USA
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.
We propose a novel framework for learning time-varying graphs from spatiotemporal measurements. Given an appropriate prior on the temporal behavior of signals, our proposed method can estimate time-varying graphs from a small number of available measurements. To achieve this, we introduce three regularization terms in convex optimization problems that constrain the sparseness of temporal variations of the time-varying networks. Moreover, a computationally scalable algorithm is introduced to solve the optimization problem efficiently. The experimental results with synthetic and real datasets (point cloud, temperature, and EEG data) demonstrate that our proposed method outperforms state-of-the-art methods.
Haruki Yokota, Koki Yamada, Yuichi Tanaka +1
Graduate School of Engineering, The University of Osaka, Osaka 565-0871, Japan · Graduate School of Engineering, Tokyo University of Agriculture and Technology, Tokyo 184-8588, Japan · Department of Electrical and Computer Engineering, University of Southern California, Los Angeles, CA 90089 USA