论物理启发的生成模型
OriginalOn Physics-Inspired Generative Models
扩散模型等物理启发生成模型训练稳定、容量大,但仍有改进空间。
- 发布时间
- 2024-05
内容介绍
扩散模型如今是生成模型里最有存在感的一支,它的底气来自两处:训练过程相对稳定,模型容量足够大。这份博士论文接着追问的是下一步——在这个家族已经成立的前提下,还有哪些地方可以改,该从哪里改。作者的判断是改进空间依然不少,并且值得从「物理启发」这个建模视角本身入手,而不是在模型外围添加模块。把「还能怎么改」当成一个正面问题来处理,而不是再提一个新模型来填充论文篇幅,是这份资料值得翻一翻的理由。
这是麻省理工学院(MIT)的一篇博士论文,作者 Xu, Yilun,导师为 Tommi S. Jaakkola。从摘要与要点看,论文覆盖的范围是物理启发的生成模型这一族,以扩散模型为代表,分析线索主要有两条:训练目标怎么写,以及隐变量结构怎么设计。作者在摘要里说明会先深入这一族模型的内部机制,再逐步展开后续工作,整体按「先理解、再改造」的顺序组织。需要说明的是,手头材料只包含摘要与要点,因此下面谈到的方法都停留在思路层面,具体的实验设置与对比结果并不在其中。
第一项工作针对训练目标。去噪分数匹配(denoising score matching)是扩散模型训练的常用目标,作者提出了它的一个推广形式,关键的观察落在方差上:在多模态数据上,这一训练目标的方差可以被显著降低。解读一下,训练目标的方差直接对应梯度的抖动程度,方差降下来意味着同样的训练过程里噪声成分更少、信号更集中;而多模态数据恰恰是方差最容易失控的场景,因为模型要在彼此分离的数据区域上都给出可靠的估计。换句话说,这项改进针对的是「数据本身分成几块」时训练信号被稀释的问题。
第二项工作走的是结构路线:把可学习的离散隐变量与连续扩散结合起来,形成一个训练框架。按通常的理解,这类设计是让离散部分承担「选哪一种模式」的职责,连续部分负责模式内部的细节刻画,分工之后模型不必用一套连续分布硬扛所有的多模态结构。换个角度看,这与前一项工作面对的是同一个困难的两种解法:一个从优化目标入手压低方差,一个从表示结构入手减少目标需要同时兼顾的东西。两条线索合在一起,构成了论文从训练目标到模型结构的主干。
这些结论的意味在于,物理启发生成模型的改进可以发生在训练目标与表示结构这两个层面,而不必以换掉整个模型家族为前提。进一步说,如果训练目标的方差是可降的,那么「多模态数据难训」就不完全是数据的问题,也有一部分出在目标函数的设计上;如果隐变量结构可以学,那么模式之间的切换就不必全靠连续分布去逼近。实践上,方差更低的训练目标对数据分布本身多模态的任务更有价值,因为这类任务最容易出现训练不稳;离散隐变量加连续扩散的框架,则为「有结构地生成」提供了一条可选路径。两者都不承诺替换现有方法,更像给同一族模型补上两个可以单独取用的零件。
适合读的是正在做扩散模型训练、并且已经不满足于调参的研究者与工程师;读完能拿走两个具体方向——训练目标的推广形式,以及离散隐变量与连续扩散耦合的框架思路。边界也要讲清楚:这是博士论文,贡献是论文尺度的,不是对整个扩散模型领域的综述;手头材料只给到摘要与要点,所以无法据此判断各项改进的实际幅度、适用的数据规模和失效情形。要评估这些方法是否值得迁移到自己的任务上,仍然得回到论文正文的实验部分去看。