A Feedback Attention Network for Classification and Visualization of Thyroid Nodules on Ultrasound Images.

Jianrong Wang,Ruixuan Zhang,Xi Wei,Xuewei Li,Mei Yu,Jian Yu,Jialin Zhu,Jie Gao,Hongqian Shen,Zhiqiang Liu,Ruiguo Yu
2019-01-01
Abstract:With the maturity of AI technology, computer-aided diagnosis technology is gradually developing, even surpassing the performance of doctors. However, due to the lack of interpretability of deep learning, we can’t understand what the model has learned, and even determine whether the diagnostic model is credible. In this paper, Feedback Attention Network(FAN) is proposed, which visualizes the lesion area that plays a decisive role in precise prediction while classifying benign and malignant thyroid ultrasound images. The degree of fineness is adjustable, which is more refined than existing visualization methods. In addition, this paper uses the attention to make the model more sensitive to the lesion area and inhibit the influence of the non-lesion area. The experimental results show that the accuracy of benign and malignant classification reaches 90.76%±2.8%. Compared with the state-of-the-art classification networks, FAN has higher accuracy and fewer parameters. Above all, the visualization results show that FAN does learn the true nodule features, especially those at the edge of nodules, which proves FAN a credible aided diagnosis model.
What problem does this paper attempt to address?