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 条
  • [11] Handling multiple objectives with particle swarm optimization
    Coello, CAC
    Pulido, GT
    Lechuga, MS
    [J]. IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2004, 8 (03) : 256 - 279
  • [12] Defferrard M, 2016, ADV NEUR IN, V29
  • [13] Multiobjective bilevel optimization
    Eichfelder, Gabriele
    [J]. MATHEMATICAL PROGRAMMING, 2010, 123 (02) : 419 - 449
  • [14] Feng Guosheng, 2021, arXiv
  • [15] Particle swarm optimization of deep neural networks architectures for image classification
    Fernandes Junior, Francisco Erivaldo
    Yen, Gary G.
    [J]. SWARM AND EVOLUTIONARY COMPUTATION, 2019, 49 : 62 - 74
  • [16] Frasca F, 2020, Arxiv, DOI arXiv:2004.11198
  • [17] Large-Scale Learnable Graph Convolutional Networks
    Gao, Hongyang
    Wang, Zhengyang
    Ji, Shuiwang
    [J]. KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, : 1416 - 1424
  • [18] Gao Y, 2020, PROCEEDINGS OF THE TWENTY-NINTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1403
  • [19] Hamilton WL, 2017, ADV NEUR IN, V30
  • [20] Henaff M, 2015, Arxiv, DOI arXiv:1506.05163