학술논문

What is The Best Data Augmentation For 3D Brain Tumor Segmentation?
Document Type
Conference
Source
2021 IEEE International Conference on Image Processing (ICIP) Image Processing (ICIP), 2021 IEEE International Conference on. :36-40 Sep, 2021
Subject
Computing and Processing
Signal Processing and Analysis
Training
Image segmentation
Three-dimensional displays
Conferences
Brightness
Standards
Tumors
Data augmentation
3D brain tumor segmentation
MRI
3D U-Net
deep learning
artificial intelligence
Language
ISSN
2381-8549
Abstract
Training segmentation networks requires large annotated datasets, which in medical imaging can be hard to obtain. Despite this fact, data augmentation has in our opinion not been fully explored for brain tumor segmentation. In this project we apply different types of data augmentation (flipping, rotation, scaling, brightness adjustment, elastic deformation) when training a standard 3D U-Net, and demonstrate that augmentation significantly improves the network’s performance in many cases. Our conclusion is that brightness augmentation and elastic deformation work best, and that combinations of different augmentation techniques do not provide further improvement compared to only using one augmentation technique. Our code is available at https://github.com/mdciri/3D-augmentation-techniques.