本論文では、正確で高速なセマンティック セグメンテーションのための新しいアーキテクチャであるShelfNetを紹介する。単一のエンコーダ-デコーダ構造とは異なり、ShelfNetは、各空間レベルでスキップ接続を持つ複数のエンコーダ-デコーダ分岐ペアを持ち、複数の列を持つ棚のような構造となっている。棚状の構造は、複数の深いパスと浅いパスのアンサンブルと見ることができ、精度を向上させることができる。チャネル数を減らすことで計算負荷を大幅に軽減すると同時に、このユニークな構造で高精度を実現している。また、性能を犠牲にすることなくパラメータ数を削減する残差ブロックの共有重み戦略を提案している。PSPNetなどの一般的な非リアルタイム手法と比較して、PASCAL VOCデータに対して同程度の精度で4倍の推論速度を実現している。BiSeNetのようなリアルタイム セグメンテーション モデルと比較して、我々のモデルはCityscapesデータセットにおいて同等の速度でより高い精度を達成しており、自律運転のための街並み把握などの速度が要求されるタスクへ応用できる。さらに、我々のShelfNetはResNet34バックボーンを用いたCityscapes Datasetにおいて79.0%のmIoUを達成しており、ResNet101のような大規模バックボーンを用いたPSPNetやBiSeNetを凌駕している。大規模な実験を通じて、ShelfNetの優れた性能を検証した。実装へのリンクを提供する。https://github.com/juntang-zhuang/ShelfNet-lw-cityscapes
In this paper, we present ShelfNet, a novel architecture for accurate fast semantic segmentation. Different from the single encoder-decoder structure, ShelfNet has multiple encoder-decoder branch pairs with skip connections at each spatial level, which looks like a shelf with multiple columns. The shelf-shaped structure can be viewed as an ensemble of multiple deep and shallow paths, thus improving accuracy. We significantly reduce computation burden by reducing channel number, at the same time achieving high accuracy with this unique structure. In addition, we propose a shared-weight strategy in the residual block which reduces parameter number without sacrificing performance. Compared with popular non real-time methods such as PSPNet, our ShelfNet achieves 4× faster inference speed with similar accuracy on PASCAL VOC dataset. Compared with real-time segmentation models such as BiSeNet, our model achieves higher accuracy at comparable speed on the Cityscapes Dataset, enabling the application in speed-demanding tasks such as street-scene understanding for autonomous driving. Furthermore, our ShelfNet achieves 79.0% mIoU on Cityscapes Dataset with ResNet34 backbone, outperforming PSPNet and BiSeNet with large backbones such as ResNet101. Through extensive experiments, we validated the superior performance of ShelfNet. We provide link to the implementation https://github.com/juntang-zhuang/ShelfNet-lw-cityscapes.