Balance-batch: An Optimized Method for Semantic Segmentation Loss Functions

Yifeng Huang,Zhirong Tang,Kaixiong Su
DOI: https://doi.org/10.1109/CVIDL51233.2020.00-74
2020-01-01
Abstract:Class-imbalanced data easily generates under-fitting problems in deep neural networks, which seriously limits the performance of the network. Several schemes have been proposed to alleviate the class-imbalance, i.e., data augmentation and network structure optimization. Our work has two main contributions: First, we proposed an optimized method Balance-batch which invests equal consideration for each class in mini-batch and tries to balance the losses of classes. Optimizing loss functions is a simpler and more effective way to tackle the class-imbalance than others. Second, we extend the binary classification Dice loss which is employed in medical images segmentation to multiple classifications. Multi-classifications dice loss reflects the improvement of Balance-batch to unstable loss functions. Our various loss function experiments on Pascal VOC2012 show the effectiveness of Balance-batch, which is beyond the advanced level of these loss function methods.
What problem does this paper attempt to address?