改进3D生成的梯度修正方法,提升图像质量与训练稳定性。
Advancing Text-to-3D Generation with Linearized Lookahead Variational Score Distillation
- 引入线性化前瞻变分得分蒸馏,优化模型更新顺序。
- 在LLFF和DTU数据集上显著提升3D生成质量,优于现有方法。
- 兼容主流文本生成3D框架,适合研究高保真3D生成的团队。
基于预训练2D扩散模型的得分蒸馏进行文本到3D生成已受到广泛关注,其中变分得分蒸馏(VSD)是一个代表性方法。VSD通过引入一个额外的基于得分的模型来修正蒸馏梯度,从而改进原始得分蒸馏。然而,在实际应用中,VSD常面临收敛缓慢甚至不稳定的难题。本文深入分析了引入的得分模型与3D模型之间的相互作用,发现实际实现中LoRA与3D分布存在不匹配问题。通过调整两者的优化顺序,使得分模型能够前瞻当前3D状态,从而产生更合理的修正。但朴素的前瞻VSD仍可能因过拟合导致训练不稳定。为此,我们提出使用模型的线性化变体进行得分蒸馏,即线性化前瞻变分得分蒸馏(L²-VSD)。该方法可利用现有深度学习库的前向模式自动微分高效实现。大量实验验证了L²-VSD的有效性,其在生成质量上明显优于先前基于得分蒸馏的方法。此外,我们的方法可无缝集成到任何其他基于VSD的文本到3D生成框架中。
原文摘要 · Abstract (English)
Text-to-3D generation based on score distillation of pre-trained 2D diffusion models has gained increasing interest, with variational score distillation (VSD) as a remarkable example. VSD proves that vanilla score distillation can be improved by introducing an extra score-based model, which characterizes the distribution of images rendered from 3D models, to correct the distillation gradient. Despite the theoretical foundations, VSD, in practice, is likely to suffer from slow and sometimes ill-posed convergence. In this paper, we perform an in-depth investigation of the interplay between the introduced score model and the 3D model, and find that there exists a mismatching problem between LoRA and 3D distributions in practical implementation. We can simply adjust their optimization order to improve the generation quality. By doing so, the score model looks ahead to the current 3D state and hence yields more reasonable corrections. Nevertheless, naive lookahead VSD may suffer from unstable training in practice due to the potential over-fitting. To address this, we propose to use a linearized variant of the model for score distillation, giving rise to the Linearized Lookahead Variational Score Distillation ($L^2$-VSD). $L^2$-VSD can be realized efficiently with forward-mode autodiff functionalities of existing deep learning libraries. Extensive experiments validate the efficacy of $L^2$-VSD, revealing its clear superiority over prior score distillation-based methods. We also show that our method can be seamlessly incorporated into any other VSD-based text-to-3D framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。