English

Long-tailed Instance Segmentation using Gumbel Optimized Loss

Computer Vision and Pattern Recognition 2022-11-04 v2

Abstract

Major advancements have been made in the field of object detection and segmentation recently. However, when it comes to rare categories, the state-of-the-art methods fail to detect them, resulting in a significant performance gap between rare and frequent categories. In this paper, we identify that Sigmoid or Softmax functions used in deep detectors are a major reason for low performance and are sub-optimal for long-tailed detection and segmentation. To address this, we develop a Gumbel Optimized Loss (GOL), for long-tailed detection and segmentation. It aligns with the Gumbel distribution of rare classes in imbalanced datasets, considering the fact that most classes in long-tailed detection have low expected probability. The proposed GOL significantly outperforms the best state-of-the-art method by 1.1% on AP , and boosts the overall segmentation by 9.0% and detection by 8.0%, particularly improving detection of rare classes by 20.3%, compared to Mask-RCNN, on LVIS dataset. Code available at: https://github.com/kostas1515/GOL

Keywords

Cite

@article{arxiv.2207.10936,
  title  = {Long-tailed Instance Segmentation using Gumbel Optimized Loss},
  author = {Konstantinos Panagiotis Alexandridis and Jiankang Deng and Anh Nguyen and Shan Luo},
  journal= {arXiv preprint arXiv:2207.10936},
  year   = {2022}
}

Comments

ECCV2022

R2 v1 2026-06-25T01:08:25.304Z