Cargando…

Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning

PURPOSE: To present the technical details of the runner‐up model in the open knowledge‐based planning (OpenKBP) challenge for the dose–volume histogram (DVH) stream. The model was designed to ensure simple and reproducible training, without the necessity of costly advanced generative adversarial net...

Descripción completa

Detalles Bibliográficos
Autores principales: Zimmermann, Lukas, Faustmann, Erik, Ramsl, Christian, Georg, Dietmar, Heilemann, Gerd
Formato: Online Artículo Texto
Lenguaje:English
Publicado: John Wiley and Sons Inc. 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8518421/
https://www.ncbi.nlm.nih.gov/pubmed/34156727
http://dx.doi.org/10.1002/mp.14774
_version_ 1784584219148681216
author Zimmermann, Lukas
Faustmann, Erik
Ramsl, Christian
Georg, Dietmar
Heilemann, Gerd
author_facet Zimmermann, Lukas
Faustmann, Erik
Ramsl, Christian
Georg, Dietmar
Heilemann, Gerd
author_sort Zimmermann, Lukas
collection PubMed
description PURPOSE: To present the technical details of the runner‐up model in the open knowledge‐based planning (OpenKBP) challenge for the dose–volume histogram (DVH) stream. The model was designed to ensure simple and reproducible training, without the necessity of costly advanced generative adversarial network (GAN) techniques. METHODS: The model was developed based on the OpenKBP challenge dataset, consisting of 200 and 40 head‐and‐neck patients for training and validation, respectively. The final model is a U‐Net with additional ResNet blocks between up‐ and down convolutions. The results were obtained by training the model with AdamW with the One Cycle scheduler. The loss function is a combination of the L1 loss with a feature loss, which uses a pretrained video classifier as a feature extractor. The performance was evaluated on another 100 patients in the OpenKBP test dataset. The DVH metrics of the test data were evaluated, where [Formula: see text] , and [Formula: see text] were calculated for the organs at risk (OARs) and [Formula: see text] , [Formula: see text] , and [Formula: see text] were computed for the target structures. DVH metric differences between predicted and true dose are reported in percentage. RESULTS: The model achieved 2nd and 4th place in the DVH and dose stream of the OpenKBP challenge, respectively. The dose and DVH score were 2.62 ± 1.10 and 1.52 ± 1.06, respectively. Mean dose differences for the different structures and DVH parameters were within ±1%. CONCLUSION: This straightforward approach produced excellent results. It incorporated One Cycle Learning, ResNet, and feature‐based losses, which are common computer vision techniques.
format Online
Article
Text
id pubmed-8518421
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher John Wiley and Sons Inc.
record_format MEDLINE/PubMed
spelling pubmed-85184212021-10-21 Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning Zimmermann, Lukas Faustmann, Erik Ramsl, Christian Georg, Dietmar Heilemann, Gerd Med Phys Special Issue: Open Knowledge‐based Planning PURPOSE: To present the technical details of the runner‐up model in the open knowledge‐based planning (OpenKBP) challenge for the dose–volume histogram (DVH) stream. The model was designed to ensure simple and reproducible training, without the necessity of costly advanced generative adversarial network (GAN) techniques. METHODS: The model was developed based on the OpenKBP challenge dataset, consisting of 200 and 40 head‐and‐neck patients for training and validation, respectively. The final model is a U‐Net with additional ResNet blocks between up‐ and down convolutions. The results were obtained by training the model with AdamW with the One Cycle scheduler. The loss function is a combination of the L1 loss with a feature loss, which uses a pretrained video classifier as a feature extractor. The performance was evaluated on another 100 patients in the OpenKBP test dataset. The DVH metrics of the test data were evaluated, where [Formula: see text] , and [Formula: see text] were calculated for the organs at risk (OARs) and [Formula: see text] , [Formula: see text] , and [Formula: see text] were computed for the target structures. DVH metric differences between predicted and true dose are reported in percentage. RESULTS: The model achieved 2nd and 4th place in the DVH and dose stream of the OpenKBP challenge, respectively. The dose and DVH score were 2.62 ± 1.10 and 1.52 ± 1.06, respectively. Mean dose differences for the different structures and DVH parameters were within ±1%. CONCLUSION: This straightforward approach produced excellent results. It incorporated One Cycle Learning, ResNet, and feature‐based losses, which are common computer vision techniques. John Wiley and Sons Inc. 2021-06-22 2021-09 /pmc/articles/PMC8518421/ /pubmed/34156727 http://dx.doi.org/10.1002/mp.14774 Text en © 2021 The Authors. Medical Physics published by Wiley Periodicals LLC on behalf of American Association of Physicists in Medicine. https://creativecommons.org/licenses/by-nc-nd/4.0/This is an open access article under the terms of the http://creativecommons.org/licenses/by-nc-nd/4.0/ (https://creativecommons.org/licenses/by-nc-nd/4.0/) License, which permits use and distribution in any medium, provided the original work is properly cited, the use is non‐commercial and no modifications or adaptations are made.
spellingShingle Special Issue: Open Knowledge‐based Planning
Zimmermann, Lukas
Faustmann, Erik
Ramsl, Christian
Georg, Dietmar
Heilemann, Gerd
Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title_full Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title_fullStr Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title_full_unstemmed Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title_short Technical Note: Dose prediction for radiation therapy using feature‐based losses and One Cycle Learning
title_sort technical note: dose prediction for radiation therapy using feature‐based losses and one cycle learning
topic Special Issue: Open Knowledge‐based Planning
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8518421/
https://www.ncbi.nlm.nih.gov/pubmed/34156727
http://dx.doi.org/10.1002/mp.14774
work_keys_str_mv AT zimmermannlukas technicalnotedosepredictionforradiationtherapyusingfeaturebasedlossesandonecyclelearning
AT faustmannerik technicalnotedosepredictionforradiationtherapyusingfeaturebasedlossesandonecyclelearning
AT ramslchristian technicalnotedosepredictionforradiationtherapyusingfeaturebasedlossesandonecyclelearning
AT georgdietmar technicalnotedosepredictionforradiationtherapyusingfeaturebasedlossesandonecyclelearning
AT heilemanngerd technicalnotedosepredictionforradiationtherapyusingfeaturebasedlossesandonecyclelearning