An Efficient Data Augmentation Network for Out-of-Distribution Image Detection

Cheng Hung Lin, Cheng Shian Lin, Po Yung Chou, Chen Chien Hsu

研究成果: 雜誌貢獻期刊論文同行評審

摘要

Since deep neural networks may classify out-of-distribution image data into in-distribution classes with high confidence scores, this problem may cause serious or even fatal hazards in certain applications, such as autonomous vehicles and medical diagnosis. Therefore, out-of-distribution detection (also called anomaly detection or outlier detection) of image classification has become a critical issue for the successful development of neural networks. In other words, a successful neural network needs to be able to distinguish anomalous data that is significantly different from the data used in training. In this paper, we propose an efficient data augmentation network to detect out-of-distribution image data by introducing a set of common geometric operations into training and testing images. The output predicted probabilities of the augmented data are combined by an aggregation function to provide a confidence score to distinguish between in-distribution and out-of-distribution image data. Different from other approaches that use out-of-distribution image data for training networks, we only use in-distribution image data in the proposed data augmentation network. This advantage makes our approach more practical than other approaches, and can be easily applied to various neural networks to improve security in practical applications. The experimental results show that the proposed data augmentation network outperforms the state-of-the-art approaches in various datasets. In addition, pre-training techniques can be integrated into the data augmentation network to make substantial improvements to large and complex data sets. The code is available at www.github.com/majic0626/Data-Augmentation-Network.git.

原文英語
文章編號9363111
頁(從 - 到)35313-35323
頁數11
期刊IEEE Access
9
DOIs
出版狀態接受/付印 - 2021

ASJC Scopus subject areas

  • 電腦科學(全部)
  • 材料科學(全部)
  • 工程 (全部)

指紋

深入研究「An Efficient Data Augmentation Network for Out-of-Distribution Image Detection」主題。共同形成了獨特的指紋。

引用此