Bone Marrow Cell Recognition: Training Deep Object Detection with A New Loss Function

TL;DR

This paper introduces a new loss function for deep learning-based bone marrow cell detection, improving accuracy by accounting for class similarities, and demonstrates superior performance over existing methods.

Contribution

It proposes a novel loss function tailored for bone marrow cell classification that considers class similarity, enhancing detection accuracy in dense and diverse cell environments.

Findings

Improved detection accuracy over existing algorithms.

Effective handling of similar and dissimilar cell classes.

Enhanced robustness in dense cell distributions.

Abstract

For a long time, bone marrow cell morphology examination has been an essential tool for diagnosing blood diseases. However, it is still mainly dependent on the subjective diagnosis of experienced doctors, and there is no objective quantitative standard. Therefore, it is crucial to study a robust bone marrow cell detection algorithm for a quantitative automatic analysis system. Currently, due to the dense distribution of cells in the bone marrow smear and the diverse cell classes, the detection of bone marrow cells is difficult. The existing bone marrow cell detection algorithms are still insufficient for the automatic analysis system of bone marrow smears. This paper proposes a bone marrow cell detection algorithm based on the YOLOv5 network, trained by minimizing a novel loss function. The classification method of bone marrow cell detection tasks is the basis of the proposed novel loss function. Since bone marrow cells are classified according to series and stages, part of the classes in adjacent stages are similar. The proposed novel loss function considers the similarity between bone marrow cell classes, increases the penalty for prediction errors between dissimilar classes, and reduces the penalty for prediction errors between similar classes. The results show that the proposed loss function effectively improves the algorithm's performance, and the proposed bone marrow cell detection algorithm has achieved better performance than other cell detection algorithms.