Effectiveness of Automatically Curated Dataset in Thyroid Nodules Classification Algorithms Using Deep Learning
Organizations: Department of Electrical & Computer Engineering, Duke University, Durham, NC, USA · Department of Radiology, Duke University Medical Center, Durham, NC, USA
Abstract
The diagnosis of thyroid nodule cancers commonly utilizes ultrasound images. Several studies showed that deep learning algorithms designed to classify benign and malignant thyroid nodules could match radiologists' performance. However, data availability for training deep learning models is often limited due to the significant effort required to curate such datasets. Our previous study proposed a method (MADLaP) to curate thyroid nodule datasets automatically. It was tested to have a 63% yield rate and 83% accuracy. However, the usefulness of the generated data for training deep learning models remains unknown. In this study, we conducted experiments to determine whether using an automatically-curated dataset improves deep learning algorithms' performance. The expert annotations of 378 patients either trained a model directly (803 images) or built MADLaP, which curated 5,228 images from 3,981 patients; the same network and training method were used throughout, and 10-model cross-validation ensembles were tested on 378 nodules from an earlier period. The model trained on the automatically-curated dataset had a significantly higher AUC than the model trained on the manually selected dataset (0.657 vs 0.607; difference 0.051, 95% CI: 0.020, 0.081; P = .001), an advantage seen in all 22 configuration pairs. On the same test set, this model was within the AUC range of four radiologists (0.63-0.66) and not significantly different from a benchmark trained on 1,278 expert-curated nodules (0.69; difference -0.036, 95% CI: -0.083, 0.011), whereas manually curated models fell significantly below it. The Stage-1 subset (86% accuracy) did not differ significantly from the full MADLaP dataset (difference 0.015, 95% CI: -0.006, 0.036). In conclusion, we showed that using an automatically-curated dataset can substantially increase the performance of deep learning algorithms.
Figures & tables
| Characteristic | Development cohort | Manual Set | MADLaP Set | S1 Set | Test set |
|---|---|---|---|---|---|
| Institution; biopsy period | Duke; Oct 2014 – Oct 2020 | Duke; development cohort | Duke; development cohort | Duke; development cohort | Duke; Jan 2004 – Oct 2014 |
| Patients | 3,981 (of 4,354 queried) | 378 (random) | run on all 3,981 | subset of MADLaP | 320 |
| Nodules | — | 404 | 2,605 | 2,066 | 378 |
| Images (benign / malignant) | — | 803 (752 / 51) | 5,228 (4,970 / 258) | 4,150 (3,958 / 192) | 756 (462 / 294) |
| Malignant nodules, n (%) | — | 26 (6.4) | 129 (5.0) | 96 (4.6) | 147 (38.9) |
| Image selection and label | — | radiologist; pathology report | MADLaP Stages 1–2 | MADLaP Stage 1 | radiologists; cytopathology |
| Model (batch/epochs) | Training images (malignant) | AUC (95% CI) | Fold models, mean SD | Difference vs Manual 256/250 (95% CI) | P |
|---|---|---|---|---|---|
| Manual 256/250 | 803 (51) | 0.607 (0.547, 0.666) | 0.600 0.015 | reference | — |
| MADLaP 512/250 (primary) | 5,228 (258) | 0.657 (0.600, 0.714) | 0.641 0.016 | 0.051 (0.020, 0.081) | .001 |
| MADLaP 256/300 (same batch size) | 5,228 (258) | 0.662 (0.605, 0.719) | 0.638 0.016 | 0.055 (0.022, 0.088) | .001 |
| MADLaP 512/350 | 5,228 (258) | 0.666 (0.608, 0.723) | 0.646 0.026 | 0.059 (0.026, 0.092) | < .001 |
| MADLaP 1,024/450 (highest CV AUC) | 5,228 (258) | 0.651 (0.594, 0.708) | 0.635 0.024 | 0.045 (0.014, 0.076) | .005 |
| S1 512/350 (highest CV AUC) | 4,150 (192) | 0.677 (0.621, 0.732) | 0.659 0.014 | 0.070 (0.033, 0.107) | < .001 |
| MADLaP 512/250 minus comparator | Manual 256/250 minus comparator | ||||
| Comparator | AUC (95% CI) | Difference (95% CI) | P | Difference (95% CI) | P |
| Radiologists (5-point score) | |||||
| Radiologist 1 | 0.630 (0.585, 0.674) | 0.027 ( 0.039, 0.094) | .42 | 0.023 ( 0.093, 0.046) | .51 |
| Radiologist 2 | 0.661 (0.609, 0.713) | 0.004 ( 0.070, 0.062) | .91 | 0.055 ( 0.121, 0.012) | .11 |
| Radiologist 3 | 0.649 (0.601, 0.697) | 0.008 ( 0.055, 0.071) | .80 | 0.042 ( 0.109, 0.024) | .21 |
| Radiologist 4 | 0.627 (0.582, 0.673) | 0.030 ( 0.033, 0.093) | .35 | 0.021 ( 0.088, 0.046) | .54 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Item | Version 1 | Version 2 | Basis |
|---|---|---|---|
| Test-set AUC | For each configuration, the highest test AUC among its ten fold models (e.g., MADLaP 0.694, S1 0.689, the batch-size points) | 10-model ensemble for every configuration; fold-model mean SD; pooled ensembles | On the 383 nodules v1 used, the fold models whose test AUCs v1 reported exceeded their ensembles by 0.012 to 0.035 |
| Manual result at batch size 512 (0.643) | Reported | Not reported; the Manual models compared have batch sizes 128 and 256 | Model not archived |
| Confidence intervals and P values | About 0.03, bootstrap with unstated unit; “P > .43” | Nodule-level DeLong ( 0.056 to 0.059 for the ensembles); paired DeLong; patient-clustered sensitivity analyses; Holm adjustment | v1 intervals could not be reproduced with a standard resampling unit; “.43” is close to a one-sided bootstrap share (0.41 to 0.46), not a two-sided P value (0.81 to 0.84) |
| Section 3.1 cross-validation | 0.78 (0.74, 0.82) for both sets, which is the cross-validation result of the benchmark study [ 4 ] | This study’s cross-validation for every configuration | — |
| Epoch selection | “selected based on ROCAUC in 10-fold cross validation” | Epochs fixed per configuration; epoch variants had been compared on the test set, so all archived configurations are reported | Training records |
| S1 vs MADLaP | S1 slightly lower; “use all the data” | No significant difference; S1 numerically higher | Ensemble estimates |
| Item | Version 1 | Version 2 | Basis |
|---|---|---|---|
| Development period | 2013–2019 | October 2014 to October 2020 | Zhang et al. [60] |
| Stage-1 accuracy | 88% | 86% (44 of 51) | Zhang et al. [60] |
| Radiologist experience | 4 years post-fellowship | 3 years post-fellowship at the time of annotation | Zhang et al. [60] |
| Benchmark AUC | 0.70 (0.64–0.75) | 0.69 (0.64, 0.75) | Weng et al. [50] |
| Annotated patients | 320 | 378 | 320 is the number of test-set patients |
| Table 1 | “Training dataset”; age rows inconsistent | Characteristics of every set; sex of the 4,354 queried patients kept; ages not reported | Sex counts sum to the 4,354 queried patients; the per-sex mean ages do not reproduce the overall mean; the source table was not available |
| Test set (378 nodules) | |||||||
| Set and batch/epochs | CV pooled out-of-fold AUC (95% CI) | Ensemble AUC (95% CI) | Fold models, mean SD | Highest fold model | P vs radiologists | P vs averaged score | P vs benchmark |
| Manual Set | |||||||
| 128/200 | 0.730 (0.633, 0.827) | 0.612 (0.553, 0.671) | 0.605 0.020 | 0.632 | .15–.65 | .001 | .001 |
| 256/250 | 0.767 (0.672, 0.863) | 0.607 (0.547, 0.666) | 0.600 0.015 | 0.629 | .11–.54 | < .001 | .001 |
| MADLaP Set | |||||||
| 256/300 | 0.709 (0.660, 0.758) * | 0.662 (0.605, 0.719) | 0.638 0.016 | 0.660 | .28–.98 | .073 | .19 |
| vs Manual 128/200 | vs Manual 256/250 | |||||||
| Automatic configuration | Difference (95% CI) | P | Holm P, DeLong | Holm P, clustered | Difference (95% CI) | P | Holm P, DeLong | Holm P, clustered |
| MADLaP 256/300 | 0.050 (0.019, 0.081) | .002 | .015 | .013 | 0.055 (0.022, 0.088) | .001 | .011 | .013 |
| MADLaP 512/250 | 0.045 (0.020, 0.071) | < .001 | .009 | .010 | 0.051 (0.020, 0.081) | .001 | .011 | .013 |
| MADLaP 512/300 | 0.047 (0.021, 0.073) | < .001 | .006 | .009 | 0.053 (0.022, 0.083) | < .001 | .009 | .013 |
| MADLaP 512/340 | 0.030 (0.003, 0.057) | .027 | .048 | .051 | 0.036 (0.005, 0.067) | .024 | .048 | .051 |
| MADLaP 512/350 | 0.053 (0.026, 0.081) | < .001 | .004 | .004 | 0.059 (0.026, 0.092) | < .001 | .009 | .010 |
| Analysis | Difference (95% CI) | P | Note |
| MADLaP 512/250 minus Manual 256/250 | |||
| Main analysis (378 nodules) | 0.051 (0.020, 0.081) | .001 | paired DeLong |
| 383 nodules (original labels) | 0.053 (0.023, 0.084) | < .001 | pooled MADLaP minus Manual 0.052 (0.024, 0.080), P < .001; positive pairs 22 of 22; Holm-significant 22 (DeLong), 22 (clustered) |
| 381 nodules with radiologist scores | 0.054 (0.024, 0.085) | < .001 | pooled MADLaP minus Manual 0.054 (0.026, 0.082), P < .001; positive pairs 22 of 22; Holm-significant 22 (DeLong), 22 (clustered) |
| 377 nodules without label disagreement | 0.050 (0.019, 0.080) | .001 | pooled MADLaP minus Manual 0.048 (0.020, 0.076), P < .001; positive pairs 22 of 22; Holm-significant 20 (DeLong), 20 (clustered) |
| Image level (756 images) | 0.049 (0.022, 0.076) | < .001 | AUC 0.649 vs 0.600; nodule-cluster bootstrap CI; P from variance clustered by nodule |
| MADLaP 512/250 | Manual 256/250 | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Radiologist decision set | Biopsies n / nodules N | Radiologist sensitivity / specificity | Sensitivity / specificity | Additional cancers | McNemar P | Holm P | Sensitivity / specificity | Additional cancers | McNemar P | Holm P |
| Radiologists’ own recommendations | ||||||||||
| Radiologist 1 | 273 / 378 | 0.735 / 0.286 | 0.823 / 0.342 | 13 | .098 | .59 | 0.782 / 0.316 | 7 | .43 | > .99 |
| Radiologist 2 | 305 / 378 | 0.830 / 0.208 | 0.878 / 0.238 | 7 | .31 | > .99 | 0.844 / 0.216 | 2 | .87 | > .99 |
| Radiologist 3 | 274 / 378 | 0.769 / 0.303 | 0.830 / 0.342 | 9 | .23 | > .99 | 0.782 / 0.312 | 2 | .89 | > .99 |
| Radiologist 4 | 270 / 378 | 0.687 / 0.268 | 0.810 / 0.346 | 18 | .027 | .19 | 0.782 / 0.329 | 14 | .10 | .72 |
| Block | Operation | Output | Parameters |
| Input | — | 160 160 1 | — |
| Conv 1 | 3 3, 8; pool | 80 80 8 | 80 |
| Conv 2 | 3 3, 12; pool | 40 40 12 | 876 |
| Conv 3 | 3 3, 16; pool | 20 20 16 | 1,744 |
| Conv 4 | 3 3, 24; pool | 10 10 24 | 3,480 |
| Conv 5 | 3 3, 32; pool | 5 5 32 | 6,944 |