Disentangled Continual Graph Neural Architecture Search with Invariant Modular Supernet
Zeyang Zhang, Xin Wang, Yijian Qin, Hong Chen, Ziwei Zhang, Xu Chu, Wenwu Zhu
Abstract
The existing graph neural architecture search (GNAS) methods assume that the graph tasks are static during the search process, ignoring the ubiquitous scenarios where sequential graph tasks come in a continual fashion. Moreover, existing GNAS works resort to entangled graph factors during the architecture search process, resulting in the catastrophic forgetting problems. In this paper, we study the problem of continual GNAS that is expected to continually search the architecture to learn new graph tasks without forgetting the past, which remains largely unexplored in the literature. However, this problem poses the challenge of architecture conflicts, i.e., the optimal architecture for the new graph task may have performance deterioration and thus sub-optimal for past tasks. To address the challenge, we propose a novel Disentangled Continual Graph Neural Architecture Search with Invariant Modular Supernet (GASIM) method, which is able to continually search the optimal architectures without forgetting past knowledge. Specifically, we first design a modular graph architecture super-network incorporating multiple modules to enable searching architecture with factor expertise. Second, we propose a factor-based task-module router that discovers the latent graph factors and routes the incoming task to the best suitable architecture module to alleviate the forgetting problem induced by architecture conflicts. Finally, we propose an invariant architecture search mechanism to capture the shared knowledge among tasks. Extensive experiments demonstrate that our method achieves state-of-the-art performance against baselines in continual graph neural architecture search.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7b4bbcdf-9fde-497c-ac75-3dbc6b841b33Cited by top-tier papers8
- Behavior Importance-Aware Graph Neural Architecture Search for Cross-Domain RecommendationChendi Ge, Xin Wang, Ziwei Zhang, Yijian Qin et al.AAAI 2025 · 3 citations
- Improving Compositional Generalization in Cross-Embodiment Learning via Mixture of Disentangled PrototypesRen Wang, Xin Wang, Tongtong Feng, Xinyue Gong et al.ACM MM 2025 · 2 citations
- DyNAS-DDI: Dynamic Pairwise Architecture Search for Generalizable Drug-Drug Interaction LLMLinxin Xiao, Xin Wang, Zeyang Zhang, Yang Yao et al.ACM MM 2025 · 1 citation
- Dynamic Mixture of Curriculum LoRA Experts for Continual Multimodal Instruction TuningChendi Ge, Xin Wang, Zeyang Zhang, Hong Chen et al.ICML 2025
- Continual-GraphLLM: Dynamic Graph Large Language Model with Invariance Regularized Adaptive Multi-Scale ExpertsTianhang Wan, Xin Wang, Haibo Chen, Longtao Huang et al.KDD 2026
Builds on34
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Out-of-Distribution Generalization via Risk Extrapolation (REx)David Krueger, Ethan Caballero, Jörn-Henrik Jacobsen, Amy Zhang et al.ICML 2021 · 1,163 citations
- GraphMAE: Self-Supervised Masked Graph AutoencodersZhenyu Hou, Xiao Liu, Yukuo Cen, Yuxiao Dong et al.KDD 2022 · 533 citations
- Discovering Invariant Rationales for Graph Neural NetworksYingxin Wu, Xiang Wang, An Zhang, Xiangnan He et al.ICLR 2022 · 313 citations
- Improving Out-of-Distribution Robustness via Selective AugmentationHuaxiu Yao, Yu Wang, Sai Li, Linjun Zhang et al.ICML 2022 · 275 citations
Related papers
- Unsupervised Graph Neural Architecture Search with Disentangled Self-SupervisionZeyang Zhang, Xin Wang, Ziwei Zhang, Guangyao Shen et al.NeurIPS 2023 · 22 citations
- Causal-aware Graph Neural Architecture Search under Distribution ShiftsPeiwen Li, Xin Wang, Zeyang Zhang, Ziwei Zhang et al.KDD 2025
- AutoGFM: Automated Graph Foundation Model with Adaptive Architecture CustomizationHaibo Chen, Xin Wang, Zeyang Zhang, Haoyang Li et al.ICML 2025
- Overcoming Catastrophic Forgetting in Graph Neural NetworksHuihui Liu, Yiding Yang, Xinchao WangAAAI 2021 · 166 citations
- Multi-task Graph Neural Architecture Search with Task-aware Collaboration and CurriculumYijian Qin, Xin Wang, Ziwei Zhang, Hong Chen et al.NeurIPS 2023 · 27 citations
