📖 IEEE TPAMI 2025
🖋️ Author: Yu Wang, Junxian Mu, Hongzhi Huang, Qilong Wang, Pengfei Zhu, Qinghua Hu
🏫 Tianjin Key Lab of Machine Learning, College of Intelligence and Computing, Tianjin University, China
🏛️ Haihe Laboratory of Information Technology Application Innovation (Haihe Lab of ITAI), Tianjin, China
Open set recognition (OSR) requires models to classify known samples while detecting unknowns. Current methods use auxiliary datasets to regularize OSR models but are sensitive to outlier selection. We propose a new perspective: Can we regularize OSR models without carefully chosen outliers? Our analysis reveals that:
-
Correlated backgrounds mislead models on 'partially' known images;
-
Unrelated backgrounds can serve as natural outliers.
Based on this, we propose Background Mix (BackMix), which mixes image foregrounds (estimated via CAMs) with random backgrounds to break fore-background correlations. This simple yet effective method improves OSR performance without inference overhead and integrates easily with existing frameworks.
BackMix first estimates and masks the foreground of the background image, then randomly cuts patches and pastes them on the target image to obtain the mixed image as the training sample.
backmix.mp4 |
| Step-by-step workflow of Backmix |
We train our models under python=3.8, pytorch=1.11, cuda=11.8. Other versions might be available as well.
- Install Pytorch and torchvision. Follow the instruction on https://pytorch.org/get-started/locally/.
# an example:
conda install -c pytorch pytorch torchvision- Install other needed packages (numpy, scikit-learn, etc.)
If you find our code / paper useful in your research, please consider cite our work. 💖
@article{wang2025backmix,
author={Wang, Yu and Mu, Junxian and Huang, Hongzhi and Wang, Qilong and Zhu, Pengfei and Hu, Qinghua},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
title={BackMix: Regularizing Open Set Recognition by Removing Underlying Fore-Background Priors},
year={2025},
pages={1-12},
doi={10.1109/TPAMI.2025.3550703}
}
We acknowledge the following repositories for providing useful components / functions in our work. 💁♀️
ARPL (Adversarial Reciprocal Points Learning for Open Set Recognition [TPAMI'2022])
CSSR (Class Specific Semantic Reconstruction for Open Set Recognition [TPAMI'2023])