Self-Classification Enhancement and Correction for Weakly Supervised Object Detection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yin, Yufei, Cheng, Lechao, Zhou, Wengang, Deng, Jiajun, Yu, Zhou, Li, Houqiang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917343000526848
author Yin, Yufei
Cheng, Lechao
Zhou, Wengang
Deng, Jiajun
Yu, Zhou
Li, Houqiang
author_facet Yin, Yufei
Cheng, Lechao
Zhou, Wengang
Deng, Jiajun
Yu, Zhou
Li, Houqiang
contents In recent years, weakly supervised object detection (WSOD) has attracted much attention due to its low labeling cost. The success of recent WSOD models is often ascribed to the two-stage multi-class classification (MCC) task, i.e., multiple instance learning and online classification refinement. Despite achieving non-trivial progresses, these methods overlook potential classification ambiguities between these two MCC tasks and fail to leverage their unique strengths. In this work, we introduce a novel WSOD framework to ameliorate these two issues. For one thing, we propose a self-classification enhancement module that integrates intra-class binary classification (ICBC) to bridge the gap between the two distinct MCC tasks. The ICBC task enhances the network's discrimination between positive and mis-located samples in a class-wise manner and forges a mutually reinforcing relationship with the MCC task. For another, we propose a self-classification correction algorithm during inference, which combines the results of both MCC tasks to effectively reduce the mis-classified predictions. Extensive experiments on the prevalent VOC 2007 & 2012 datasets demonstrate the superior performance of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2505_16294
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Self-Classification Enhancement and Correction for Weakly Supervised Object Detection
Yin, Yufei
Cheng, Lechao
Zhou, Wengang
Deng, Jiajun
Yu, Zhou
Li, Houqiang
Computer Vision and Pattern Recognition
In recent years, weakly supervised object detection (WSOD) has attracted much attention due to its low labeling cost. The success of recent WSOD models is often ascribed to the two-stage multi-class classification (MCC) task, i.e., multiple instance learning and online classification refinement. Despite achieving non-trivial progresses, these methods overlook potential classification ambiguities between these two MCC tasks and fail to leverage their unique strengths. In this work, we introduce a novel WSOD framework to ameliorate these two issues. For one thing, we propose a self-classification enhancement module that integrates intra-class binary classification (ICBC) to bridge the gap between the two distinct MCC tasks. The ICBC task enhances the network's discrimination between positive and mis-located samples in a class-wise manner and forges a mutually reinforcing relationship with the MCC task. For another, we propose a self-classification correction algorithm during inference, which combines the results of both MCC tasks to effectively reduce the mis-classified predictions. Extensive experiments on the prevalent VOC 2007 & 2012 datasets demonstrate the superior performance of our framework.
title Self-Classification Enhancement and Correction for Weakly Supervised Object Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.16294