开源项目

DiffusionOPD:扩散模型在线策略蒸馏的官方实现

ali-vilab/DiffusionOPD

概述

DiffusionOPD 是 ali-vilab 团队发布的官方代码仓库,实现扩散模型的在线策略蒸馏(Online Policy Distillation)方法,并给出完整复现流程,覆盖训练与评估环节。在线策略蒸馏让学生的训练信号来自教师采样过程中的实时反馈,与离线蒸馏相比更贴合扩散模型的迭代生成特性。对做扩散模型小型化与端侧部署的团队有直接参考价值。