Memory-efficient Segmentation of High-resolution Volumetric MicroCT Images
Not on the map. Why?
Abstract (arXiv)
In recent years, 3D convolutional neural networks have become the dominant approach for volumetric medical image segmentation. However, compared to their 2D counterparts, 3D networks introduce substantially more training parameters and higher requirement for the GPU memory. This has become a major limiting factor for designing and training 3D networks for high-resolution volumetric images. In this work, we propose a novel memory-efficient network architecture for 3D high-resolution image segmentation. The network incorporates both global and local features via a two-stage U-net-based cascaded framework and at the first stage, a memory-efficient U-net (meU-net) is developed. The features learnt at the two stages are connected via post-concatenation, which further improves the information flow. The proposed segmentation method is evaluated on an ultra high-resolution microCT dataset with typically 250 million voxels per volume. Experiments show that it outperforms state-of-the-art 3D segmentation methods in terms of both segmentation accuracy and memory efficiency.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 19 of its 24 references that are in the map. First posted on arXiv 31 May 2022.
Builds on
19 of 24 references.
References · 19
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
- 2017Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentationscited by 614
- 2015Holistically-Nested Edge Detectioncited by 590
- 2014Deeply-Supervised Netscited by 364
- 2021CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentationcited by 227
7 more
- 2021Multi-Compound Transformer for Accurate Biomedical Image Segmentationcited by 42
- 2017Automatic 3D Cardiovascular MR Segmentation with Densely-Connected Volumetric ConvNetscited by 32
- 2017Deeply-Supervised CNN for Prostate Segmentationcited by 30
- 2018A multi-scale pyramid of 3D fully convolutional networks for abdominal multi-organ segmentationcited by 17
- 2019A Partially Reversible U-Net for Memory-Efficient Volumetric Image Segmentationcited by 16
- 2021OSS-Net: Memory Efficient High Resolution Semantic Segmentation of 3D Medical Datacited by 9
- 2019CU-Net: Cascaded U-Net with Loss Weighted Sampling for Brain Tumor Segmentationcited by 6
Built on by
Citing papers per year: 2026: 1.
1 paper.