基于切片 Wasserstein 引导的颜色条件生成
计算机视觉与模式识别
2026-05-04 v2
摘要
我们提出 SW-Guidance，一种无需训练的方法，用于根据参考图像的颜色分布进行图像条件生成。虽然可以通过先根据文本提示生成图像再应用颜色风格迁移方法来生成具有固定颜色的图像，但这种方法往往会导致生成图像中出现语义上无意义的颜色。我们的方法通过修改扩散模型的采样过程，将生成图像的颜色分布与参考调色板之间的可微切片 1-Wasserstein 距离纳入其中，从而解决了这一问题。我们的方法在颜色与参考的相似度方面优于颜色条件生成的现有最先进技术，所生成的图像不仅匹配参考颜色，还保持了与原始文本提示的语义一致性。源代码可在 https://github.com/alobashev/sw-guidance/ 获取。
引用
@article{arxiv.2503.19034,
title = {Color Conditional Generation with Sliced Wasserstein Guidance},
author = {Alexander Lobashev and Maria Larchenko and Dmitry Guskov},
journal= {arXiv preprint arXiv:2503.19034},
year = {2026}
}
备注
NeurIPS 2025, spotlight