Golden Cudgel Network for Real-Time Semantic Segmentation
Guoyu Yang, Yuan Wang, Daming Shi, Yanzhong Wang
TL;DR
GCNet tackles the real-time semantic segmentation bottleneck by using training-time vertical multi-convolutions and horizontal multi-paths that are reparameterized into a single $3 \times 3$ convolution for inference. It introduces the Golden Cudgel Block (GCBlock), which enables self-enlargement during training and self-contraction during inference, effectively acting as its own teacher without external models. Empirical results on Cityscapes, CamVid, and Pascal VOC 2012 show GCNet achieving a favorable balance of high mIoU and fast FPS, with GCNet-S reaching 77.3% mIoU at 193.3 FPS and GCNet-L achieving 79.6% mIoU, while maintaining strong zero-shot and training efficiency relative to comparable real-time methods. The work demonstrates that reparameterizable, multi-path training can realize the benefits of complex blocks for learning without sacrificing inference speed, offering practical impact for deployment in latency-critical applications.
Abstract
Recent real-time semantic segmentation models, whether single-branch or multi-branch, achieve good performance and speed. However, their speed is limited by multi-path blocks, and some depend on high-performance teacher models for training. To overcome these issues, we propose Golden Cudgel Network (GCNet). Specifically, GCNet uses vertical multi-convolutions and horizontal multi-paths for training, which are reparameterized into a single convolution for inference, optimizing both performance and speed. This design allows GCNet to self-enlarge during training and self-contract during inference, effectively becoming a "teacher model" without needing external ones. Experimental results show that GCNet outperforms existing state-of-the-art models in terms of performance and speed on the Cityscapes, CamVid, and Pascal VOC 2012 datasets. The code is available at https://github.com/gyyang23/GCNet.
