Multi-Task/Single-Task Joint Learning of Ultrasound BI-RADS Features

Qinghua Huang, Liping Ye

Research output: Contribution to journalArticlepeer-review

25 Scopus citations

Abstract

Breast cancer is one of the most common cancers among women in the world. The breast imaging reporting and data system (BI-RADS) features effectively improve the accuracy and sensitivity of breast tumors. Based on the description of signs in BI-RADS, a quantitative scoring scheme is proposed based on ultrasound (US) data. This scheme includes feature extraction of high-level semantic, that is, an intermediate step interpreting the subsequent diagnosis. However, the scheme requires doctors to score the features of breast data, which is labor-intensive. To reduce the burden of doctors, we design a multi-task learning (MTL) framework, which can directly output the scores of different BI-RADS features from the raw US images. The MTL framework consists of a shared network that learns global features and <inline-formula> <tex-math notation="LaTeX">{K} </tex-math></inline-formula> soft attention networks for different BI-RADS features. Thus, it enables the network not only to learn the potential correlation among different BI-RADS features, but also learn the unique specificity of each feature, which can assist each other and jointly improve the scoring accuracy. In addition, we group different BI-RADS features according to the correlation among tasks and build a multi-task/single-task joint framework. Experimental results on the US breast tumor dataset collected from 1859 patients with 4458 US images show that the proposed BI-RADS feature scoring framework achieves an average scoring accuracy of 84.91% for 11 BI-RADS features on the test dataset, which is helpful for the subsequent diagnosis of breast tumors.

Original languageEnglish
Pages (from-to)691-701
Number of pages11
JournalIEEE Transactions on Ultrasonics, Ferroelectrics, and Frequency Control
Volume69
Issue number2
DOIs
StatePublished - 1 Feb 2022

Keywords

  • Breast cancer
  • Breast tumors
  • Feature extraction
  • Malignant tumors
  • Medical services
  • Multitasking
  • Tumors

Fingerprint

Dive into the research topics of 'Multi-Task/Single-Task Joint Learning of Ultrasound BI-RADS Features'. Together they form a unique fingerprint.

Cite this