Název: Strategies for Training Deep Learning Models in Medical Domains with Small Reference Datasets
Autoři: Zwettler, Gerald A.
Holmes III, David R.
Backfrieder, Werner
Citace zdrojového dokumentu: Journal of WSCG. 2020, vol. 28, no. 1-2, p. 37-46.
Datum vydání: 2020
Nakladatel: Václav Skala - UNION Agency
Typ dokumentu: článek
article
URI: http://wscg.zcu.cz/WSCG2020/2020-J_WSCG-1-2.pdf
http://hdl.handle.net/11025/38423
ISSN: 1213-6972 (print)
1213-6980 (CD-ROM)
1213-6964 (on-line)
Klíčová slova: segmentace lékařského obrazu;hluboké učení;Generativní kontradiktorní sítě;metoda Graph cut
Klíčová slova v dalším jazyce: medical image segmentation;deep learning;Generative Adversial Networks;Graph cut
Abstrakt v dalším jazyce: With the steady progress of Deep Learning (DL), powerful tools are now present for sophisticated segmentation tasks. Nevertheless, the generally very high demand for training data and precise reference segmentations often cannot be met in medical domains when processing small and individual studies or acquisition protocols. As common strategies, reinforcement learning or transfer learning are applicable but coherent with immense effort due to domain-specific adjustment. In this work the applicability of a U-net cascade for training on a very low amount of abdominal MRI datasets of the parenchyma is evaluated and strategies to compensate for the lack of training data are discussed. Although the model accuracy when training on 13 MRI volumes with achievable JI=89.41 is rather low, results are still good enough for manual post-processing utilizing a Graph cut (GC) approach with medium demand for user interaction. This way, the DL models are retrained, when additional test data sets become available to subsequently improve the classification accuracy. With only 2 additional GC postprocessed datasets, the accuracy after model re-training is increased to JI= 89.87. Besides, the applicability of Generative Adversial Networks (GAN) in the medical domain is evaluated discussing to synthesize axial CT slices together with perfect ground truth reference segmentations. It is shown for abdominal CT slices of the parenchyma, that in case of lack of training data, synthesized slices, that can be derived at arbitrary number, help to significantly improve the DL training process when only an insufficient amount of data is available. While training on 2,200 real images only leads to accuracy JI=88.75, the enrichment with 2,200 additional images synthesized from a GAN trained on 5,000 datasets only leads to an increase up to JI=92.02. Even if the DL model is exclusively trained on 4,400 computer-generated images, the classification accuracy on real-world data is notable with JI=90.81.
Práva: © Václav Skala - UNION Agency
Vyskytuje se v kolekcích:Volume 28, Number 1-2 (2020)

Soubory připojené k záznamu:
Soubor Popis VelikostFormát 
Zwettler.pdfPlný text4,37 MBAdobe PDFZobrazit/otevřít


Použijte tento identifikátor k citaci nebo jako odkaz na tento záznam: http://hdl.handle.net/11025/38423

Všechny záznamy v DSpace jsou chráněny autorskými právy, všechna práva vyhrazena.