Large Batch and Patch Size Training for Medical Image Segmentation
Multi-organ segmentation enables organ evaluation, accounts the relationship between multiple organs, and facilitates accurate diagnosis and treatment decisions. However, only few models can perform segmentation accurately because of the lack of datasets and computational resources. On AMOS2022 challenge, which is a large-scale, clinical, and diverse abdominal multiorgan segmentation benchmark, we trained a 3D-UNet model with large batch and patch sizes using multi-GPU distributed training. Segmentation performance tended to increase for models with large batch and patch sizes compared with the baseline settings. The accuracy was further improved by using ensemble models that were trained with different settings. These results provide a reference for parameter selection in organ segmentation.
Code (0)
등록된 구현이 없습니다.
Tasks
GPUImage SegmentationMedical Image SegmentationOrgan SegmentationSegmentationSemantic SegmentationSimilar Papers 제목 키워드 기반
VariViT: A Vision Transformer for Variable Image Sizes
Vision Transformers (ViTs) have emerged as the state-of-the-art architecture in representation learning, leveraging self-attention mechanisms to excel in various tasks. ViTs split images into fixed-size patches, constrai…
Brain Tumor ClassificationRepresentation LearningFast and Accurate 3D Medical Image Segmentation with Data-swapping Method
Deep neural network models used for medical image segmentation are large because they are trained with high-resolution three-dimensional (3D) images. Graphics processing units (GPUs) are widely used to accelerate the tra…
CPUGPUImage SegmentationMedical Image Segmentation+2Modelling brain lesion volume in patches with CNN-based Poisson Regression
Monitoring the progression of lesions is important for clinical response. Summary statistics such as lesion volume are objective and easy to interpret, which can help clinicians assess lesion growth or decay. CNNs are co…
Image SegmentationMedical Image SegmentationModel Selectionregression+2Memory-efficient GAN-based Domain Translation of High Resolution 3D Medical Images
Generative adversarial networks (GANs) are currently rarely applied on 3D medical images of large size, due to their immense computational demand. The present work proposes a multi-scale patch-based GAN approach for esta…
GPUImage GenerationTranslationPhysics-informed Neural Networks for Solving Inverse Problems of Nonlinear Biot's Equations: Batch Training
In biomedical engineering, earthquake prediction, and underground energy harvesting, it is crucial to indirectly estimate the physical properties of porous media since the direct measurement of those are usually impracti…
Earthquake prediction