Unlearning Graph Classifiers with Limited Data Resources
Chao Pan, Eli Chien, Olgica Milenkovic
Abstract
As the demand for user privacy grows, controlled data removal (machine unlearning) is becoming an important feature of machine learning models for data-sensitive Web applications such as social networks and recommender systems. Nevertheless, at this point it is still largely unknown how to perform efficient machine unlearning of graph neural networks (GNNs); this is especially the case when the number of training samples is small, in which case unlearning can seriously compromise the performance of the model. To address this issue, we initiate the study of unlearning the Graph Scattering Transform (GST), a mathematical framework that is efficient, provably stable under feature or graph topology perturbations, and offers graph classification performance comparable to that of GNNs. Our main contribution is the first known nonlinear approximate graph unlearning method based on GSTs. Our second contribution is a theoretical analysis of the computational complexity of the proposed unlearning mechanism, which is hard to replicate for deep neural networks. Our third contribution are extensive simulation results which show that, compared to complete retraining of GNNs after each removal request, the new GST-based approach offers, on average, a 10.38x speed-up and leads to a 2.6% increase in test accuracy during unlearning of 90 out of 100 training graphs from the IMDB dataset (10% training ratio). Our implementation is available online at https://doi.org/10.5281/zenodo.7613150 . CCS CONCEPTS • Security and privacy → Human and societal aspects of security and privacy; • Computing methodologies → Machine learning.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers16
- Differentially Private Decoupled Graph Convolutions for Multigranular Topology ProtectionEli Chien, Wei-Ning Chen, Chao Pan, Pan Li et al.NeurIPS 2023 · 33 citations
- Breaking the Trilemma of Privacy, Utility, and Efficiency via Controllable Machine UnlearningZheyuan Liu, Guangyao Dou, Eli Chien, Chunhui Zhang et al.WWW 2024 · 32 citations
- Towards Certified Unlearning for Deep Neural NetworksBinchi Zhang, Yushun Dong, Tianhao Wang, Jundong LiICML 2024 · 31 citations
- Certified Edge Unlearning for Graph Neural NetworksKun Wu, Jie Shen, Yue Ning, Ting Wang et al.KDD 2023 · 24 citations
- Verification of Machine Unlearning is FragileBinchi Zhang, Zihan Chen, Cong Shen, Jundong LiICML 2024 · 21 citations
Builds on11
- Certified Data Removal from Machine Learning ModelsChuan Guo, Tom Goldstein, Awni Y. Hannun, Laurens van der MaatenICML 2020 · 633 citations
- Remember What You Want to Forget: Algorithms for Machine UnlearningAyush Sekhari, Jayadev Acharya, Gautam Kamath, Ananda Theertha SureshNeurIPS 2021 · 516 citations
- Knowledge-aware Coupled Graph Neural Network for Social RecommendationChao Huang, Huance Xu, Yong Xu, Peng Dai et al.AAAI 2021 · 215 citations
- Scattering GCN: Overcoming Oversmoothness in Graph Convolutional NetworksYimeng Min, Frederik Wenkel, Guy WolfNeurIPS 2020 · 141 citations
- Graph UnlearningMin Chen, Zhikun Zhang, Tianhao Wang, Michael Backes et al.CCS 2022 · 103 citations
Related papers
- Efficient Model Updates for Approximate Unlearning of Graph-Structured DataEli Chien, Chao Pan, Olgica MilenkovicICLR 2023
- Is Graph Unlearning Ready for Practice? A Benchmark on Efficiency, Utility, and ForgettingSamyak Jain, Ronak Kalvani, sainyam galhotra, Sayan RanuICLR 2026
- Scalable and Certifiable Graph Unlearning: Overcoming the Approximation Error BarrierLu Yi, Zhewei WeiICLR 2025
- IDEA: A Flexible Framework of Certified Unlearning for Graph Neural NetworksYushun Dong, Binchi Zhang, Zhenyu Lei, Na Zou et al.KDD 2024 · 11 citations
- Dynamic Graph Unlearning: A General and Efficient Post-Processing Method via Gradient TransformationHe Zhang, Bang Wu, Xiangwen Yang, Xingliang Yuan et al.WWW 2025 · 16 citations
