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...
Autores principales: | , , , , |
---|---|
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 |