arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2301.13622cs.LGcs.CVstat.ML
收藏

使用联合扩散模型学习数据表示

Learning Data Representations with Joint Diffusion Models

Warsaw University of Technology(华沙理工大学);IDEAS NCBR(IDEAS国家生物医学研究中心);Eindhoven University of Technology(埃因霍温理工大学)

Kamil Deja, Tomasz Trzcinski, Jakub M. Tomczak

AI总结:

本文提出一种联合扩散模型,通过引入分类器实现共享参数化的稳定端到端训练,在分类和生成质量上超越现有混合方法,并支持视觉反事实解释。

摘要:

允许合成和分类数据的联合机器学习模型通常在这两项任务之间表现不均,或者训练不稳定。在本研究中,我们从一组经验观察出发,这些观察表明当代基于深度扩散的生成模型构建的内部表示不仅对生成有用,对预测也有用。然后,我们提出用分类器扩展原始扩散模型,从而在两个目标之间通过共享参数化实现稳定的联合端到端训练。由此产生的联合扩散模型在所有评估基准上的分类和生成质量方面均优于近期最先进的混合方法。在我们的联合训练方法基础上,我们通过引入视觉反事实解释方法,展示了如何直接从共享的生成和判别表示中获益。

英文摘要:

Joint machine learning models that allow synthesizing and classifying data often offer uneven performance between those tasks or are unstable to train. In this work, we depart from a set of empirical observations that indicate the usefulness of internal representations built by contemporary deep diffusion-based generative models not only for generating but also predicting. We then propose to extend the vanilla diffusion model with a classifier that allows for stable joint end-to-end training with shared parameterization between those objectives. The resulting joint diffusion model outperforms recent state-of-the-art hybrid methods in terms of both classification and generation quality on all evaluated benchmarks. On top of our joint training approach, we present how we can directly benefit from shared generative and discriminative representations by introducing a method for visual counterfactual explanations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑