I'm also starting a similar project, but it just involves writing DDPM from scratch. In the past few days I saw some papers regarding diffusion in the medical domain, maybe you can skim through that and see how they are used in that domain.