Saved in:
Bibliographic Details
Main Authors: Huang, Paul Kuo-Ming, Chen, Si-An, Lin, Hsuan-Tien
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2307.04081
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914670167719936
author Huang, Paul Kuo-Ming
Chen, Si-An
Lin, Hsuan-Tien
author_facet Huang, Paul Kuo-Ming
Chen, Si-An
Lin, Hsuan-Tien
contents Score-based generative models (SGMs) are a popular family of deep generative models that achieve leading image generation quality. Early studies extend SGMs to tackle class-conditional generation by coupling an unconditional SGM with the guidance of a trained classifier. Nevertheless, such classifier-guided SGMs do not always achieve accurate conditional generation, especially when trained with fewer labeled data. We argue that the problem is rooted in the classifier's tendency to overfit without coordinating with the underlying unconditional distribution. To make the classifier respect the unconditional distribution, we propose improving classifier-guided SGMs by letting the classifier regularize itself. The key idea of our proposed method is to use principles from energy-based models to convert the classifier into another view of the unconditional SGM. Existing losses for unconditional SGMs can then be leveraged to achieve regularization by calibrating the classifier's internal unconditional scores. The regularization scheme can be applied to not only the labeled data but also unlabeled ones to further improve the classifier. Across various percentages of fewer labeled data, empirical results show that the proposed approach significantly enhances conditional generation quality. The enhancements confirm the potential of the proposed self-calibration technique for generative modeling with limited labeled data.
format Preprint
id arxiv_https___arxiv_org_abs_2307_04081
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Score-based Conditional Generation with Fewer Labeled Data by Self-calibrating Classifier Guidance
Huang, Paul Kuo-Ming
Chen, Si-An
Lin, Hsuan-Tien
Computer Vision and Pattern Recognition
Machine Learning
Score-based generative models (SGMs) are a popular family of deep generative models that achieve leading image generation quality. Early studies extend SGMs to tackle class-conditional generation by coupling an unconditional SGM with the guidance of a trained classifier. Nevertheless, such classifier-guided SGMs do not always achieve accurate conditional generation, especially when trained with fewer labeled data. We argue that the problem is rooted in the classifier's tendency to overfit without coordinating with the underlying unconditional distribution. To make the classifier respect the unconditional distribution, we propose improving classifier-guided SGMs by letting the classifier regularize itself. The key idea of our proposed method is to use principles from energy-based models to convert the classifier into another view of the unconditional SGM. Existing losses for unconditional SGMs can then be leveraged to achieve regularization by calibrating the classifier's internal unconditional scores. The regularization scheme can be applied to not only the labeled data but also unlabeled ones to further improve the classifier. Across various percentages of fewer labeled data, empirical results show that the proposed approach significantly enhances conditional generation quality. The enhancements confirm the potential of the proposed self-calibration technique for generative modeling with limited labeled data.
title Score-based Conditional Generation with Fewer Labeled Data by Self-calibrating Classifier Guidance
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2307.04081