Efficient Semantic Segmentation using Gradual Grouping

Best Runner-Up Award, Oral Presentation
IEEE Embedded Vision Workshop, CVPR'18
May, 2018

Bibtex

@InProceedings{Vallurupalli_2018_CVPR_Workshops, author = {Vallurupalli, Nikitha and Annamaneni, Sriharsha and Varma, Girish and Jawahar, C.V. and Mathew, Manu and Nagori, Soyeb}, title = {Efficient Semantic Segmentation Using Gradual Grouping}, booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops}, month = {June}, year = {2018} }

Abstract

Deep CNNs for semantic segmentation have high memory and run time requirements. Various approaches have been proposed to make CNNs efficient like grouped, shuffled, depth-wise separable convolutions. We study the effectiveness of these techniques on a real-time semantic segmentation architecture like ERFNet for improving runtime by over 5X. We apply these techniques to CNN layers partially or fully and evaluate the testing accuracies on Cityscapes dataset. We obtain accuracy vs parameters/FLOPs trade offs, giving accuracy scores for models that can run under specified runtime budgets. We further propose a novel training procedure which starts out with a dense convolution but gradually evolves towards a grouped convolution. We show that our proposed training method and efficient architecture design can improve accuracies by over 8% with depthwise separable convolutions applied on the encoder of ERFNet and attaching a light weight decoder. This results in a model which has a 5X improvement in FLOPs while only suffering a 4% degradation in accuracy with respect to ERFNet.