Policy-Mask：扩散模型特征在判别任务上的预算自适应方法
Policy-Mask: Budget-Aware Adaptation of Diffusion Features to Discriminative Tasks
AI 导读
研究者提出 Policy-Mask，一种在给定通道预算下联合选择扩散模型去噪轨迹中时间步、网络块和通道子集的策略方法。它在 ADE20K 分割上达到 38.9 mIoU，CIFAR-100 在 128 维下达到 75.4% top-1，SPair-71k 上达到 76.3 [email protected]，并已适配 PixArt-$\Sigma$ 和 SD3.5 MMDiT 等扩散 Transformer。