Tumor Detection Based on Deep Mutual Learning in Automated Breast Ultrasound

Yanfeng Li,Zilu Zhang,Jia Sun,Houjin Chen,Ziwei Chen,Jiayu Wei
DOI: https://doi.org/10.1007/s11042-024-18377-8
IF: 2.577
2024-01-01
Multimedia Tools and Applications
Abstract:Tumor detection in automated breast ultrasound (ABUS) images is critical for computer-aided diagnosis. A single deep learning based network is prone to make overconfident prediction. In this paper, a DML-YOLOX model based on deep mutual learning (DML) is proposed. To alleviate the overconfidence of a single network, the dual-model collaborative training strategy is designed based on YOLOX baseline. To implement the interaction between the dual models, the exploration loss is developed and combined with the consistency loss for supervision. The exploration loss aims at encouraging the models to explore and learn different feature representations in the feature extraction stage. The consistency loss aims at constraining the models to have consistent output representations for the same input. According to the ellipse-like characteristic of ABUS tumors, a rotation augmentation method is designed, which can decrease the overestimate of the bounding box. Besides, in order to reduce the regression error caused by large angle rotation, a rotation discriminant (RD) measurement is developed. The proposed method is verified on two ABUS datasets, one of which is the private ABUS dataset with 68 normal volumes and 68 tumor volumes, including 43,248 slices, and the other is the public ABUS dataset. On the private ABUS dataset, it achieves a promising detection result with a sensitivity of 0.90 and the false positives per slice (FPs/S) at 0.15.
What problem does this paper attempt to address?