We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: RG-Flow: A hierarchical and explainable flow model based on renormalization group and sparse prior

Abstract: Flow-based generative models have become an important class of unsupervised learning approaches. In this work, we incorporate the key ideas of renormalization group (RG) and sparse prior distribution to design a hierarchical flow-based generative model, RG-Flow, which can separate information at different scales of images and extract disentangled representations at each scale. We demonstrate our method on synthetic multi-scale image datasets and the CelebA dataset, showing that the disentangled representations enable semantic manipulation and style mixing of the images at different scales. To visualize the latent representations, we introduce receptive fields for flow-based models and show that the receptive fields of RG-Flow are similar to those of convolutional neural networks. In addition, we replace the widely adopted isotropic Gaussian prior distribution by the sparse Laplacian distribution to further enhance the disentanglement of representations. From a theoretical perspective, our proposed method has $O(\log L)$ complexity for inpainting of an image with edge length $L$, compared to previous generative models with $O(L^2)$ complexity.
Comments: 31 pages, 20 figures, 3 tables
Subjects: Machine Learning (cs.LG); Disordered Systems and Neural Networks (cond-mat.dis-nn); Artificial Intelligence (cs.AI); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (stat.ML)
Journal reference: Mach. Learn.: Sci. Technol. 3 035009 (2022)
DOI: 10.1088/2632-2153/ac8393
Cite as: arXiv:2010.00029 [cs.LG]
  (or arXiv:2010.00029v5 [cs.LG] for this version)

Submission history

From: Dian Wu [view email]
[v1] Wed, 30 Sep 2020 18:04:04 GMT (18742kb,D)
[v2] Fri, 2 Oct 2020 18:27:26 GMT (18898kb,D)
[v3] Wed, 7 Oct 2020 00:59:10 GMT (18905kb,D)
[v4] Fri, 18 Dec 2020 20:27:11 GMT (11165kb,D)
[v5] Mon, 15 Aug 2022 09:50:27 GMT (6792kb,D)

Link back to: arXiv, form interface, contact.