Sample Efficient Semantic Segmentation using Rotation Equivariant Convolutional Networks
We propose a semantic segmentation model that exploits rotation and reflection symmetries. We demonstrate significant gains in sample efficiency due to increased weight sharing, as well as improvements in robustness to symmetry transformations. The group equivariant CNN framework is extended for segmentation by introducing a new equivariant (G->Z2)-convolution that transforms feature maps on a group to planar feature maps. Also, equivariant transposed convolution is formulated for up-sampling in an encoder-decoder network. To demonstrate improvements in sample efficiency we evaluate on multiple data regimes of a rotation-equivariant segmentation task: cancer metastases detection in histopathology images. We further show the effectiveness of exploiting more symmetries by varying the size of the group.
Code (0)
등록된 구현이 없습니다.
Tasks
DecoderSegmentationSemantic SegmentationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Leveraging SO(3)-steerable convolutions for pose-robust semantic segmentation in 3D medical data
Convolutional neural networks (CNNs) allow for parameter sharing and translational equivariance by using convolutional kernels in their linear layers. By restricting these kernels to be SO(3)-steerable, CNNs can further …
Data AugmentationMedical Image AnalysisSegmentationSemantic SegmentationDeeply Supervised Rotation Equivariant Network for Lesion Segmentation in Dermoscopy Images
Automatic lesion segmentation in dermoscopy images is an essential step for computer-aided diagnosis of melanoma. The dermoscopy images exhibits rotational and reflectional symmetry, however, this geometric property has …
Lesion SegmentationSegmentationSkin Lesion SegmentationImproved Vessel Segmentation with Symmetric Rotation-Equivariant U-Net
Automated segmentation plays a pivotal role in medical image analysis and computer-assisted interventions. Despite the promising performance of existing methods based on convolutional neural networks (CNNs), they neglect…
Medical Image AnalysisSegmentationRotation-Invariant Point Convolution With Multiple Equivariant Alignments
Recent attempts at introducing rotation invariance or equivariance in 3D deep learning approaches have shown promising results, but these methods still struggle to reach the performances of standard 3D neural networks. I…
Deep LearningSemantic Segmentation4D Panoptic Segmentation as Invariant and Equivariant Field Prediction
In this paper, we develop rotation-equivariant neural networks for 4D panoptic segmentation. 4D panoptic segmentation is a benchmark task for autonomous driving that requires recognizing semantic classes and object insta…
4D Panoptic SegmentationAutonomous DrivingPanoptic SegmentationSegmentation