Abstract
In this paper, we study the problem of out-of-distribution (OOD) detection in skin lesion images. Publicly available medical datasets have a limited number of lesion classes compared to the number of possible diseases in real-life clinical applications. It is thus essential to develop methods that leverage available disease classes in existing datasets to detect previously-unseen types in an unsupervised manner. Toward this goal, we propose an unsupervised and non-parametric OOD detection approach, called DeepIF, which learns the normal distribution of features in a pre-trained CNN using Isolation Forests. We conduct comprehensive experiments on two different datasets and compare our DeepIF against four baseline models. Results demonstrate state-of-the-art performance of our proposed approach on the task of detecting unseen skin lesions.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Blei, D.M., Jordan, M.I., et al.: Variational inference for dirichlet process mixtures. Bayesian Anal. 1(1), 121–143 (2006)
DeVries, T.: Learning confidence for out-of-distribution detection in neural networks (2018). https://github.com/uoguelph-mlrg/confidence_estimation
DeVries, T., Taylor, G.W.: Learning confidence for out-of-distribution detection in neural networks. arXiv preprint (2018). arXiv:1802.04865
Esteva, A., et al.: Dermatologist-level classification of skin cancer with deep neural networks. Nature 542(7639), 115–118 (2017)
Han, S.S., et al.: Deep neural networks show an equivalent and often superior performance to dermatologists in onychomycosis diagnosis: automatic construction of onychomycosis datasets by region-based convolutional deep neural network. PloS one 13(1), e0191493 (2018)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 770–778 (2016)
Hendrycks, D., Gimpel, K.: A baseline for detecting misclassified and out-of-distribution examples in neural networks. arXiv preprint (2016). arXiv:1610.02136
Hendrycks, D., Mazeika, M., Dietterich, T.: Deep anomaly detection with outlier exposure. arXiv preprint (2018). arXiv:1812.04606
Hendrycks, D., Mazeika, M., Kadavath, S., Song, D.: Using self-supervised learning can improve model robustness and uncertainty. In: Advances in Neural Information Processing Systems, pp. 15637–15648 (2019)
Lee, K.: A simple unified framework for detecting out-of-distribution samples and adversarial attacks (2019). https://github.com/pokaxpoka/deep_Mahalanobis_detector
Lee, K., Lee, K., Lee, H., Shin, J.: A simple unified framework for detecting out-of-distribution samples and adversarial attacks. In: Advances in Neural Information Processing Systems, pp. 7167–7177 (2018)
Liang, S., Li, Y., Srikant, R.: Enhancing the reliability of out-of-distribution image detection in neural networks. In: 6th International Conference on Learning Representations, ICLR 2018 (2018)
Liu, F.T., Ting, K.M., Zhou, Z.H.: Isolation forest. In: 2008 Eighth IEEE International Conference on Data Mining, pp. 413–422. IEEE (2008)
Lu, Y., Xu, P.: Anomaly detection for skin disease images using variational autoencoder. arXiv preprint (2018). arXiv:1807.01349
Marsland, S., Shapiro, J., Nehmzow, U.: A self-organising network that grows when required. Neural Netw. 15(8–9), 1041–1058 (2002)
Masana, M., Ruiz, I., Serrat, J., van de Weijer, J., Lopez, A.M.: Metric learning for novelty and anomaly detection. arXiv preprint (2018). arXiv:1808.05492
Oakley, A.: Dermnet new zealand (2016)
Ouardini, K., et al.: Towards practical unsupervised anomaly detection on retinal images. In: Wang, Q., et al. (eds.) DART/MIL3ID -2019. LNCS, vol. 11795, pp. 225–234. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-33391-1_26
Pacheco, A.G., Ali, A.R., Trappenberg, T.: Skin cancer detection based on deep learning and entropy to detect outlier samples. arXiv preprint (2019). arXiv:1909.04525
Ren, J., et al.: Likelihood ratios for out-of-distribution detection. In: Advances in Neural Information Processing Systems, pp. 14680–14691 (2019)
Ruff, L., et al.: Deep one-class classification. In: International conference on machine learning, pp. 4393–4402 (2018)
Tschandl, P., Rosendahl, C., Kittler, H.: The ham10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions. Sci. Data 5, 180161 (2018)
Vyas, A., Jammalamadaka, N., Zhu, X., Das, D., Kaul, B., Willke, T.L.: Out-of-distribution detection using an ensemble of self supervised leave-out classifiers. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 550–564 (2018)
Zhang, X., Wang, S., Liu, J., Tao, C.: Towards improving diagnosis of skin diseases by combining deep neural network and human knowledge. BMC Med. Inform. Decis. Making 18(2), 59 (2018)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Li, X., Lu, Y., Desrosiers, C., Liu, X. (2020). Out-of-Distribution Detection for Skin Lesion Images with Deep Isolation Forest. In: Liu, M., Yan, P., Lian, C., Cao, X. (eds) Machine Learning in Medical Imaging. MLMI 2020. Lecture Notes in Computer Science(), vol 12436. Springer, Cham. https://doi.org/10.1007/978-3-030-59861-7_10
Download citation
DOI: https://doi.org/10.1007/978-3-030-59861-7_10
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-59860-0
Online ISBN: 978-3-030-59861-7
eBook Packages: Computer ScienceComputer Science (R0)