TY - JOUR
T1 - Binary Graph Convolutional Network With Capacity Exploration
AU - Wang, Junfu
AU - Guo, Yuanfang
AU - Yang, Liang
AU - Wang, Yunhong
N1 - Publisher Copyright:
© 1979-2012 IEEE.
PY - 2024/5/1
Y1 - 2024/5/1
N2 - The current success of Graph Neural Networks (GNNs) usually relies on loading the entire attributed graph for processing, which may not be satisfied with limited memory resources, especially when the attributed graph is large. This paper pioneers to propose a Binary Graph Convolutional Network (Bi- GCN), which binarizes both the network parameters and input node attributes and exploits binary operations instead of floatingpoint matrix multiplications for network compression and acceleration. Meanwhile, we also propose a new gradient approximation based back-propagation method to properly train our Bi-GCN. According to the theoretical analysis, our Bi-GCN can reduce the memory consumption by an average of ∼31x for both the network parameters and input data, and accelerate the inference speed by an average of ∼51x, on three citation networks, i.e., Cora, PubMed, and CiteSeer. Besides, we introduce a general approach to generalize our binarization method to other variants of GNNs, and achieve similar efficiencies. Although the proposed Bi-GCN and Bi-GNNs are simple yet efficient, these compressed networks may also possess a potential capacity problem, i.e., they may not have enough storage capacity to learn adequate representations for specific tasks. To tackle this capacity problem, an Entropy Cover Hypothesis is proposed to predict the lower bound of the width of Bi-GNN hidden layers. Extensive experiments have demonstrated that our Bi-GCN and Bi-GNNs can give comparable performances to the corresponding full-precision baselines on seven node classification datasets and verified the effectiveness of our Entropy Cover Hypothesis for solving the capacity problem.
AB - The current success of Graph Neural Networks (GNNs) usually relies on loading the entire attributed graph for processing, which may not be satisfied with limited memory resources, especially when the attributed graph is large. This paper pioneers to propose a Binary Graph Convolutional Network (Bi- GCN), which binarizes both the network parameters and input node attributes and exploits binary operations instead of floatingpoint matrix multiplications for network compression and acceleration. Meanwhile, we also propose a new gradient approximation based back-propagation method to properly train our Bi-GCN. According to the theoretical analysis, our Bi-GCN can reduce the memory consumption by an average of ∼31x for both the network parameters and input data, and accelerate the inference speed by an average of ∼51x, on three citation networks, i.e., Cora, PubMed, and CiteSeer. Besides, we introduce a general approach to generalize our binarization method to other variants of GNNs, and achieve similar efficiencies. Although the proposed Bi-GCN and Bi-GNNs are simple yet efficient, these compressed networks may also possess a potential capacity problem, i.e., they may not have enough storage capacity to learn adequate representations for specific tasks. To tackle this capacity problem, an Entropy Cover Hypothesis is proposed to predict the lower bound of the width of Bi-GNN hidden layers. Extensive experiments have demonstrated that our Bi-GCN and Bi-GNNs can give comparable performances to the corresponding full-precision baselines on seven node classification datasets and verified the effectiveness of our Entropy Cover Hypothesis for solving the capacity problem.
KW - Binary graph neural networks
KW - graph representation learning
KW - information storage capacity
UR - https://www.scopus.com/pages/publications/85179796765
U2 - 10.1109/TPAMI.2023.3342224
DO - 10.1109/TPAMI.2023.3342224
M3 - 文章
C2 - 38090833
AN - SCOPUS:85179796765
SN - 0162-8828
VL - 46
SP - 3031
EP - 3046
JO - IEEE Transactions on Pattern Analysis and Machine Intelligence
JF - IEEE Transactions on Pattern Analysis and Machine Intelligence
IS - 5
M1 - 10356827
ER -