Strengthen Out-of-Distribution Detection Capability with Progressive Self-Knowledge Distillation
Yang Yang, Haonan Xu
Abstract
Out-of-distribution (OOD) detection aims to ensure AI system reliability by rejecting inputs outside the training distribution. Recent work shows that memorizing atypical samples during later stages of training can hurt OOD detection, while strategies for forgetting them show promising improvements. However, directly forgetting atypical samples sacrifices ID generalization and limits the model's OOD detection capability. To address this issue, we propose Progressive Self-Knowledge Distillation (PSKD) framework, which strengthens the OOD detection capability by leveraging self-provided uncertaintyembedded targets. Specifically, PSKD adaptively selects a self-teacher model from the training history using pseudo-outliers, facilitating the learning of uncertainty knowledge via multi-level distillation applied to features and responses. As a result, PSKD achieves better ID generalization and uncertainty estimation for OOD detection. Moreover, PSKD is orthogonal to most existing methods and can be integrated as a plugin to collaborate with them. Experimental results from multiple OOD scenarios verify the effectiveness and general applicability of PSKD.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a6a1c225-9cf9-4e67-86b9-d40a4b542e8aCited by top-tier papers4
- TrustworthyQENN: A Quantum Evidential Neural Network Based on Complex-Valued Contrastive Learning for Uncertainty Pattern ClassificationXiaolong Chen, Fuyuan Xiao, Xiaohong Zhang, Zehong Cao et al.ICML 2026
- Shaping Parameter Contribution Patterns for Out-of-Distribution DetectionHaonan Xu, Yang YangAAAI 2026
- PAF: Perturbation-Aware Filtering for Open-Set Semi-Supervised LearningYinan Han, Qingyuan JiangCVPR 2026
- GR-LoRA: Gradient-Recycling Low-Rank Adaptation for Class-Incremental LearningYipeng Lin, Fengqiang Wan, Yang YangICML 2026
Builds on27
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Energy-based Out-of-distribution DetectionWeitang Liu, Xiaoyun Wang, John D. Owens, Yixuan LiNeurIPS 2020 · 2,213 citations
- Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self DistillationLinfeng Zhang, Jiebo Song, Anni Gao, Jingwei Chen et al.ICCV 2019 · 1,069 citations
- Out-of-Distribution Detection with Deep Nearest NeighborsYiyou Sun, Yifei Ming, Xiaojin Zhu, Yixuan LiICML 2022 · 789 citations
- Open-Set Recognition: A Good Closed-Set Classifier is All You NeedSagar Vaze, Kai Han, Andrea Vedaldi, Andrew ZissermanICLR 2022 · 594 citations
Related papers
- Self-Knowledge Distillation with Progressive Refinement of TargetsKyungyul Kim, Byeongmoon Ji, Doyoung Yoon, Sangheum HwangICCV 2021 · 251 citations
- Multi-Level Knowledge Distillation for Out-of-Distribution Detection in TextQianhui Wu, Huiqiang Jiang, Haonan Yin, Börje Karlsson et al.ACL 2023 · 7 citations
- PseDet: Revisiting the Power of Pseudo Label in Incremental Object DetectionQiuchen Wang, Zehui Chen, Chenhongyi Yang, Jiaming Liu et al.ICLR 2025
- Refine Myself by Teaching Myself: Feature Refinement via Self-Knowledge DistillationMingi Ji, Seungjae Shin, Seunghyun Hwang, Gibeom Park et al.CVPR 2021
- Multi-label Self Knowledge DistillationXucong Wang, Pengkun Wang, Shurui Zhang, Miao Fang et al.AAAI 2025 · 2 citations
