Automated detection and recognition of thyroid nodules in ultrasound images using Improve Cascade Mask R-CNN

Yinghao Zheng,Lina Qin,Taorong Qiu,Aiyun Zhou,Pan Xu,Zhixin Xue
DOI: https://doi.org/10.1007/s11042-021-10939-4
IF: 2.577
2021-05-01
Multimedia Tools and Applications
Abstract:Accurate diagnosis of thyroid nodules using ultrasonography heavily relies on the superb skills and rich experience of senior radiologists, considering the low contrast, high noise of the ultrasound image, and the diverse appearance of the nodules. Computer-aided diagnosis systems could diagnose thyroid nodules based on ultrasound characteristics to assist radiologists. However, the existing learning-based approaches for detecting and recognizing thyroid nodules have the problems of inaccurate localization and low recognition accuracy. In this study, we propose an Improved Cascade Mask R-CNN for effectively detecting and recognizing thyroid nodules. Firstly, a more effective detector is designed to better classify the ROIs and better correct the bounding boxes. Secondly, a more effective balanced L1 loss function is used to increase the gradient of the easy sample and solve the problem of imbalance between hard samples and easy samples during training. Finally, a more effective soft non-maximum suppression (Soft-NMS) method is used to set an attenuation function for adjacent bounding boxes, which solves the problem of possible missing detection in non-maximum suppression (NMS). The improved model is trained and verified by using real 1408 images collected from the known hospital. Under the localization accuracy of the IoU threshold of 0.5, the mAP reaches 87.1%, and the recognition accuracy reaches 98.67%. The experiment results show that the improved model is effective and highly valuable to help the doctors for the recognition of benign and malignant thyroid nodules.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?