Classification of white blood cells using capsule networks

• Published in: Computerized medical imaging and graphics 2020-03, Vol.80, p.101699-101699, Article 101699
• Main Authors: Baydilli, Yusuf Yargı, Atila, Ümit
• Format: Article
• Language: English
• Subjects: Artificial neural networks; Capsule networks; Case studies; Classification; Data analysis; Data processing; Datasets; Deep learning; Diagnostic software; Diagnostic systems; Image acquisition; Learning algorithms; Leukocytes; Machine learning; Medical image analysis; Medical imaging; Medical research; Neural networks; Researchers; Transfer learning; White blood cells (WBC)
• ISSN: 0895-6111; 1879-0771
• DOI: 10.1016/j.compmedimag.2020.101699

[Display omitted] •Success of deep learning methods is directly proportional to the data set size.•This is a major obstacle for researchers who analyze medical data with deep learning.•In this study, a modified Capsule Network was employed on a very small WBC data set.•The proposed model was achieved 96.86% accuracy rate for five classes. While the number and structural features of white blood cells (WBC) can provide important information about the health status of human beings, the ratio of sub-types of these cells and the deformations that can be observed serve as a good indicator in the diagnosis process of some diseases. Hence, correct identification and classification of the WBC types is of great importance. In addition, the fact that the diagnostic process that is carried out manually is slow, and the success is directly proportional to the expert's skills makes this problem an excellent field of application for computer-aided diagnostic systems. Unfortunately, both the ethical reasons and the cost of image acquisition process is one of the biggest obstacles to the fact that researchers working with medical images are able to collect enough data to produce a stable model. For that reasons, researchers who want to perform a successful analysis with small data sets using classical machine learning methods need to undergo their data a long and error-prone pre-process, while those using deep learning methods need to increase the data size using augmentation techniques. As a result, there is a need for a model that does not need pre-processing and can perform a successful classification in small data sets. WBCs were classified under five categories using a small data set via capsule networks, a new deep learning method. We improved the model using many techniques and compared the results with the most known deep learning methods. Both the above-mentioned problems were overcame and higher success rates were obtained compared to other deep learning models. While, convolutional neural networks (CNN) and transfer learning (TL) models suffered from over-fitting, capsule networks learned well training data and achieved a high accuracy on test data (96.86%). In this study, we briefly discussed the abilities of capsule networks in a case study. We showed that capsule networks are a quite successful alternative for deep learning and medical data analysis when the sample size is limited.