DWS-YOLO: A Lightweight Detector for Blood Cell Detection

Yihai Mao,Hongyi Zhang,Wanqing Wu,Xingen Gao,Zhibin Lin,Juqiang Lin
DOI: https://doi.org/10.1080/08839514.2024.2318673
IF: 2.777
2024-02-24
Applied Artificial Intelligence
Abstract:Peripheral blood cell detection is an essential component of medical practice and is used to diagnose and treat diseases, as well as to monitor the progress of therapies. Our objective is to construct an efficient deep learning model for peripheral blood cell analysis that achieves an optimized balance between inference speed, computational complexity, and detection accuracy. In this article, we propose the DWS-YOLO blood detector, which is a lightweight blood detector. Our model includes several improved modules, including the lightweight C3 module, the increased combined attention mechanism, the Scylla-IoU loss function, and the improved soft non-maximum suppression. Improved attention, loss function, and suppression enhance detection accuracy, while lightweight C3 module reduces computation time. The experiment results demonstrate that our proposed modules can enhance a detector's detection performance, and obtain new state-of-the-art (SOTA) results and excellent robustness performance on the BCCD dataset. On the white blood cell detection dataset (Raabin-WBC), the proposed detector's generalization performance was confirmed to be satisfactory. Our proposed blood detector achieves high detection accuracy while requiring few computational resources and is very suitable for resource-limited but efficient medical device environments, providing a reliable and advanced solution for blood detection that greatly improves the efficiency and effectiveness of peripheral blood cell analysis in clinical practice.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?
This paper aims to address several key issues in blood cell detection, particularly seeking the optimal balance in improving detection accuracy, reducing computational complexity, and accelerating inference speed. Specifically, the paper proposes a lightweight blood cell detector, DWS-YOLO, with the main objectives including: 1. **Improving detection accuracy**: By introducing the Enhanced Composite Attention Mechanism (ECAM), it effectively captures sensitive information, thereby improving the accuracy of blood cell detection. 2. **Reducing computational complexity**: Designed depthwise separable convolutions (DSC) and inverted residual blocks to reduce the number of parameters in the C3 module, making the model more lightweight. 3. **Improving the loss function**: Adopted the Scylla Intersection over Union (SIoU) loss to enhance the model's convergence speed and regression accuracy. 4. **Optimizing non-maximum suppression techniques**: Introduced the improved Soft Non-Maximum Suppression (Im-Soft-NMS) technique to enhance detection accuracy. In summary, this research aims to develop a blood cell detection method suitable for resource-limited but highly efficient medical device environments, significantly improving the efficiency and effectiveness of peripheral blood cell analysis in clinical practice.