Oct, 2023

预训练扩散模型中不可监督地发现可解释的 h - 空间方向

TL;DR我们提出了第一个无监督的基于学习的方法,用于识别预训练扩散模型的可解释方向。通过联合优化移位控制模块和再构造器,模型可以自主发现解耦和可解释的方向。我们还提出了一种基于梯度检查点技术的通用 VRAM 高效训练算法来解决 GPU VRAM 存储问题。与现有的扩散模型相关工作相比,我们的方法具有全局化和可扩展性的优势,不需要任何复杂的额外步骤,并且广泛实验证明了我们方法的有效性。