DPAM-PSPNet: ultrasonic image segmentation of thyroid nodule based on dual-path attention mechanism

Phys Med Biol. 2023 Jul 31;68(16). doi: 10.1088/1361-6560/ace6f1.

Abstract

Objective.Deep learning has demonstrated its versatility in the medical field, particularly in medical image segmentation, image classification, and other forms of automated diagnostics. The clinical diagnosis of thyroid nodules requires radiologists to locate nodules, diagnose conditions based on nodule boundaries, textures and their experience. This task is labor-intensive and tiring; therefore, an automated system for accurate thyroid nodule segmentation is essential. In this study, a model named DPAM-PSPNet was proposed, which automatically segments nodules in thyroid ultrasound images and enables to segment malignant nodules precisely.Approach.In this paper, accurate segmentation of nodule edges is achieved by introducing the dual path attention mechanism (DPAM) in PSPNet. In one channel, it captures global information with a lightweight cross-channel interaction mechanism. In other channel, it focus on nodal margins and surrounding information through the residual bridge network. We also updated the integrated loss function to accommodate the DPAM-PSPNet.Main results.The DPAM-PSPNet was tested against the classical segmentation model. Ablation experiments were designed for the two-path attention mechanism and the new loss function, and generalization experiments were designed on the public dataset. Our experimental results demonstrate that DPAM-PSPNet outperforms other existing methods in various evaluation metrics. In the model comparison experiments, it achieved performance with an mIOU of 0.8675, mPA of 0.9357, mPrecision of 0.9202, and Dice coefficient of 0.9213.Significance.The DPAM-PSPNet model can segment thyroid nodules in ultrasound images with little training data and generate accurate boundary regions for these nodules.

Keywords: attention mechanism; computer-aided diagnosis; deep learning; thyroid nodules; ultrasound.