Dual-branch network via pseudo-label training for thyroid nodule detection in ultrasound image

Ruoning Song,Chuang Zhu,Long Zhang,Tong Zhang,Yihao Luo,Jun Liu,Jie Yang
DOI: https://doi.org/10.1007/s10489-021-02967-2
IF: 5.3
2022-01-29
Applied Intelligence
Abstract:Automated nodule detection in the ultrasound image is essential for computer-aided thyroid tumor diagnosis. However, in the ultrasound image, the solid nodule has imaging characteristics similar to other tissues, making it challenging to detect such nodules. Therefore, we proposed a feature-enhanced dual-branch network (FDnet) to complete the nodule detection task by adding a semantic segmentation branch and a feature enhancement mechanism into the detection network. This design improves the target area's proposal score and suppresses the interference of similar tissues, which can reduce the false-positive rate of the proposed bounding box and finally obtain more reliable detection results. Additionally, to solve the lack of fine-grained mask information for semantic segmentation branch training in the actual scenario, we also proposed an iterative training strategy that combines the ground-truth boundary box with the branch results to generate a pseudo-label mask. Finally, we carried out various comparative experiments to verify the feasibility of the proposed network and training strategy. A series of experiments showed that FDnet could achieve competitive detection performance (mAP: 61.8/92.5/65.9), which metrics are better than the state-of-the-art detection methods. Besides, the performance using the pseudo-label mask training is also close to using the ground-truth mask in the public dataset, and the inference speed per image is also comparable to that of other networks both in the two datasets. This result shows that our method can improve the efficiency of thyroid nodule detection without fine-grained annotation, and the output result of the trained semantic segmentation branch can guide the further segmentation of nodule edge, which has practical clinical significance. We will release the source code and the public dataset at https://github.com/songruoning/Thyroid_Solid_Nodule.
computer science, artificial intelligence
What problem does this paper attempt to address?