Details about the code for the sample strategy

wenhaomin / DiffSTG

Code for DiffSTG

MIT License

54 stars 10 forks source link

Details about the code for the sample strategy #3

Closed Wanghkkkkkkk closed 5 months ago

Wanghkkkkkkk commented 5 months ago

Hi, @wenhaomin I was studying your code and noticed that you use three methods in the sample_strategy, namely 'ddim_multi', 'ddim_one' and 'ddpm'. I would like to ask which method is used in the paper? Looking forward to your reply, thank you!

Wanghkkkkkkk commented 5 months ago

And why use ‘ddpm’ when training and ‘ddim_multi’ when testing in code?

wenhaomin commented 5 months ago

Hi, thanks for the question. ddim_multi is used in the method. For sample strategy, to estimate the data distribution from a trained model, i.e., getting S samples.

ddpm needs to samling S*N diffusion steps;
ddim needs to sample S * M (M < N) diffusion steps;
ddim_multi need (S/k) * M diffusion steps, where k is the number of samples used in one reverse process.

So ddim_multi can be used in test for its sampling speed is much faster.