Out of distribution detection for skin and malaria images

Abstract

Deep neural networks have shown promising results in disease detection and classification using medical image data. However, they still suffer from the challenges of handling real-world scenarios especially reliably detecting out-of-distribution (OoD) samples. We propose an approach to robustly classify OoD samples in skin and malaria images without the need to access labeled OoD samples during training. Specifically, we use metric learning along with logistic regression to force the deep networks to learn much rich class representative features. To guide the learning process against the OoD examples, we generate ID similar-looking examples by either removing class-specific salient regions in the image or permuting image parts and distancing them away from in-distribution samples. During inference time, the K-reciprocal nearest neighbor is employed to detect out-of-distribution samples. For skin cancer OoD detection, we employ two standard benchmark skin cancer ISIC datasets as ID, and six different datasets with varying difficulty levels were taken as out of distribution. For malaria OoD detection, we use the BBBC041 malaria dataset as ID and five different challenging datasets as out of distribution. We achieved state-of-the-art results, improving 5% and 4% in TNR@ TPR95% over the previous state-of-the-art for skin cancer and malaria OoD detection respectively.

Out-of-Distribution (OOD) Detection Based on Deep Learning: A Review

“…However, medical image interpretation is limited by physician subjectivity, cognitive differences, and fatigue. Muhammad et al [83] proposed an approach to robustly classify OOD samples in skin and malaria images without accessing labeled OOD samples during training. This method has reached its most advanced level in detecting skin cancer and malaria.…”

Data-Driven Quantitation of Movement Abnormality after Stroke

“…OOD detection has been increasingly used in medical image processing, for example, to identify and score brain lesions in MRIs [ 31 ], to distinguish malignancies in histopathological samples [ 32 ], to detect skin cancer in images [ 33 ], and to detect malaria in blood smears [ 33 ]. OOD detection has also been applied to motion data captured by sensors [ 34 ] or video [ 35 ] to detect new activities unrelated to trained activities.…”