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.
引用
收藏
页码:7918 / 7930
页数:13
相关论文
共 75 条
[1]  
Alon Uri, 2021, INT C LEARN REPR
[2]   A graph-based toy model of chemistry [J].
Benkö, G ;
Flamm, C ;
Stadler, PF .
JOURNAL OF CHEMICAL INFORMATION AND COMPUTER SCIENCES, 2003, 43 (04) :1085-1093
[3]  
Bruna J., 2014, INT C LEARN REPR ICL
[4]  
Cai Rongshen, 2021, P PAC RIM INT C ART, P500
[5]   Rethinking Graph Neural Architecture Search from Message-passing [J].
Cai, Shaofei ;
Li, Liang ;
Deng, Jincan ;
Zhang, Beichen ;
Zha, Zheng-Jun ;
Su, Li ;
Huang, Qingming .
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 [J].
Chen, Jie ;
Chen, Shouzhen ;
Bai, Mingyuan ;
Pu, Jian ;
Zhang, Junping ;
Gao, Junbin .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) :9859-9873
[7]   AutoGSR: Neural Architecture Search for Graph-based Session Recommendation [J].
Chen, Jingfan ;
Zhu, Guanghui ;
Hou, Haojun ;
Yuan, Chunfeng ;
Huang, Yihua .
PROCEEDINGS OF THE 45TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '22), 2022, :1694-1704
[8]  
Chen M, 2020, PR MACH LEARN RES, V119
[9]  
Chen M, 2020, ADV NEUR IN, V33
[10]   How Do the Open Source Communities Address Usability and UX Issues? An Exploratory Study [J].
Cheng, Jinghui ;
Guo, Jin L. C. .
CHI 2018: EXTENDED ABSTRACTS OF THE 2018 CHI CONFERENCE ON HUMAN FACTORS IN COMPUTING SYSTEMS, 2018,