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

Qinghua Huang, Liping Ye

科研成果: 期刊稿件文章同行评审

25 引用 (Scopus)

摘要

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.

源语言英语
页(从-至)691-701
页数11
期刊IEEE Transactions on Ultrasonics, Ferroelectrics, and Frequency Control
69
2
DOI
出版状态已出版 - 1 2月 2022

指纹

探究 'Multi-Task/Single-Task Joint Learning of Ultrasound BI-RADS Features' 的科研主题。它们共同构成独一无二的指纹。

引用此