A Multi-Scale Framework for Out-of-Distribution Detection in Dermoscopic Images

Abstract

The automatic detection of skin diseases via dermoscopic images can improve the efficiency in diagnosis and help doctors make more accurate judgments. However, conventional skin disease recognition systems may produce high confidence for out-of-distribution (OOD) data, which may become a major security vulnerability in practical applications. In this paper, we propose a multi-scale detection framework to detect out-of-distribution skin disease image data to ensure the robustness of the system. Our framework extracts features from different layers of the neural network. In the early layers, rectified activation is used to make the output features closer to the well-behaved distribution, and then an one-class SVM is trained to detect OOD data; in the penultimate layer, an adapted Gram matrix is used to calculate the features after rectified activation, and finally the layer with the best performance is chosen to compute a normality score. Experiments show that the proposed framework achieves superior performance when compared with other state-of-the-art methods in the task of skin disease recognition.

arXiv commentPaper accepted by the 4th International Conference on Machine Learning for Cyber Security (ML4CS 2022), Guangzhou, China

References in corpus (11)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions
• A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networks
• A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks
• Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networks
• Energy-based Out-of-distribution Detection
• Generalized Out-of-Distribution Detection: A Survey
• BCN20000: Dermoscopic Lesions in the Wild
• Semi-supervised Medical Image Classification with Relation-driven Self-ensembling Model
• Does Your Dermatology Classifier Know What It Doesn't Know? Detecting the Long-Tail of Unseen Conditions
• Detecting Out-of-Distribution Inputs in Deep Neural Networks Using an Early-Layer Output