Falcon: lightweight and accurate convolution based on depthwise separable convolution

Jun-Gi Jang,Chun Quan,Hyun Dong Lee,U. Kang
DOI: https://doi.org/10.1007/s10115-022-01818-x
IF: 2.7
2023-01-19
Knowledge and Information Systems
Abstract:How can we efficiently compress convolutional neural network (CNN) using depthwise separable convolution, while retaining their accuracy on classification tasks? Depthwise separable convolution, which replaces a standard convolution with a depthwise convolution and a pointwise convolution, has been used for building lightweight architectures. However, previous works based on depthwise separable convolution are limited when compressing a trained CNN model since (1) they are mostly heuristic approaches without a precise understanding of their relations to standard convolution, and (2) their accuracies do not match that of the standard convolution. In this paper, we propose Falcon , an accurate and lightweight method to compress CNN based on depthwise separable convolution. Falcon uses generalized elementwise product (GEP), our proposed mathematical formulation to approximate the standard convolution kernel, to interpret existing convolution methods based on depthwise separable convolution. By exploiting the knowledge of a trained standard model and carefully determining the order of depthwise separable convolution via GEP, Falcon achieves sufficient accuracy close to that of the trained standard model. Furthermore, this interpretation leads to developing a generalized version rank- k Falcon which performs k independent Falcon operations and sums up the result. Experiments show that Falcon (1) provides higher accuracy than existing methods based on depthwise separable convolution and tensor decomposition and (2) reduces the number of parameters and FLOPs of standard convolution by up to a factor of 8 while ensuring similar accuracy. We also demonstrate that rank- k Falcon further improves the accuracy while sacrificing a bit of compression and computation reduction rates.
computer science, information systems, artificial intelligence
What problem does this paper attempt to address?