Residential College | false |
Status | 已發表Published |
AOAM: Automatic Optimization of Adjacency Matrix for Graph Convolutional Network | |
Yuhang Zhang1,2; Hongshuai Ren2; Jiexia Ye2; Xitong Gao2; Yang Wang2; Kejiang Ye2; Cheng-Zhong Xu3 | |
2021-05 | |
Conference Name | 25th International Conference on Pattern Recognition, ICPR 2020 |
Source Publication | Proceedings - International Conference on Pattern Recognition |
Pages | 5130 - 5136 |
Conference Date | Jan 10-15, 2021 |
Conference Place | Milan |
Country | Italy |
Publication Place | IEEE COMPUTER SOC, 10662 LOS VAQUEROS CIRCLE, PO BOX 3014, LOS ALAMITOS, CA 90720-1264 USA |
Publisher | IEEE |
Abstract | Graph Convolutional Network (GCN) is adopted to tackle the problem of convolution operation in non-Euclidean space. Previous works on GCN have made some progress, however, one of their limitations is that the design of Adjacency Matrix (AM) as GCN input requires domain knowledge and such process is cumbersome, tedious and error-prone. In addition, entries of a fixed Adjacency Matrix are generally designed as binary values (i.e., ones and zeros) which can not reflect the real relationship between nodes. Meanwhile, many applications require a weighted and dynamic Adjacency Matrix instead of an unweighted and fixed AM, and there are few works focusing on designing a more flexible Adjacency Matrix. To that end, we propose an end-to-end algorithm to improve the GCN performance by focusing on the Adjacency Matrix. We first provide a calculation method called node information entropy to update the matrix. Then, we perform the search strategy in a continuous space and introduce the Deep Deterministic Policy Gradient (DDPG) method to overcome the drawback of the discrete space search. Finally, we integrate the GCN and reinforcement learning into an end-to-end framework. Our method can automatically define the Adjacency Matrix without prior knowledge. At the same time, the proposed approach can deal with any size of the matrix and provide a better AM for network. Four popular datasets are selected to evaluate the capability of our algorithm. The method in this paper achieves the state-of-the-art performance on Cora and Pubmed datasets, with the accuracy of 84.6% and 81.6% respectively. |
Keyword | Graph Convolutional Network Adjacency Matrix End-to-end Node Information Entropy |
DOI | 10.1109/ICPR48806.2021.9412046 |
URL | View the original |
Indexed By | CPCI-S |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering ; Imaging Science & Photographic Technology |
WOS Subject | Computer Science, Artificial Intelligence ; Engineering, Electrical & Electronic ; Imaging Science & Photographic Technology |
WOS ID | WOS:000678409205034 |
Scopus ID | 2-s2.0-85110495370 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) Faculty of Science and Technology |
Corresponding Author | Kejiang Ye |
Affiliation | 1.University of Chinese Academy of Sciences 2.Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences 3.State Key Lab of IoTSC, University of Macau |
Recommended Citation GB/T 7714 | Yuhang Zhang,Hongshuai Ren,Jiexia Ye,et al. AOAM: Automatic Optimization of Adjacency Matrix for Graph Convolutional Network[C], IEEE COMPUTER SOC, 10662 LOS VAQUEROS CIRCLE, PO BOX 3014, LOS ALAMITOS, CA 90720-1264 USA:IEEE, 2021, 5130 - 5136. |
APA | Yuhang Zhang., Hongshuai Ren., Jiexia Ye., Xitong Gao., Yang Wang., Kejiang Ye., & Cheng-Zhong Xu (2021). AOAM: Automatic Optimization of Adjacency Matrix for Graph Convolutional Network. Proceedings - International Conference on Pattern Recognition, 5130 - 5136. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment