Automatic Design of Deep Graph Neural Networks With Decoupled Mode

被引:1
作者
Tao, Qian [1 ,2 ,3 ]
Cai, Rongshen [4 ]
Lin, Zicheng [1 ,2 ]
Tang, Yufei [5 ]
机构
[1] South China Univ Technol, Sch Software, Software Serv Engn, Guangzhou 510006, Peoples R China
[2] South China Univ Technol, Sch Software, Cloud Comp Lab, Guangzhou 510006, Peoples R China
[3] Pazhou Lab, Guangzhou 510335, Peoples R China
[4] Ant Grp Inc, Int Business Grp, Shenzhen 518000, Guangdong, Peoples R China
[5] Florida Atlantic Univ, Dept Elect Engn & Comp Sci, Boca Raton, FL 33431 USA
基金
中国国家自然科学基金;
关键词
Computer architecture; Training; Task analysis; Computational modeling; Graph neural networks; Data models; Accuracy; Decoupled mode; graph mining; graph neural networks (GNNs); neural architecture search (NAS); swarm optimization; ARCHITECTURES;
D O I
10.1109/TNNLS.2024.3438609
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph neural networks (GNNs), a class of deep learning models designed for performing information interaction on non-Euclidean graph data, have been successfully applied to node classification tasks in various applications such as citation networks, recommender systems, and natural language processing. Graph node classification is an important research field for node-level tasks in graph data mining. Recently, due to the limitations of shallow GNNs, many researchers have focused on designing deep graph learning models. Previous GNN architecture search works only solve shallow networks (e.g., less than four layers). It is challenging and nonefficient to manually design deep GNNs for challenges like over-smoothing and information squeezing, which greatly limits their capabilities on large-scale graph data. In this article, we propose a novel neural architecture search (NAS) method for designing deep GNNs automatically and further exploit the application potential on various node classification tasks. Our innovations lie in two aspects, where we first redesign the deep GNNs search space for architecture search with a decoupled mode based on propagation and transformation processes, and we then formulate and solve the problem as a multiobjective optimization to balance accuracy and computational efficiency. Experiments on benchmark graph datasets show that our method performs very well on various node classification tasks, and exploiting large-scale graph datasets further validates that our proposed method is scalable.
引用
收藏
页数:13
相关论文
共 75 条
  • [1] Alon U., 2021, P INT C LEARN REPR, P1
  • [2] [Anonymous], 2000, Journal of Social Structure
  • [3] A graph-based toy model of chemistry
    Benkö, G
    Flamm, C
    Stadler, PF
    [J]. JOURNAL OF CHEMICAL INFORMATION AND COMPUTER SCIENCES, 2003, 43 (04): : 1085 - 1093
  • [4] Bruna J, 2014, Arxiv, DOI [arXiv:1312.6203, 10.48550/arXiv.1312.6203]
  • [5] Rethinking Graph Neural Architecture Search from Message-passing
    Cai, Shaofei
    Li, Liang
    Deng, Jincan
    Zhang, Beichen
    Zha, Zheng-Jun
    Su, Li
    Huang, Qingming
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 6653 - 6662
  • [6] Graph Decoupling Attention Markov Networks for Semisupervised Graph Node Classification
    Chen, Jie
    Chen, Shouzhen
    Bai, Mingyuan
    Pu, Jian
    Zhang, Junping
    Gao, Junbin
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) : 9859 - 9873
  • [7] AutoGSR: Neural Architecture Search for Graph-based Session Recommendation
    Chen, Jingfan
    Zhu, Guanghui
    Hou, Haojun
    Yuan, Chunfeng
    Huang, Yihua
    [J]. PROCEEDINGS OF THE 45TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '22), 2022, : 1694 - 1704
  • [8] Chen M, 2020, ADV NEUR IN, V33
  • [9] Chen M, 2020, PR MACH LEARN RES, V119
  • [10] How Do the Open Source Communities Address Usability and UX Issues? An Exploratory Study
    Cheng, Jinghui
    Guo, Jin L. C.
    [J]. CHI 2018: EXTENDED ABSTRACTS OF THE 2018 CHI CONFERENCE ON HUMAN FACTORS IN COMPUTING SYSTEMS, 2018,