| Literature DB >> 35381046 |
Vemund Fredriksen1, Svein Ole M Sevle1, André Pedersen2,3,4, Thomas Langø2,5, Gabriel Kiss1,5, Frank Lindseth1,6.
Abstract
PURPOSE: Cancer is among the leading causes of death in the developed world, and lung cancer is the most lethal type. Early detection is crucial for better prognosis, but can be resource intensive to achieve. Automating tasks such as lung tumor localization and segmentation in radiological images can free valuable time for radiologists and other clinical personnel. Convolutional neural networks may be suited for such tasks, but require substantial amounts of labeled data to train. Obtaining labeled data is a challenge, especially in the medical domain.Entities:
Mesh:
Year: 2022 PMID: 35381046 PMCID: PMC8982833 DOI: 10.1371/journal.pone.0266147
Source DB: PubMed Journal: PLoS One ISSN: 1932-6203 Impact factor: 3.240
Fig 1The method overview during training.
Firstly, the teacher was trained using the strong dataset (semantic annotated images) represented with blue lines. The teacher was then used to make semantic annotations for the weak dataset (bounding box annotated images). The student was then trained on both the pseudo-annotated dataset D, represented by the orange lines, and the strong dataset D.
Tumor sizes of the three datasets.
| Dataset | Volume [ | Diameter [ |
|---|---|---|
| MSD-Lung | 21.98 ± 51.66 | 37.63 ± 20.08 |
| NSCLC-Radiomics | 75.37 ± 96.30 | 63.63 ± 29.62 |
| Lung-PET-CT-Dx | 63.67 ± 86.26 | 48.66 ± 19.85 |
Fig 2The network architectures.
The single output (SO) Student is highlighted in colors whereas the decoder branch of the dual output (DO) Student is implicated in gray. For the DO Student the ouput of the two decoder branches are concatenated to form a dual-channeled output. The teachers have the same architecture as the SO Student, but with three downsamplings rather than four.
Teacher results.
| Dataset | Model | DSC |
|---|---|---|
| MSD-Lung | Point Guided | 74.78 ± 11.83 |
| Box Guided |
| |
| NSCLC-Radiomics | Point Guided | 59.57 ± 23.90 |
| Box Guided |
| |
| Both | Point Guided | 61.48 ± 23.29 |
| Box Guided |
|
The best performing model with respect to mean dice similarity coefficient (DSC) is highlighted in bold.
Student results.
| Dataset | Model | DSC | DSC-TP | F1-score | Recall | Precision |
|---|---|---|---|---|---|---|
| MSD-Lung | Baseline |
|
|
|
|
|
| SO Student | 64.27±16.05 | 71.32±8.06 |
|
|
| |
| DO Student | 55.37 ± 29.03 | 70.49 ± 8.82 | 74.07 ± 40.91 | 77.78 ± 41.57 | 72.22 ± 41.57 | |
| NSCLC-Radiomics | Baseline | 51.06 ± 28.22 | 68.81 ± 18.27 | 63.56 ± 36.36 |
| 56.68 ± 38.65 |
| SO Student |
|
| 64.18 ± 37.37 | 79.90 ± 39.66 | 58.76 ± 39.28 | |
| DO Student | 52.25 ± 30.18 | 68.69 ± 19.47 |
| 79.17 ± 39.75 |
| |
| Both | Baseline | 52.96 ± 27.98 | 69.34 ± 17.97 | 65.66 ± 36.32 |
| 59.14 ± 38.76 |
| SO Student |
|
| 66.20 ± 37.19 | 80.95 ± 38.90 | 60.98 ± 39.21 | |
| DO Student | 52.61 ± 30.06 | 68.90 ± 18.59 |
| 79.00 ± 39.97 |
|
For each respective metric, the best performing models are highlighted in bold.
Scarcely trained teacher results.
| Dataset | Model | DSC |
|---|---|---|
| MSD-Lung | Scarce Point Guided | 48.52 ± 31.18 |
| Scarce Box Guided |
| |
| NSCLC-Radiomics | Scarce Point Guided | 43.83 ± 25.65 |
| Scarce Box Guided |
| |
| Both | Scarce Point Guided | 44.42 ± 26.45 |
| Scarce Box Guided |
|
The best performing model with respect to mean dice similarity coefficient (DSC) is highlighted in bold.
Scarcely trained student results.
| Dataset | Model | DSC | DSC-TP | F1-score | Recall | Precision |
|---|---|---|---|---|---|---|
| MSD-Lung | Baseline | 26.45 ± 26.56 | 75.24 ± 15.90 | 09.10 ± 13.26 | 33.33 ± 47.14 | 05.31 ± 07.80 |
| SO Student | 64.74 ± 11.82 | 71.56 ± 10.40 | 61.85 ± 16.49 |
| 47.22 ± 20.79 | |
| DO Student |
|
|
| 88.89 ± 31.43 |
| |
| NSCLC-Radiomics | Baseline | 28.23 ± 28.05 | 55.39 ± 22.80 | 32.13 ± 36.99 | 51.47 ± 49.24 | 26.99 ± 35.68 |
| SO Student | 51.06 ± 30.75 |
| 62.65 ± 36.73 | 79.41 ± 39.51 | 56.67 ± 38.79 | |
| DO Student |
| 67.44 ± 21.70 |
|
|
| |
| Both | Baseline | 28.02 ± 27.89 | 56.91 ± 22.96 | 29.44 ± 35.82 | 49.35 ± 49.34 | 24.45 ± 34.35 |
| SO Student | 52.66 ± 29.51 |
| 62.55 ± 34.98 | 81.82 ± 37.72 | 55.56 ± 37.26 | |
| DO Student |
| 68.56 ± 20.71 |
|
|
|
For each respective metric, the best performing model is highlighted in bold.
Fig 3A sample of the results produced by the scarce students on the test set.
The figure shows the input image, bounding box, and ground truth (GT) mask in the three top rows, respectively. The baseline model, single output (SO) Student, and dual output (DO) Students corresponding outputs are shown in the three bottom rows, respectively.