English

Large Batch and Patch Size Training for Medical Image Segmentation

Image and Video Processing 2022-10-25 v1 Computer Vision and Pattern Recognition Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2210.13364,
  title  = {Large Batch and Patch Size Training for Medical Image Segmentation},
  author = {Junya Sato and Shoji Kido},
  journal= {arXiv preprint arXiv:2210.13364},
  year   = {2022}
}
R2 v1 2026-06-28T04:22:36.165Z