A comparison of U-net backbone architectures for the automatic white blood cells segmentation
Keywords:white blood cells segmentation, deep Learning, transfer learning, U-NET, Loss Function, cytological image’s dataset
Reliable recognition of white blood cells is an essential step in the diagnosis of several types of cancer. Therefore, the segmentation of white blood cells plays an essential role and is an important part of the medical diagnostic system. Manual cell diagnosis involves doctors visually examining microscopic images to detect any cellular abnormalities. This step is costly and time-consuming. An automated system based on white blood cell identification provides a more accurate result than the manual method. Image segmentation is one of the crucial contributions of a deep learning community to the medical field. In this paper, we demonstrate how the U-Net type architecture can be improved by the use of the pre-trained encoder, a comparison of several efficient methods for automatic recognition of white blood cells using the original U-NET, different pre-trained classification networks are used as the backbone to obtain better performance. The architecture of RESNET-50 obtains the best segmentation results on testing data for automatic recognition in cytological images with a less amount of training epochs.
Al-Dulaimi, K. A. K., Banks, J., Chandran, V., Tomeo-Reyes, I., & Nguyen Thanh, K. (2018). Classification of white blood cell types from microscope images: Techniques and challenges. Microscopy science: Last approaches on educational programs and applied research (Microscopy Book Series, 8), 17-25.
Blumenreich, M. S. (1990). The white blood cell and differential count. Clinical Methods: The History, Physical, and Laboratory Examinations. 3rd edition..
Yu, T. C., Chou, W. C., Yeh, C. Y., Yang, C. K., Huang, S. C., Tien, F. M., ... & Chou, S. C. (2019). Automatic Bone Marrow Cell Identification and Classification By Deep Neural Network.
Ramakrishna, R. R., Abd Hamid, Z., Zaki, W. M. D. W., Huddin, A. B., & Mathialagan, R. (2020). Stem cell imaging through convolutional neural networks: current issues and future directions in artificial intelligence technology. PeerJ, 8, e10346.
Anilkumar, K. K., Manoj, V. J., & Sagi, T. M. (2020). A survey on image segmentation of blood and bone marrow smear images with emphasis to automated detection of Leukemia. Biocybernetics and Biomedical Engineering.
Benazzouz, M., Baghli, I., & Chikh, M. A. (2013). Microscopic image segmentation based on pixel classification and dimensionality reduction. International journal of imaging systems and technology, 23(1), 22-28.
Baghli, I., Nakib, A., Sellam, E., Benazzouz, M., Chikh, A., & Petit, E. (2014, March). Hybrid framework based on evidence theory for blood cell image segmentation. In Medical Imaging 2014: Biomedical Applications in Molecular, Structural, and Functional Imaging (Vol. 9038, p. 903815). International Society for Optics and Photonics.
Benomar, M. L., Chikh, A., Descombes, X., & Benazzouz, M. (2021). Multi-feature-based approach for white blood cells segmentation and classification in peripheral blood and bone marrow images. International Journal of Biomedical Engineering and Technology, 35(3), 223-241.
Settouti, N., Bechar, M. E. A., Daho, M. E. H., & Chikh, M. A. (2020). An optimised pixel-based classification approach for automatic white blood cells segmentation. International Journal of Biomedical Engineering and Technology, 32(2), 144-160.
Saidi, M., Bechar, M. E. A., Settouti, N., & Chikh, M. A. (2018). Instances selection algorithm by ensemble margin. Journal of Experimental & Theoretical Artificial Intelligence, 30(3), 457-478.
Settouti, N., Saidi, M., Bechar, M. E. A., Daho, M. E. H., & Chikh, M. A. (2020). An instance and variable selection approach in pixel-based classification for automatic white blood cells segmentation. Pattern Analysis and Applications, 1-18.
Bechar, M. E. A., Settouti, N., Daho, M. E. H., Adel, M., & Chikh, M. A. (2019). Influence of normalization and color features on super-pixel classification: application to cytological image segmentation. Australasian physical & engineering sciences in medicine, 42(2), 427-441.
Bechar, M. E. A., Settouti, N., Daho, M. E. H., & CHIKH, M. A. (2018, October). Semi-supervised Super-pixels classification for White Blood Cells segmentation. In 2018 3rd International Conference on Pattern Analysis and Intelligent Systems (PAIS) (pp. 1-8). IEEE.
Khouani, A., Daho, M. E. H., Mahmoudi, S. A., Chikh, M. A., & Benzineb, B. (2020). Automated recognition of white blood cells using deep learning. Biomedical Engineering Letters, 10(3), 359-367.
Ronneberger, O., Fischer, P., & Brox, T. (2015, October). U-net: Convolutional networks for biomedical image segmentation. In International Conference on Medical image computing and computer-assisted intervention (pp. 234-241). Springer, Cham.
Ioffe, S., & Szegedy, C. (2015, June). Batch normalization: Accelerating deep network training by reducing internal covariate shift. In International conference on machine learning (pp. 448-456). PMLR.
Iglovikov, V., & Shvets, A. (2018). Ternausnet: U-net with vgg11 encoder pre-trained on imagenet for image segmentation. arXiv preprint arXiv:1801.05746.
Chang, S. W., & Liao, S. W. (2019, October). KUnet: microscopy image segmentation with deep unet based convolutional networks. In 2019 IEEE International Conference on Systems, Man and Cybernetics (SMC) (pp. 3561-3566). IEEE.
Howard, A. G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., ... & Adam, H. (2017). Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv preprint arXiv:1704.04861.
Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556.
He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778).
Russakovsky, O., Deng, J., Su, H., Krause, J., Satheesh, S., Ma, S., ... & Fei-Fei, L. (2015). Imagenet large scale visual recognition challenge. International journal of computer vision, 115(3), 211-252.
Yakubovskiy, P. Segmentation Models. (2019). https://github.com/qubvel/ segmentation_models.
Prechelt, L. (1998). Early stopping-but when?. In Neural Networks: Tricks of the trade (pp. 55-69). Springer, Berlin, Heidelberg.
Kingma, D. P., & Ba, J. (2014). Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980.
Yi-de, M., Qing, L., & Zhi-Bai, Q. (2004, October). Automated image segmentation using improved PCNN model based on cross-entropy. In Proceedings of 2004 International Symposium on Intelligent Multimedia, Video and Speech Processing, 2004. (pp. 743-746). IEEE.
Sudre, C. H., Li, W., Vercauteren, T., Ourselin, S., & Cardoso, M. J. (2017). Generalised dice overlap as a deep learning loss function for highly unbalanced segmentations. In Deep learning in medical image analysis and multimodal learning for clinical decision support (pp. 240-248). Springer, Cham.
How to Cite
Copyright (c) 2021 WAS Science Nature (WASSN) ISSN: 2766-7715
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.