Policy-Mask: Budget-Aware Adaptation of Diffusion Features to Discriminative Tasks

Ivan Listopadov, Maksim Nakhodnov, Aibek Alanov

Selecting intermediate features from a frozen diffusion model for discriminative tasks is non-trivial because the useful signal is distributed across the denoising trajectory, netw…