TransMixNet: an Attention Based Double-Branch Model for White Blood Cell Classification and Its Training with the Fuzzified Training Data

Hua Chen,Juan Liu,Chunbing Hua,Zhiqun Zuo,Jing Feng,Baochuan Pang,Di Xiao
DOI: https://doi.org/10.1109/bibm52615.2021.9669587
2021-01-01
Abstract:White blood cells (WBCs) play a critical part in the human immune system, protecting human from bacteria and viruses. In general, WBCs can be divided into five categories: basophil, monocyte, eosinophil, neutrophil, and lymphocyte. The proportion of different kinds of WBCs is closely related to human health, thus analyzing the numbers and percentages of WBCs is usually included in the blood routine examination. Therefore, accurate identification of different types of WBCs has become an important step in the statistical analysis. However, the insufficiency and imbalance of WBC samples in medical computer vision domains is still a challenge to classify WBCs intelligently and accurately. This paper presents an attention based double-branch network, called TransMixNet, to build the classification model for WBCs recognition. During the model construction, we adopt transfer learning and data augmentation strategies to overcome the problems of insufficient and unbalanced samples. In addition, mixup strategy is used to model the domain relationship between different types of training samples to improve the generalization performance of the model. The evaluation experiment results show that our model is promising and applicable to clinical applications.
What problem does this paper attempt to address?