Boundary regression-based reep neural network for thyroid nodule segmentation in ultrasound images

Zhihao Jin, Xuechen Li, Yudi Zhang, Lin Lin Shen, Zhihui Lai, Heng Kong

Research output: Journal PublicationArticlepeer-review

Abstract

Due to the noises captured in ultrasound device and image reconstruction process, the edges of thyroid nodule are usually not distinctive and it is very difficult for existing approaches to well segment them in ultrasound images. While deep neural networks like U-Net have been successfully applied in many medical image segmentation tasks, their segmentation performances on ultrasound images are still not satisfactory. To address this issue, we propose in this paper a boundary field regression branch to provide useful boundary information to help improve the segmentation performance of existing networks. Without requirement of additional labeling costs, our approach firstly generates boundary field heatmap from available segmentation masks, which are then used as a supervision to train the regression branch. As a general architecture, our branch can be integrated with all encoder-decoder like segmentation networks. A dataset consisting of 3169 images from 2004 patients is used for experiments. We integrate our branch with U-Net, Attention U-Net, U-Net++ and DeepLabv3+; consistent improvements of Dice metrics were observed. The memory and computation costs required by adding our branch are marginal as well.

Original languageEnglish
Pages (from-to)22357-22366
Number of pages10
JournalNeural Computing and Applications
Volume34
Issue number24
DOIs
Publication statusPublished - Dec 2022
Externally publishedYes

Keywords

  • Boundary
  • Segmentation
  • Thyroid nodule

ASJC Scopus subject areas

  • Software
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Boundary regression-based reep neural network for thyroid nodule segmentation in ultrasound images'. Together they form a unique fingerprint.

Cite this