Dec, 2023

SteinDreamer:通过 Stein 标识进行文本至 3D 分数提炼的方差减少

TL;DR通过方差约减的视角,本文揭示了分数蒸馏中的梯度估计与高方差之间的内在关系,并提出使用 Stein's identity 进行变量约减的方法,称为 Stein Score Distillation (SSD)。通过构建由 Stein identity 构造的控制变量,SSD 可以灵活地引入指导先验和网络结构以明确优化方差约减,实验结果表明 SSD 可以有效地减少蒸馏方差,提高生成视觉质量。同时,我们证明了由于更稳定的梯度更新,SteinDreamer 相比现有方法更快地收敛。