arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.11978cs.CV

The Devil is in Classification: A Simple Framework for Long-tail Object Detection and Instance Segmentation

NGS, National University of Singapore(新加坡国立大学);ECE Department, National University of Singapore(新加坡国立大学电机与计算机工程系);Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);Salesforce Research Asia(Salesforce亚洲研究院)

Tao Wang, Yu Li, Bingyi Kang, Junnan Li, Junhao Liew, Sheng Tang, Steven Hoi, Jiashi Feng

英文摘要:

Most existing object instance detection and segmentation models only work well on fairly balanced benchmarks where per-category training sample numbers are comparable, such as COCO. They tend to suffer performance drop on realistic datasets that are usually long-tailed. This work aims to study and address such open challenges. Specifically, we systematically investigate performance drop of the state-of-the-art two-stage instance segmentation model Mask R-CNN on the recent long-tail LVIS dataset, and unveil that a major cause is the inaccurate classification of object proposals. Based on such an observation, we first consider various techniques for improving long-tail classification performance which indeed enhance instance segmentation results. We then propose a simple calibration framework to more effectively alleviate classification head bias with a bi-level class balanced sampling approach. Without bells and whistles, it significantly boosts the performance of instance segmentation for tail classes on the recent LVIS dataset and our sampled COCO-LT dataset. Our analysis provides useful insights for solving long-tail instance detection and segmentation problems, and the straightforward \emph{SimCal} method can serve as a simple but strong baseline. With the method we have won the 2019 LVIS challenge. Codes and models are available at https://github.com/twangnh/SimCal.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑