TAG diffusion model