search space is the same as diffusion