Search papers, labs, and topics across Lattice.
This paper introduces StaticSegFormer, a novel static structured pruning method specifically designed for attention layers in semantic segmentation networks. By optimizing the SegFormer architecture, the method achieves a significant increase in frame rate鈥攗p to 34% on the Cityscapes dataset鈥攚hile maintaining the same mean Intersection over Union (mIoU) performance and reducing computational complexity. This advancement highlights the potential of static pruning techniques to outperform dynamic methods in both efficiency and speed without sacrificing accuracy.
StaticSegFormer boosts segmentation frame rates by 34% without compromising accuracy, challenging the dominance of dynamic pruning methods.
Structured pruning enhances the efficiency of deep neural networks (DNNs) by eliminating groups of parameters during inference. Previous methods mostly reduce computational complexity (FLOPs), while semantic segmentation performance (mIoU) slightly drops. Accordingly, recent dynamic structured pruning methods aim at reducing the performance drop, while lowering the FLOPs even more. However, on the ADE20K and Cityscapes benchmarks, our study reveals that on a GPU platform such dynamic methods exhibit a surprisingly low frame rate far below a simple static approach, while having comparable results in mIoU and FLOPs. To address this issue, we propose a static structured pruning method for attention layers, that achieves both, a lower FLOPs and a high frame rate [fps] of the SegFormer network, the latter increased by up to 34% relative on the Cityscapes dataset, while having no mIoU performance drop at all. Our so-called StaticSegFormer method is strongest for small encoders and large images.
