Long-Tailed Instance Segmentation Using Gumbel Optimized Loss

Kostas Alexandridis, Jiankang Deng, Anh Nguyen, Shan Luo*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference paperpeer-review

17 Citations (Scopus)

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.

Original languageEnglish
Title of host publicationComputer Vision – ECCV 2022 - 17th European Conference, Proceedings
EditorsShai Avidan, Gabriel Brostow, Moustapha Cissé, Giovanni Maria Farinella, Tal Hassner
PublisherSpringer Science and Business Media Deutschland GmbH
Pages353-369
Number of pages17
ISBN (Print)9783031200793
DOIs
Publication statusPublished - 2022
Event17th European Conference on Computer Vision, ECCV 2022 - Tel Aviv, Israel
Duration: 23 Oct 202227 Oct 2022

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13670 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference17th European Conference on Computer Vision, ECCV 2022
Country/TerritoryIsrael
CityTel Aviv
Period23/10/202227/10/2022

Keywords

  • Gumbel activation
  • Long-tailed distribution
  • Long-tailed instance segmentation

Fingerprint

Dive into the research topics of 'Long-Tailed Instance Segmentation Using Gumbel Optimized Loss'. Together they form a unique fingerprint.

Cite this