Folia
← 返回头版

探索性建模:从K个候选中训练最优输出

研究人员提出了探索性建模这一新型生成模型范式,通过在训练阶段探索K个可能的输出结果,仅对其中最优匹配进行梯度更新,从而解决生成模型在多模态分布下的平均化问题1。该方法实现了数据效率提升6.2倍、计算效率提升4.1倍、参数效率提升47%1。

在ImageNet 256数据集上,探索性建模达到1.43 FID的性能指标1。随着数据规模增长,性能提升从7%增至36%;随参数规模增长,性能提升从13%增至23%1。在行为克隆任务中,探索性策略仅需1次前向传播便能匹配100步扩散策略的效果1。在世界模型任务上,该方法实现了16-256倍的前向传播次数减少1。

该研究定义了生成表达性的概念,即模型能够捕获的不同答案数量1。这一新范式使生成模型能够实现端到端的训练和推理1。


评论