IoT Application of Transfer Learning in Hybrid Artificial Intelligence Systems for Acute Lymphoblastic Leukemia Classification

Sensors (Basel). 2021 Dec 1;21(23):8025. doi: 10.3390/s21238025.

Abstract

Acute lymphoblastic leukemia is the most common cancer in children, and its diagnosis mainly includes microscopic blood tests of the bone marrow. Therefore, there is a need for a correct classification of white blood cells. The approach developed in this article is based on an optimized and small IoT-friendly neural network architecture. The application of learning transfer in hybrid artificial intelligence systems is offered. The hybrid system consisted of a MobileNet v2 encoder pre-trained on the ImageNet dataset and machine learning algorithms performing the role of the head. These were the XGBoost, Random Forest, and Decision Tree algorithms. In this work, the average accuracy was over 90%, reaching 97.4%. This work proves that using hybrid artificial intelligence systems for tasks with a low computational complexity of the processing units demonstrates a high classification accuracy. The methods used in this study, confirmed by the promising results, can be an effective tool in diagnosing other blood diseases, facilitating the work of a network of medical institutions to carry out the correct treatment schedule.

Keywords: ALL-IDB database; IoT; MobileNet v2; hybrid artificial intelligence system; leukemia; low-resource dataset; lymphocyte cells.

MeSH terms

  • Algorithms
  • Artificial Intelligence*
  • Humans
  • Machine Learning
  • Neural Networks, Computer
  • Precursor Cell Lymphoblastic Leukemia-Lymphoma* / diagnosis