Breast Tumor Classification using Short-ResNet with Pixel-based Tumor Probability Map in Ultrasound Images.

Journal: Ultrasonic imaging
Published Date:

Abstract

Breast cancer is the most common form of cancer and is still the second leading cause of death for women in the world. Early detection and treatment of breast cancer can reduce mortality rates. Breast ultrasound is always used to detect and diagnose breast cancer. The accurate breast segmentation and diagnosis as benign or malignant is still a challenging task in the ultrasound image. In this paper, we proposed a classification model as short-ResNet with DC-UNet to solve the segmentation and diagnosis challenge to find the tumor and classify benign or malignant with breast ultrasonic images. The proposed model has a dice coefficient of 83% for segmentation and achieves an accuracy of 90% for classification with breast tumors. In the experiment, we have compared with segmentation task and classification result in different datasets to prove that the proposed model is more general and demonstrates better results. The deep learning model using short-ResNet to classify tumor whether benign or malignant, that combine DC-UNet of segmentation task to assist in improving the classification results.

Authors

  • You-Wei Wang
    Department of Computer Science and Information Engineering, National Taiwan University, Taipei, Taiwan.
  • Tsung-Ter Kuo
    Department of Medical Imaging and Radiological Technology, Yuanpei University of Medical Technology, Hsinchu, Taiwan.
  • Yi-Hong Chou
    Department of Radiology, Taipei Veterans General Hospital and National Yang Ming University, Taipei 112, Taiwan.
  • Yu Su
    Honghui Hospital, Xi'an Jiaotong University, Xi'an, China.
  • Shing-Hwa Huang
    Department of Breast Surgery, En Chu Kong Hospital, New Taipei City, Taiwan.
  • Chii-Jen Chen
    Department of Medical Imaging and Radiological Technology, Yuanpei University of Medical Technology, Hsinchu, Taiwan. Electronic address: cjchen@mail.ypu.edu.tw.