Timezone: »

Structured Convolutions for Efficient Neural Network Design
Yash Bhalgat · Yizhe Zhang · Jamie Menjay Lin · Fatih Porikli

Wed Dec 09 09:00 PM -- 11:00 PM (PST) @ Poster Session 4 #1161

In this work, we tackle model efficiency by exploiting redundancy in the implicit structure of the building blocks of convolutional neural networks. We start our analysis by introducing a general definition of Composite Kernel structures that enable the execution of convolution operations in the form of efficient, scaled, sum-pooling components. As its special case, we propose Structured Convolutions and show that these allow decomposition of the convolution operation into a sum-pooling operation followed by a convolution with significantly lower complexity and fewer weights. We show how this decomposition can be applied to 2D and 3D kernels as well as the fully-connected layers. Furthermore, we present a Structural Regularization loss that promotes neural network layers to leverage on this desired structure in a way that, after training, they can be decomposed with negligible performance loss. By applying our method to a wide range of CNN architectures, we demonstrate 'structured' versions of the ResNets that are up to 2x smaller and a new Structured-MobileNetV2 that is more efficient while staying within an accuracy loss of 1% on ImageNet and CIFAR-10 datasets. We also show similar structured versions of EfficientNet on ImageNet and HRNet architecture for semantic segmentation on the Cityscapes dataset. Our method performs equally well or superior in terms of the complexity reduction in comparison to the existing tensor decomposition and channel pruning methods.

Author Information

Yash Bhalgat (Qualcomm AI Research)

I am a research scientist at Qualcomm AI Research. I do algorithm and system design to develop efficient deep networks for computer vision use-cases. I have developed pipelines and published papers on low-bit quantization [LSQ+, QKD], structured [StructConv] and unstructured pruning [LTP] of deep networks. The usecases I have worked on include image and video classification / segmentation, pose estimation, gaze estimation, etc. My current research focuses on developing compute-adaptive deep networks.

Yizhe Zhang (Qualcomm AI Research)
Jamie Menjay Lin (Qualcomm AI Research)
Fatih Porikli (Qualcomm CR&D)