Patchwise Generative ConvNet: Training Energy-Based Models From a Single Natural Image for Internal Learning
Zilong Zheng, Jianwen Xie, Ping Li
Abstract
Exploiting internal statistics of a single natural image has long been recognized as a significant research paradigm where the goal is to learn the internal distribution of patches within the image without relying on external training data. Different from prior works that model such a distribution implicitly with a top-down latent variable model (e.g., generator), this paper proposes to explicitly represent the statistical distribution within a single natural image by using an energy-based generative framework, where a pyramid of energy functions, each parameterized by a bottom-up deep neural network, are used to capture the distributions of patches at different resolutions. Meanwhile, a coarse-to-fine sequential training and sampling strategy is presented to train the model efficiently. Besides learning to generate random samples from white noise, the model can learn in parallel with a self-supervised task (e.g., recover the input image from its corrupted version), which can further improve the descriptive power of the learned model. The proposed model is simple and natural in that it does not require an auxiliary model (e.g., discriminator) to assist the training. Besides, it also unifies internal statistics learning and image generation in a single framework. Experimental results presented on various image generation and manipulation tasks, including super-resolution, image editing, harmonization, style transfer, etc, have demonstrated the effectiveness of our model for internal learning.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers10
- Learning Generative Vision Transformer with Energy-Based Latent Space for Saliency PredictionJing Zhang, Jianwen Xie, Nick Barnes, Ping LiNeurIPS 2021 · 117 citations
- SinDDM: A Single Image Denoising Diffusion ModelVladimir Kulikov, Shahar Yadin, Matan Kleiner, Tomer MichaeliICML 2023 · 113 citations
- Single Motion DiffusionSigal Raab, Inbal Leibovitch, Guy Tevet, Moab Arar et al.ICLR 2024 · 81 citations
- A Tale of Two Flows: Cooperative Learning of Langevin Flow and Normalizing Flow Toward Energy-Based ModelJianwen Xie, Yaxuan Zhu, Jun Li, Ping LiICLR 2022 · 53 citations
- PetsGAN: Rethinking Priors for Single Image GenerationZicheng Zhang, Yinglu Liu, Congying Han, Hailin Shi et al.AAAI 2022 · 27 citations
Builds on6
- SinGAN: Learning a Generative Model From a Single Natural ImageTamar Rott Shaham, Tali Dekel, Tomer MichaeliICCV 2019 · 933 citations
- Your classifier is secretly an energy based model and you should treat it like oneWill Grathwohl, Kuan-Chieh Wang, Jörn-Henrik Jacobsen, David Duvenaud et al.ICLR 2020 · 643 citations
- InGAN: Capturing and Retargeting the "DNA" of a Natural ImageAssaf Shocher, Shai Bagon, Phillip Isola, Michal IraniICCV 2019 · 146 citations
- Learning Energy-Based Models by Diffusion Recovery LikelihoodRuiqi Gao, Yang Song, Ben Poole, Ying Nian Wu et al.ICLR 2021 · 144 citations
- Learning Energy-Based Model with Variational Auto-Encoder as Amortized SamplerJianwen Xie, Zilong Zheng, Ping LiAAAI 2021 · 57 citations
Related papers
- SinIR: Efficient General Image Manipulation with Single Image ReconstructionJihyeong Yoo, Qifeng ChenICML 2021 · 25 citations
- Meta Internal LearningRaphael Bensadoun, Shir Gur, Tomer Galanti, Lior WolfNeurIPS 2021 · 8 citations
- Efficient and Training-Free Single-Image Diffusion ModelsHaojun Qiu, Kiriakos N. Kutulakos, David B. LindellCVPR 2026 · 1 citation
- Energy-Inspired Self-Supervised Pretraining for Vision ModelsZe Wang, Jiang Wang, Zicheng Liu, Qiang QiuICLR 2023 · 1 citation
- Learning Probabilistic Models from Generator Latent Spaces with Hat EBMMitch Hill, Erik Nijkamp, Jonathan Mitchell, Bo Pang et al.NeurIPS 2022 · 14 citations
