Exploring Convolutional Neural Networks for the Classification of Acute Lymphoblast Leukemia Blood Cell Images
A. A. Trubnikov, Dmitry Savelyev · Journal of Biomedical Photonics & Engineering · 2024
This paper introduces a novel approach to blood cell classification using convolutional neural networks (CNNs). Our emphasis lies in methodological insights derived from comparative analyses across various CNN architectures, implemented in Python with PyTorch such as ResNet152V2, Xception, EfficientNetB5, and EfficientNetV2M. In exploring model capabilities and limitations, we observe that the simple and shallow architectures is not sufficient to learn patterns compared to more complex networks. During research we have discovered that EfficientNetV2M demonstrate stable results with mean F 1 =0.891 score, which is higher compare to other models. Saliency maps are applied to reveal significant regions within or near cells, offering nuanced insights into morphological influences like cell shape. Extensive data augmentation allows one effectively mitigate overfitting, aligning train learning curves with validation and test splits. Our study is methodologically rich, comprising a meticulous data-specific overview, a robustness analysis through multiple model training, a systematic architecture comparison, and an in-depth examination using saliency maps.