MUM: Mix Image Tiles and UnMix Feature Tiles for Semi-Supervised Object Detection
Jongmok Kim, Jooyoung Jang, Seunghyeon Seo, Jisoo Jeong, Jongkeun Na, Nojun Kwak
Abstract
Many recent semi-supervised learning (SSL) studies build teacher-student architecture and train the student net-work by the generated supervisory signal from the teacher. Data augmentation strategy plays a significant role in the SSL framework since it is hard to create a weak-strong aug-mented input pair without losing label information. Espe-cially when extending SSL to semi-supervised object de-tection (SSOD), many strong augmentation methodologies related to image geometry and interpolation-regularization are hard to utilize since they possibly hurt the location information of the bounding box in the object detection task. To address this, we introduce a simple yet effective data augmentation method, Mix/UnMix (MUM), which un-mixes feature tiles for the mixed image tiles for the SSOD framework. Our proposed method makes mixed input image tiles and reconstructs them in the feature space. Thus, MUM can enjoy the interpolation-regularization effect from non-interpolated pseudo-labels and successfully generate a meaningful weak-strong pair. Furthermore, MUM can be easily equipped on top of various SSOD methods. Exten-sive experiments on MS-COCO and PASCAL VOC datasets demonstrate the superiority of MUM by consistently im-proving the mAP performance over the baseline in all the tested SSOD benchmark protocols. The code is released at https.//github.com/JongMokKim/mix-unmix.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers7
- DTG-SSOD: Dense Teacher Guidance for Semi-Supervised Object DetectionGang Li, Xiang Li, Yujie Wang, Yichao Wu et al.NeurIPS 2022 · 31 citations
- Adapting Object Size Variance and Class Imbalance for Semi-supervised Object DetectionYuxiang Nie, Chaowei Fang, Lechao Cheng, Liang Lin et al.AAAI 2023 · 19 citations
- FedCD: Federated Semi-Supervised Learning with Class Awareness Balance via Dual TeachersYuzhi Liu, Huisi Wu, Jing QinAAAI 2024 · 16 citations
- Learning with Noisy Data for Semi-Supervised 3D Object DetectionZehui Chen, Zhenyu Li, Shuo Wang, Dengpan Fu et al.ICCV 2023 · 14 citations
- Cycle Self-Training for Semi-Supervised Object Detection with Distribution Consistency ReweightingHao Liu, Bin Chen, Bo Wang, Chunpeng Wu et al.ACM MM 2022 · 8 citations
Builds on13
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- CutMix: Regularization Strategy to Train Strong Classifiers With Localizable FeaturesSangdoo Yun, Dongyoon Han, Sanghyuk Chun, Seong Joon Oh et al.ICCV 2019 · 5,843 citations
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and ConfidenceKihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang et al.NeurIPS 2020 · 5,129 citations
- RandAugment: Practical Automated Data Augmentation with a Reduced Search SpaceEkin Dogus Cubuk, Barret Zoph, Jonathon Shlens, Quoc LeNeurIPS 2020 · 4,453 citations
- Unsupervised Data Augmentation for Consistency TrainingQizhe Xie, Zihang Dai, Eduard H. Hovy, Thang Luong et al.NeurIPS 2020 · 2,774 citations
Related papers
- Instant-Teaching: An End-to-End Semi-Supervised Object Detection FrameworkQiang Zhou, Chaohui Yu, Zhibin Wang, Qi Qian et al.CVPR 2021
- Unbiased Teacher for Semi-Supervised Object DetectionYen-Cheng Liu, Chih-Yao Ma, Zijian He, Chia-Wen Kuo et al.ICLR 2021 · 603 citations
- Hierarchical Supervision and Shuffle Data Augmentation for 3D Semi-Supervised Object DetectionChuandong Liu, Chenqiang Gao, Fangcen Liu, Pengcheng Li et al.CVPR 2023
- Supervision Interpolation via LossMix: Generalizing Mixup for Object Detection and BeyondThanh Vu, Baochen Sun, Bodi Yuan, Alex Ngai et al.AAAI 2024 · 8 citations
- Interpolation-Based Semi-Supervised Learning for Object DetectionJisoo Jeong, Vikas Verma, Minsung Hyun, Juho Kannala et al.CVPR 2021
