| <h1 style="border-bottom: 2px solid black; font-size: 100px;" align="center"> HDUNet PBS </h1> | |
| _Trained by Margerie Huet Dastarac ._ <br> | |
| _Training date: 06/06/2023 ._ | |
| ## 1. Task Description | |
| Prediction of the dose distribution with proton therapy beam scanning treatment. | |
| ## 2. Model | |
| ### 2.1. Architecture | |
| <img width="100%" src="https://cdn-uploads.huggingface.co/production/uploads/65c9dbefd6cbf9dfed67367e/xGGUE5spRY6tar5R4VeKX.png" alt="error while loading image"> | |
| _Figure 1: HDUNet PBS architecture_ | |
| ### 2.2. Input | |
| <ul> | |
| <li> CT</li> | |
| <li> Target volumes contours and prescription</li> | |
| <li> Organs at risks contours</li> | |
| </ul> | |
| ### 2.3. Output | |
| <ul> | |
| <li> DOSE</li> | |
| </ul> | |
| ### 2.4 Training details | |
| <ul> | |
| <li> Number of epoch: 400 </li> | |
| <li> Loss function: MSE loss </li> | |
| <li> Optimizer: AdamW </li> | |
| <li> Learning Rate: 0.0001 </li> | |
| <li> Dropout: No </li> | |
| <li> Patch size in voxels: (128,128,128) </li> | |
| <li> Data augmentation used: | |
| <ul> | |
| <li> RandCrop</li> | |
| </ul> | |
| </li> | |
| </ul> | |
| ## 3. Dataset | |
| <ul> | |
| <li> Location: Oropharynx </li> | |
| <li> Training set size: 57 </li> | |
| <li> Resolution in mm: 3x3x3 </li> | |
| </ul> | |
| ## Performance | |
| + TBD | |