Method Details


Details for method 'Axial-DeepLab-L [Mapillary Vistas]'

 

Method overview

name Axial-DeepLab-L [Mapillary Vistas]
challenge pixel-level semantic labeling
details Convolution exploits locality for efficiency at a cost of missing long range context. Self-attention has been adopted to augment CNNs with non-local interactions. Recent works prove it possible to stack self-attention layers to obtain a fully attentional network by restricting the attention to a local region. In this paper, we attempt to remove this constraint by factorizing 2D self-attention into two 1D self-attentions. This reduces computation complexity and allows performing attention within a larger or even global region. In companion, we also propose a position-sensitive self-attention design. Combining both yields our position-sensitive axial-attention layer, a novel building block that one could stack to form axial-attention models for image classification and dense prediction. We demonstrate the effectiveness of our model on four large-scale datasets. In particular, our model outperforms all existing stand-alone self-attention models on ImageNet. Our Axial-DeepLab improves 2.8% PQ over bottom-up state-of-the-art on COCO test-dev. This previous state-of-the-art is attained by our small variant that is 3.8x parameter-efficient and 27x computation-efficient. Axial-DeepLab also achieves state-of-the-art results on Mapillary Vistas and Cityscapes.
publication Axial-DeepLab: Stand-Alone Axial-Attention for Panoptic Segmentation
Huiyu Wang, Yukun Zhu, Bradley Green, Hartwig Adam, Alan Yuille, Liang-Chieh Chen
ECCV 2020 (spotlight)
https://arxiv.org/abs/2003.07853
project page / code https://github.com/csrhddlam/axial-deeplab
used Cityscapes data fine annotations
used external data ImageNet, Mapillary Vistas
runtime n/a
subsampling no
submission date March, 2020
previous submissions

 

Average results

Metric Value
IoU Classes 83.1423
iIoU Classes 64.023
IoU Categories 92.3349
iIoU Categories 78.9062

 

Class results

Class IoU iIoU
road 98.8119 -
sidewalk 87.8007 -
building 94.1522 -
wall 59.7815 -
fence 68.0794 -
pole 73.4194 -
traffic light 79.4888 -
traffic sign 82.6183 -
vegetation 93.948 -
terrain 72.7257 -
sky 96.0848 -
person 88.9012 71.1657
rider 77.5147 58.2308
car 96.4877 87.0429
truck 76.8575 51.2779
bus 91.2153 64.0995
train 92.6248 63.8267
motorcycle 74.7286 57.8978
bicycle 74.4632 58.6424

 

Category results

Category IoU iIoU
flat 98.7371 -
nature 93.6701 -
object 78.642 -
sky 96.0848 -
construction 94.4179 -
human 88.8617 72.3049
vehicle 95.931 85.5074

 

Links

Download results as .csv file

Benchmark page