天大团队构建决策模型代码库，支持多种扩散模型和网络架构

文章预览

近年来，扩散模型一直很火热，在图像生成或视频生成方面，例如，被人们所熟知的 Stable Diffusion、Sora 等应用。 近期，天津大学团队构建了一种面向决策领域的扩散模型代码库 CleanDiffuser。 通过重新审视扩散模型在决策领域中的角色，识别出一系列关键的基本子模块，即扩散模型、网络架构、引导采样方法。 此外，CleanDiffuser 让所有部分都相互解耦，让新算法开发与算法组件消融变得更简单。 CleanDiffuser 的代码和文档已经在 GitHub 网站开源。 “希望更多开发者能够加入，共同将代码库完善，以促进更多科研成果，甚至是落地的机器人应用。”该论文共同第一作者、天津大学博士生董子斌表示。 图丨董子斌（左）和袁逸夫（来源：董子斌、袁逸夫） 日前，相关论文以《CleanDiffuser: 易于使用的决策扩散模型模块化库》（ CleanDiffuser: An Easy-to-use Modulari ………………………………