Beyond CNNs: Exploiting Further Inherent Symmetries in Medical Images
for Segmentation
- URL: http://arxiv.org/abs/2005.03924v1
- Date: Fri, 8 May 2020 09:36:50 GMT
- Title: Beyond CNNs: Exploiting Further Inherent Symmetries in Medical Images
for Segmentation
- Authors: Shuchao Pang, Anan Du, Mehmet A. Orgun, Yan Wang, Quanzheng Sheng,
Shoujin Wang, Xiaoshui Huang, Zhemei Yu
- Abstract summary: We propose a novel group equivariant segmentation framework for medical tumor segmentation.
By exploiting further symmetries, novel segmentation CNNs can dramatically reduce the sample complexity and the redundancy of filters.
We show that a newly built GER-UNet outperforms its regular CNN-based counterpart and the state-of-the-art segmentation methods on real-world clinical data.
- Score: 11.797343325320474
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Automatic tumor segmentation is a crucial step in medical image analysis for
computer-aided diagnosis. Although the existing methods based on convolutional
neural networks (CNNs) have achieved the state-of-the-art performance, many
challenges still remain in medical tumor segmentation. This is because regular
CNNs can only exploit translation invariance, ignoring further inherent
symmetries existing in medical images such as rotations and reflections. To
mitigate this shortcoming, we propose a novel group equivariant segmentation
framework by encoding those inherent symmetries for learning more precise
representations. First, kernel-based equivariant operations are devised on
every orientation, which can effectively address the gaps of learning
symmetries in existing approaches. Then, to keep segmentation networks globally
equivariant, we design distinctive group layers with layerwise symmetry
constraints. By exploiting further symmetries, novel segmentation CNNs can
dramatically reduce the sample complexity and the redundancy of filters (by
roughly 2/3) over regular CNNs. More importantly, based on our novel framework,
we show that a newly built GER-UNet outperforms its regular CNN-based
counterpart and the state-of-the-art segmentation methods on real-world
clinical data. Specifically, the group layers of our segmentation framework can
be seamlessly integrated into any popular CNN-based segmentation architectures.
Related papers
- Heterogenous Memory Augmented Neural Networks [84.29338268789684]
We introduce a novel heterogeneous memory augmentation approach for neural networks.
By introducing learnable memory tokens with attention mechanism, we can effectively boost performance without huge computational overhead.
We show our approach on various image and graph-based tasks under both in-distribution (ID) and out-of-distribution (OOD) conditions.
arXiv Detail & Related papers (2023-10-17T01:05:28Z) - Self-supervised Semantic Segmentation: Consistency over Transformation [3.485615723221064]
We propose a novel self-supervised algorithm, textbfS$3$-Net, which integrates a robust framework based on the proposed Inception Large Kernel Attention (I-LKA) modules.
We leverage deformable convolution as an integral component to effectively capture and delineate lesion deformations for superior object boundary definition.
Our experimental results on skin lesion and lung organ segmentation tasks show the superior performance of our method compared to the SOTA approaches.
arXiv Detail & Related papers (2023-08-31T21:28:46Z) - Implicit Anatomical Rendering for Medical Image Segmentation with
Stochastic Experts [11.007092387379078]
We propose MORSE, a generic implicit neural rendering framework designed at an anatomical level to assist learning in medical image segmentation.
Our approach is to formulate medical image segmentation as a rendering problem in an end-to-end manner.
Our experiments demonstrate that MORSE can work well with different medical segmentation backbones.
arXiv Detail & Related papers (2023-04-06T16:44:03Z) - Self-Supervised Correction Learning for Semi-Supervised Biomedical Image
Segmentation [84.58210297703714]
We propose a self-supervised correction learning paradigm for semi-supervised biomedical image segmentation.
We design a dual-task network, including a shared encoder and two independent decoders for segmentation and lesion region inpainting.
Experiments on three medical image segmentation datasets for different tasks demonstrate the outstanding performance of our method.
arXiv Detail & Related papers (2023-01-12T08:19:46Z) - Reliable Joint Segmentation of Retinal Edema Lesions in OCT Images [55.83984261827332]
In this paper, we propose a novel reliable multi-scale wavelet-enhanced transformer network.
We develop a novel segmentation backbone that integrates a wavelet-enhanced feature extractor network and a multi-scale transformer module.
Our proposed method achieves better segmentation accuracy with a high degree of reliability as compared to other state-of-the-art segmentation approaches.
arXiv Detail & Related papers (2022-12-01T07:32:56Z) - Attentive Symmetric Autoencoder for Brain MRI Segmentation [56.02577247523737]
We propose a novel Attentive Symmetric Auto-encoder based on Vision Transformer (ViT) for 3D brain MRI segmentation tasks.
In the pre-training stage, the proposed auto-encoder pays more attention to reconstruct the informative patches according to the gradient metrics.
Experimental results show that our proposed attentive symmetric auto-encoder outperforms the state-of-the-art self-supervised learning methods and medical image segmentation models.
arXiv Detail & Related papers (2022-09-19T09:43:19Z) - Beyond CNNs: Exploiting Further Inherent Symmetries in Medical Image
Segmentation [21.6412682130116]
We propose a novel group equivariant segmentation framework by encoding those inherent symmetries for learning more precise representations.
Based on our novel framework, extensive experiments conducted on real-world clinical data demonstrate that a Group Equivariant Res-UNet (named GER-UNet) outperforms its regular CNN-based counterpart.
The newly built GER-UNet also shows potential in reducing the sample complexity and the redundancy of filters.
arXiv Detail & Related papers (2022-07-29T04:28:20Z) - Two-Stream Graph Convolutional Network for Intra-oral Scanner Image
Segmentation [133.02190910009384]
We propose a two-stream graph convolutional network (i.e., TSGCN) to handle inter-view confusion between different raw attributes.
Our TSGCN significantly outperforms state-of-the-art methods in 3D tooth (surface) segmentation.
arXiv Detail & Related papers (2022-04-19T10:41:09Z) - SynthSeg: Domain Randomisation for Segmentation of Brain MRI Scans of
any Contrast and Resolution [7.070890465817133]
Convolutional neural networks (CNNs) have difficulties generalising to unseen target domains.
We introduce SynthSeg, the first segmentation CNN to brain MRI scans of any contrast and resolution.
We demonstrate SynthSeg on 5,500 scans of 6 modalities and 10 resolutions, where it exhibits unparalleled generalisation.
arXiv Detail & Related papers (2021-07-20T15:22:16Z) - Automatic size and pose homogenization with spatial transformer network
to improve and accelerate pediatric segmentation [51.916106055115755]
We propose a new CNN architecture that is pose and scale invariant thanks to the use of Spatial Transformer Network (STN)
Our architecture is composed of three sequential modules that are estimated together during training.
We test the proposed method in kidney and renal tumor segmentation on abdominal pediatric CT scanners.
arXiv Detail & Related papers (2021-07-06T14:50:03Z) - Segmentation-Renormalized Deep Feature Modulation for Unpaired Image
Harmonization [0.43012765978447565]
Cycle-consistent Generative Adversarial Networks have been used to harmonize image sets between a source and target domain.
These methods are prone to instability, contrast inversion, intractable manipulation of pathology, and steganographic mappings which limit their reliable adoption in real-world medical imaging.
We propose a segmentation-renormalized image translation framework to reduce inter-scanner harmonization while preserving anatomical layout.
arXiv Detail & Related papers (2021-02-11T23:53:51Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.