Haar Wavelet Feature Compression for Quantized Graph Convolutional Networks

IEEE Trans Neural Netw Learn Syst. 2024 Apr;35(4):4542-4553. doi: 10.1109/TNNLS.2023.3285874. Epub 2024 Apr 4.

Abstract

Graph convolutional networks (GCNs) are widely used in a variety of applications and can be seen as an unstructured version of standard convolutional neural networks (CNNs). As in CNNs, the computational cost of GCNs for large input graphs (such as large point clouds or meshes) can be high and inhibit the use of these networks, especially in environments with low computational resources. To ease these costs, quantization can be applied to GCNs. However, aggressive quantization of the feature maps can lead to a significant degradation in performance. On a different note, the Haar wavelet transforms are known to be one of the most effective and efficient approaches to compress signals. Therefore, instead of applying aggressive quantization to feature maps, we propose to use Haar wavelet compression and light quantization to reduce the computations involved with the network. We demonstrate that this approach surpasses aggressive feature quantization by a significant margin, for a variety of problems ranging from node classification to point cloud classification and both part and semantic segmentation.