Abstract
Deep neural networks for image classification only learn to map in-distribution inputs to their corresponding ground-truth labels in training without differentiating out-of-distribution samples from in-distribution ones. This results from the assumption that all samples are independent and identically distributed (IID) without distributional distinction. Therefore, a pretrained network learned from in-distribution samples treats out-of-distribution samples as in-distribution and makes high-confidence predictions on them in the test phase. To address this issue, we draw out-of-distribution samples from the vicinity distribution of training in-distribution samples for learning to reject the prediction on out-of-distribution inputs. A cross-class vicinity distribution is introduced by assuming that an out-of-distribution sample generated by mixing multiple in-distribution samples does not share the same classes of its constituents. We, thus, improve the discriminability of a pretrained network by finetuning it with out-of-distribution samples drawn from the cross-class vicinity distribution, where each out-of-distribution input corresponds to a complementary label. Experiments on various in-/out-of-distribution datasets show that the proposed method significantly outperforms the existing methods in improving the capacity of discriminating between in-and out-of-distribution samples.
| Original language | English |
|---|---|
| Pages (from-to) | 13777-13788 |
| Number of pages | 12 |
| Journal | IEEE Transactions on Neural Networks and Learning Systems |
| Volume | 35 |
| Issue number | 10 |
| Early online date | 26 May 2023 |
| DOIs | |
| Publication status | Published - Oct 2024 |
Fingerprint
Dive into the research topics of 'Out-of-distribution detection by cross-class vicinity distribution of in-distribution data'. Together they form a unique fingerprint.Projects
- 2 Finished
-
Deep Interaction Learning in Unlabelled Big Data and Complex Systems (FT190100734)
Cao, L. (Primary Chief Investigator)
1/06/20 → 31/05/24
Project: Research
-
Deep analytics of non-occurring but important behaviours (DP190101079)
Cao, L. (Primary Chief Investigator)
1/01/20 → 30/09/23
Project: Research
Cite this
- APA
- Author
- BIBTEX
- Harvard
- Standard
- RIS
- Vancouver