English

A Deep Learning Framework for Thyroid Nodule Segmentation and Malignancy Classification from Ultrasound Images

Image and Video Processing 2025-11-18 v1 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Ultrasound-based risk stratification of thyroid nodules is a critical clinical task, but it suffers from high inter-observer variability. While many deep learning (DL) models function as "black boxes," we propose a fully automated, two-stage framework for interpretable malignancy prediction. Our method achieves interpretability by forcing the model to focus only on clinically relevant regions. First, a TransUNet model automatically segments the thyroid nodule. The resulting mask is then used to create a region of interest around the nodule, and this localised image is fed directly into a ResNet-18 classifier. We evaluated our framework using 5-fold cross-validation on a clinical dataset of 349 images, where it achieved a high F1-score of 0.852 for predicting malignancy. To validate its performance, we compared it against a strong baseline using a Random Forest classifier with hand-crafted morphological features, which achieved an F1-score of 0.829. The superior performance of our DL framework suggests that the implicit visual features learned from the localised nodule are more predictive than explicit shape features alone. This is the first fully automated end-to-end pipeline for both detecting thyroid nodules on ultrasound images and predicting their malignancy.

Keywords

Cite

@article{arxiv.2511.11937,
  title  = {A Deep Learning Framework for Thyroid Nodule Segmentation and Malignancy Classification from Ultrasound Images},
  author = {Omar Abdelrazik and Mohamed Elsayed and Noorul Wahab and Nasir Rajpoot and Adam Shephard},
  journal= {arXiv preprint arXiv:2511.11937},
  year   = {2025}
}

Comments

5 pages, 2 figures, 2 tables

R2 v1 2026-07-01T07:38:33.350Z