|
1 | 1 | # Importance of Aligning Training Strategy with Evaluation for Diffusion Models in 3D Multiclass Segmentation |
2 | 2 |
|
3 | | -<div> |
4 | | -<img src="images/figure2.png" width="600" alt="figure2"></img> |
5 | | -</div> |
6 | | - |
7 | | -## Results |
8 | | - |
9 | | -DS, NSD, HD, and CD represents Dice score, normalised surface Dice, 95% |
10 | | -Hausdorff Distance, and centroid distance. The mean and standard deviations |
11 | | -values are reported. |
| 3 | +:tada: This work has been accepted at |
| 4 | +[Deep Generative Models workshop at MICCAI 2023](https://dgm4miccai.github.io/). |
12 | 5 |
|
13 | | -### Prostate MR Data Set |
| 6 | +:bookmark_tabs: An updated manuscript has also been uploaded at |
| 7 | +[arXiv](https://arxiv.org/abs/2303.06040). |
14 | 8 |
|
15 | | -| Model | Diffusion Model | DS | NSD | HD | CD | |
16 | | -| :---- | :-------------- | :------------ | :------------ | :------------ | :------------ | |
17 | | -| 2D | No | 0.831 (0.098) | 0.638 (0.113) | 6.044 (1.031) | 2.824 (1.239) | |
18 | | -| 2D | Yes | 0.818 (0.102) | 0.615 (0.118) | 6.658 (0.839) | 3.012 (1.174) | |
19 | | -| 3D | No | 0.838 (0.088) | 0.648 (0.110) | 5.197 (1.184) | 2.675 (0.927) | |
20 | | -| 3D | Yes | 0.830 (0.094) | 0.626 (0.112) | 5.424 (1.176) | 3.009 (1.165) | |
| 9 | +:mag_right: We are working on a follow-up work, stay tuned. |
21 | 10 |
|
22 | | -### Abdominal CT Data Set |
| 11 | +<div> |
| 12 | +<img src="images/method_x0.png" width="600" alt="figure2"></img> |
| 13 | +</div> |
23 | 14 |
|
24 | | -| Model | Diffusion Model | DS | NSD | HD | CD | |
25 | | -| :---- | :-------------- | :------------ | :------------ | :------------- | :------------ | |
26 | | -| 3D | Yes | 0.801 (0.109) | 0.540 (0.095) | 9.125 (2.564) | 4.836 (2.273) | |
27 | | -| 2D | Yes | 0.769 (0.127) | 0.520 (0.091) | 12.039 (2.932) | 5.121 (2.102) | |
28 | | -| 3D | No | 0.816 (0.100) | 0.596 (0.084) | 9.091 (2.475) | 4.275 (1.870) | |
29 | | -| 2D | No | 0.804 (0.109) | 0.577 (0.082) | 9.885 (2.587) | 4.416 (1.914) | |
| 15 | +<div> |
| 16 | +<img src="images/figure2.png" width="600" alt="figure2"></img> |
| 17 | +</div> |
30 | 18 |
|
31 | | -### Reproduction |
| 19 | +## Reproduction |
32 | 20 |
|
33 | 21 | Install the environment and build the dataset following the documentation. Then |
34 | 22 | run one of the following sets of commands. |
@@ -317,11 +305,10 @@ pytest --cov=imgx -n 4 tests |
317 | 305 |
|
318 | 306 | ## Acknowledgement |
319 | 307 |
|
320 | | -This work was supported by the Wellcome/EPSRC Centre for Interventional and |
321 | | -Surgical Sciences (203145Z/16/Z), the EPSRC funded Centre for Doctoral Training |
322 | | -in Intelligent, Integrated Imaging in Healthcare (i4Health) (EP/S021930/1), the |
323 | | -EPSRC grant EP/T029404/1), the International Alliance for Cancer Early |
324 | | -Detection, an alliance between Cancer Research UK [C28070/A30912; |
325 | | -C73666/A31378], Canary Center at Stanford University, the University of |
326 | | -Cambridge, OHSU Knight Cancer Institute, University College London and the |
327 | | -University of Manchester, and Cloud TPUs from Google's TPU Research Cloud (TRC). |
| 308 | +This work was supported by the EPSRC grant (EP/T029404/1), the Wellcome/EPSRC |
| 309 | +Centre for Interventional and Surgical Sciences (203145Z/16/Z), the |
| 310 | +International Alliance for Cancer Early Detection, an alliance between Cancer |
| 311 | +Research UK (C28070/A30912, C73666/A31378), Canary Center at Stanford |
| 312 | +University, the University of Cambridge, OHSU Knight Cancer Institute, |
| 313 | +University College London and the University of Manchester, and Cloud TPUs from |
| 314 | +Google’s TPU Research Cloud (TRC). |
0 commit comments