Sep, 2023
辅助扩散模型的异常想象
Dream the Impossible: Outlier Imagination with Diffusion Models
Xuefeng Du, Yiyou Sun, Xiaojin Zhu, Yixuan Li
TL;DR使用辅助离群数据集来规范机器学习模型为数据分布外(OOD)检测和安全预测提供了有希望的结果。本文提出了一个名为 DREAM-OOD 的新框架,能够通过扩散模型在像素空间中仅使用分布内(ID)数据和类别想象出逼真的异常值。与先前的工作不同,DREAM-OOD 能够在像素空间中直接可视化和理解想象的异常值,并展示通过 DREAM-OOD 生成的样本训练可以提高 OOD 检测性能。