Attention-based image segmentation and classification model for the preoperative risk stratification of thyroid nodules
- Author(s)
- Jassal, K; Di Muzio, B; Edwards, M; Brown, W; Serpell, J; Koohestani, A; Lee, JC;
- Details
- Publication Year 2025-04,Volume 49,Issue #4,Page 1024-1033
- Journal Title
- World Journal of Surgery
- Publication Type
- Research article
- Abstract
- BACKGROUND: Despite widespread use of standardized classification systems, risk stratification of thyroid nodules is nuanced and often requires diagnostic surgery. Genomic sequencing is available for this dilemma however, costs and access restricts global applicability. Artificial intelligence (AI) has the potential to overcome this issue nevertheless, the need for black-box interpretability is pertinent. We aimed to create an ultrasonographic segmentation and classification model that offers explainability and risk accountability. METHODOLOGY: Four hundred and fourteen ultrasonography images were collected from 105 patients undergoing thyroidectomy, divided into training and testing groups. Classification ground truth used is exclusively surgical histopathology. Relevant nodules were manually annotated by a dedicated study radiologist and surgeon. Three AI architectures with and without block attention modules were trained to identify the relevant nodule and the best performing was selected for the subsequent task in classifying identified nodules into benign or malignant. Gradient-Weighted Class Activation Map is used to provide saliency mapping for visual interpretability. FINDINGS: Superior performance was recorded by the block attention model which stratified thyroid nodules into benign versus malignant with an accuracy of 93% versus 90%, F-score 90% versus 89%, sensitivity 93% versus 91% and specificity 92% versus 91% on a training dataset versus a testing dataset respectively. GRADCAM: Visual interpretability maps demonstrate salient areas for a benign nodule diagnosis overlaps spongiform areas and malignant diagnosis salient areas overlap solid components of a partially cystic-solid nodule and microcalcifications within nodules. These findings are consistent with established diagnostic criteria for benign and malignant nodules. CONCLUSION: We developed an image segmentation and classification model for the risk stratification of thyroid nodules benchmarking surgical histopathology as ground truth and providing visual interpretability.
- Publisher
- Wiley
- Keywords
- Humans; *Thyroid Nodule/diagnostic imaging/surgery/classification/pathology; Risk Assessment/methods; Ultrasonography/methods; Thyroidectomy; Female; Male; Middle Aged; Adult; *Artificial Intelligence; Preoperative Care/methods; Sensitivity and Specificity; Aged; endocrine
- Department(s)
- Surgical Oncology
- Publisher's Version
- https://doi.org/10.1002/wjs.12464
- Terms of Use/Rights Notice
- Refer to copyright notice on published article.
Creation Date: 2025-01-21 06:44:54
Last Modified: 2025-05-13 01:31:57