Video-based AI module with raw-scale and ROI-scale information for thyroid nodule diagnosis

Linghu Wu,Yuli Zhou,Mengmeng Liu,Sijing Huang,Youhuan Su,Xiaoshu Lai,Song Bai,Keen Yang,Yitao Jiang,Chen Cui,Siyuan Shi,Jinfeng Xu,Nan Xu,Fajin Dong
DOI: https://doi.org/10.1016/j.heliyon.2024.e37924
IF: 3.776
2024-09-19
Heliyon
Abstract:Objectives: Ultrasound examination is a primary method for detecting thyroid lesions in clinical practice. Incorrect ultrasound diagnosis may lead to delayed treatment or unnecessary biopsy punctures. Therefore, our objective is to propose an artificial intelligence model to increase the precision of thyroid ultrasound diagnosis and reduce puncture rates. Methods: We consecutively collected ultrasound recordings from 672 patients with 845 nodules across two Chinese hospitals. This dataset was divided into training, validation, and internal test sets in a ratio of 7:1:2. We constructed and tested six different model variants based on different video feature distillation strategies and whether additional information from ROI (Region of Interest) scales was used. The models' performances were evaluated using the internal test set and an additional external test set containing 126 nodules from a third hospital. Results: The dual-stream model, which contains both raw-scale and ROI-scale streams with the time dimensional convolution layer, achieved the best performance on both internal and external test sets. On the internal test set, it achieved an AUROC (Area Under Receiver Operating Characteristic Curve) of 0.969 (95 % confidence interval, CI: 0.944-0.993) and an accuracy of 92.6 %, outperforming other variants (AUROC: 0.936-0.955, accuracy: 80.2%-88.3 %) and experienced radiologists (accuracy: 91.9 %). The AUROC of the best model in the external test was 0.931 (95 % CI: 0.890-0.972). Conclusion: Integrating a dual-stream model with additional ROI scale information and the time dimensional convolution layer can improve performance in diagnosing thyroid ultrasound videos.
What problem does this paper attempt to address?