Hugging Face · 每日论文· 媒体 / 作者来源·精选
Policy-Mask: 预算感知的扩散特征适应判别任务
原文标题
Policy-Mask: Budget-Aware Adaptation of Diffusion Features to Discriminative Tasks
AI 导读
研究团队提出Policy-Mask方法，通过预算条件策略联合选择时间步、网络块和通道子集，解决了冻结扩散模型特征选择难题。在分割、分类和语义对应任务中，该方法在相同预算下优于固定提取和特征过滤基线，在ADE20K上达到38.9 mIoU，CIFAR-100上达到75.4% top-1准确率，SPair-71k上达到76.3 PCK@0.1。研究发现256个保留通道是实际操作点，并将方法扩展到PixArt-Σ和SD3.5 MMDiT等扩散变压器。
来源：Hugging Face · 每日论文 · arxiv.org
导读与选稿说明
Policy-Mask通过联合优化特征选择参数，显著提升了扩散模型在判别任务中的性能，并确定了实际操作点。
导读由模型辅助整理，非独立核验。完整信息见原文。