Thyroid Nodule Segmentation in Ultrasound Images with Conformal Uncertainty Estimation
Organizations: Department of Statistical Sciences, Sapienza University of Rome, Rome, Italy.
Abstract
Thyroid nodules affect a large proportion of the general population. Although most are benign, a non-negligible fraction is malignant. Ultrasound imaging is the primary diagnostic modality, and accurate nodule segmentation is a prerequisite for CAD systems to classify malignancy and assess metastatic risk.\ In this work, we propose a thyroid nodule segmentation framework that integrates a DeepLabV3+, channel-attention modules and post-hoc conformal prediction, providing both accurate mask and pixel-level uncertainty estimates. We address thyroid ultrasound nodule segmentation as the first stage of a pipeline that also includes malignancy classification and metastatic risk prediction. We adopt DeepLabV3+, whose Atrous Spatial Pyramid Pooling (ASPP) captures multi-scale context, with an EfficientNet-B5 backbone for its accuracy-efficiency trade-off. Additionally, we test three channel-attention modules: Squeeze-and-Excitation, CBAM and Grouped Channel Attention (GCA), paired with two loss formulations (Dice-BCE and Boundary).\ Since binary masks do not indicate prediction confidence, we wrap the trained model with a split conformal prediction layer that produces pixel-level uncertainty maps with statistical coverage guarantees.\ We evaluate all configurations on the TN3K public benchmark using a fixed split and directly compare them with state-of-the-art methods. The plain backbone already improves upon previously reported results, with CBAM + Dice-BCE emerging as the best overall configuration.
Figures & tables
| Attention Mechanism | Loss Function | Dice | MCC | ||||
|---|---|---|---|---|---|---|---|
| Dice g | mDice | mDice 50 | mDice 10 | MCC g | mMCC | ||
| None | Dice-BCE | 88.76 | 83.43 | 89.70 | 63.83 | 0.872 | 0.829 |
| Boundary | 89.46 | 84.72 | 91.19 | 65.80 | 0.880 | 0.843 | |
| GCA | Dice-BCE | 89.50 | 84.66 | 90.54 | 64.32 | 0.881 | 0.842 |
| Boundary | 89.43 | 84.48 | 90.70 | 64.87 | 0.880 | 0.840 | |
| SE | Dice-BCE | 89.18 | 84.42 | 90.42 | 64.51 | 0.877 | 0.839 |
| Attention Mechanism | Loss Function | DR by size | DR by count | |||
|---|---|---|---|---|---|---|
| DR s | DR m | DR l | DR 1 | DR M | ||
| None | Dice-BCE | 0.861 | 0.919 | 0.970 | 0.961 | 0.726 |
| Boundary | 0.891 | 0.946 | 0.980 | 0.971 | 0.801 | |
| GCA | Dice-BCE | 0.883 | 0.900 | 0.985 | 0.958 | 0.774 |
| Boundary | 0.883 | 0.923 | 0.990 | 0.971 | 0.767 | |
| SE | Dice-BCE | 0.898 | 0.928 | 0.980 | 0.963 | 0.815 |