mlsanityData to 7 Oct 2026

Clinical DVH metrics as a loss function for 3D dose prediction in head and neck radiotherapy

arXiv 2603.29670 ↗First posted 31 Mar 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Purpose: Deep-learning-based three-dimensional (3D) dose prediction is widely used in automated radiotherapy workflows. However, most existing models are trained with voxel-wise regression losses, which are poorly aligned with clinical plan evaluation criteria based on dose-volume histogram (DVH) metrics. This study aims to develop a clinically guided loss formulation that directly optimizes clinically used DVH metrics while remaining computationally efficient for head and neck (H&N) dose prediction. Methods: We propose a clinical DVH metric loss (CDM loss) that incorporates differentiable textitD-metrics and surrogate textitV-metrics, together with a lossless bit-mask region-of-interest (ROI) encoding to improve training efficiency. The method was evaluated on 174 H&N patients using a temporal split (137 training, 37 testing). Results: Compared with MAE- and DVH-curve based losses, CDM loss substantially improved target coverage and satisfied all clinical constraints. Using a standard 3D U-Net, the PTV Score was reduced from 1.544 (MAE) to 0.491 (MAE + CDM), while OAR sparing remained comparable. Bit-mask encoding reduced training time by 83% and lowered GPU memory usage. Conclusion: Directly optimizing clinically used DVH metrics enables 3D dose predictions that are better aligned with clinical treatment planning criteria than conventional voxel-wise or DVH-curve-based supervision. The proposed CDM loss, combined with efficient ROI bit-mask encoding, provides a practical and scalable framework for H&N dose prediction.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 8 of its 20 references that are in the map. First posted on arXiv 31 Mar 2026.

Builds on

8 of 20 references.

References · 8

  1. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 823
  2. 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 161
  3. 2020OpenKBP: The open-access knowledge-based planning grand challengecited by 30
  4. 2019Incorporating human and learned domain knowledge into training deep neural networks: A differentiable dose volume histogram and adversarial inspired framework for generating Pareto optimal dose distributions in radiation therapycited by 10
  5. 2022Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Functioncited by 5
  6. 2020Comparing Normalization Methods for Limited Batch Size Segmentation Neural Networkscited by 3
  7. 2021LENAS: Learning-based Neural Architecture Search and Ensemble for 3D Radiotherapy Dose Predictioncited by 3
  8. 2023A Cascade Transformer-based Model for 3D Dose Distribution Prediction in Head and Neck Cancer Radiotherapycited by 2

Built on by

Not cited yet.