AI4MIRO/Dose_prediction_han_PBS

HDUNet PBS

Trained by Margerie Huet Dastarac .
Training date: 06/06/2023 .

1. Task Description

Prediction of the dose distribution with proton therapy beam scanning treatment.

2. Model

2.1. Architecture

Figure 1: HDUNet PBS architecture

2.2. Input

• CT
• Target volumes contours and prescription
• Organs at risks contours

2.3. Output

• DOSE

2.4 Training details

• Number of epoch: 400
• Loss function: MSE loss
• Optimizer: AdamW
• Learning Rate: 0.0001
• Dropout: No
• Patch size in voxels: (128,128,128)
• Data augmentation used:
  • RandCrop

3. Dataset

• Location: Oropharynx
• Training set size: 57
• Resolution in mm: 3x3x3

Performance

• TBD