diffusion size prediction