AnoMalNet: Outlier Detection based Malaria Cell Image Classification Method Leveraging Deep Autoencoder

Class imbalance is a pervasive issue in the field of disease classification from medical images. It is necessary to balance out the class distribution while training a model for decent results. However, in the case of rare medical diseases, images from affected patients are much harder to come by compared to images from non-affected patients, resulting in unwanted class imbalance. Various processes of tackling class imbalance issues have been explored so far, each having its fair share of drawbacks. In this research, we propose an outlier detection based binary medical image classification technique which can handle even the most extreme case of class imbalance. We have utilized a dataset of malaria parasitized and uninfected cells. An autoencoder model titled AnoMalNet is trained with only the uninfected cell images at the beginning and then used to classify both the affected and non-affected cell images by thresholding a loss value. We have achieved an accuracy, precision, recall, and F1 score of 98.49%, 97.07%, 100%, and 98.52% respectively, performing better than large deep learning models and other published works. As our proposed approach can provide competitive results without needing the disease-positive samples during training, it should prove to be useful in binary disease classification on imbalanced datasets.

[1]  Swati Lipsa,et al.  MalNet – an Optimized CNN based method for Malaria Diagnosis , 2022, 2022 2nd International Conference on Intelligent Technologies (CONIT).

[2]  A. Nahid,et al.  Malaria cell image classification by explainable artificial intelligence , 2021, Health and Technology.

[3]  H. A. Nugroho,et al.  GGB Color Normalization and Faster-RCNN Techniques for Malaria Parasite Detection , 2021, 2021 IEEE International Biomedical Instrumentation and Technology Conference (IBITeC).

[4]  Siak Wang Khor,et al.  Malaria Parasite Detection using Residual Attention U-Net , 2021, International Conference on Signal and Image Processing Applications.

[5]  Matt Gardner,et al.  QA Dataset Explosion: A Taxonomy of NLP Resources for Question Answering and Reading Comprehension , 2021, ACM Comput. Surv..

[6]  Alice Othmani,et al.  Leveraging Recent Advances in Deep Learning for Audio-Visual Emotion Recognition , 2021, Pattern Recognit. Lett..

[7]  Hao Guan,et al.  Domain Adaptation for Medical Image Analysis: A Survey , 2021, IEEE Transactions on Biomedical Engineering.

[8]  Wookey Lee,et al.  Biosignal Sensors and Deep Learning-Based Speech Recognition: A Review , 2021, Sensors.

[9]  Xiangbin Liu,et al.  A Review of Deep-Learning-Based Medical Image Segmentation Methods , 2021, Sustainability.

[10]  Z. Tan,et al.  An Overview of Deep-Learning-Based Audio-Visual Speech Enhancement and Separation , 2020, IEEE/ACM Transactions on Audio, Speech, and Language Processing.

[11]  Quan Quan,et al.  An Effective Convolutional Neural Network for Classifying Red Blood Cells in Malaria Diseases , 2020, Interdisciplinary Sciences: Computational Life Sciences.

[12]  Russell C. Hardie,et al.  Performance analysis of machine learning and deep learning architectures for malaria detection on cell images , 2019, Optical Engineering + Applications.

[13]  Taghi M. Khoshgoftaar,et al.  A survey on Image Data Augmentation for Deep Learning , 2019, Journal of Big Data.

[14]  D. Juliet,et al.  Transfer Learning with ResNet-50 for Malaria Cell-Image Classification , 2019, 2019 International Conference on Communication and Signal Processing (ICCSP).

[15]  George R. Thoma,et al.  Pre-trained convolutional neural networks as feature extractors toward improved malaria parasite detection in thin blood smear images , 2018, PeerJ.

[16]  Constantine Bekas,et al.  BAGAN: Data Augmentation with Balancing GAN , 2018, ArXiv.

[17]  Andrew G. Howard,et al.  Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation , 2018, ArXiv.

[18]  Scott Lundberg,et al.  A Unified Approach to Interpreting Model Predictions , 2017, NIPS.

[19]  Madhu S. Nair,et al.  Malaria Parasite Detection From Peripheral Blood Smear Images Using Deep Belief Networks , 2017, IEEE Access.

[20]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[21]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[22]  Marouane Birjali,et al.  A comprehensive survey on sentiment analysis: Approaches, challenges and trends , 2021, Knowl. Based Syst..

[23]  Mahendra Kumar Gourisaria,et al.  A Deep Learning Model for Malaria Disease Detection and Analysis using Deep Convolutional Neural Networks , 2020 .