Question about the beta schedule about ddm2 HOT 4 OPEN

Hi, thanks for your interest in our work! It is expected behavior that less noisy images should be matched with small state numbers (since they are more close to the clean reference). Do you encounter any problem when running Stage3? Can you observe visual improvements between Stage3 results and Stage1 results?

from ddm2.

Thanks for the reply! I do have tried to just use the few matched states for stage 3, but the generated images are not so good. Did you change the beta schedule? I noticed you use beta_0 = 5e-5, beta_T = 1e-2, which seems different from that of DDPM (I remember they are 1e-4 and 2e-2). Does it mean changing the starting and ending of the schedule can make stage 2&3 have larger matched states?

from ddm2.

In our original implementation, as you mentioned, we changed the beta scheduler to have large matched states (more generation steps). Yes, changing beta schedule in your case may help denoise less noisy images.

from ddm2.

Thanks so much! I also plan to change the beta schedule. Can you share any insights about how to modify the beta schedule based on your estimated noise level? Also, I am wondering generally how many steps you will match for your dataset (DWI images). Thanks!

from ddm2.