Papers › Wavelet Transform-assisted Adaptive Generative Modeling for Colorization

Wavelet Transform-assisted Adaptive Generative Modeling for Colorization

9 Jul 2021arXiv:2107.04261archive 2025-07-28

Jin Li, Wanyun Li, Zichen Xu, Yuhao Wang, Qiegen Liu

Unsupervised deep learning has recently demonstrated the promise of producing high-quality samples. While it has tremendous potential to promote the image colorization task, the performance is limited owing to the high-dimension of data manifold and model capability. This study presents a novel scheme that exploits the score-based generative model in wavelet domain to address the issues. By taking advantage of the multi-scale and multi-channel representation via wavelet transform, the proposed model learns the richer priors from stacked coarse and detailed wavelet coefficient components jointly and effectively. This strategy also reduces the dimension of the original manifold and alleviates the curse of dimensionality, which is beneficial for estimation and sampling. Moreover, dual consistency terms in the wavelet domain, namely data-consistency and structure-consistency are devised to leverage colorization task better. Specifically, in the training phase, a set of multi-channel tensors consisting of wavelet coefficients is used as the input to train the network with denoising score matching. In the inference phase, samples are iteratively generated via annealed Langevin dynamics with data and structure consistencies. Experiments demonstrated remarkable improvements of the proposed method on both generation and colorization quality, particularly in colorization robustness and diversity.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

yqx7150/WACM officialmentioned in paperpytorch report
yqx7150/JGM mentioned on GitHubpytorch report
yqx7150/hgm mentioned on GitHubpytorch report
yqx7150/ugm-pi mentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

ColorizationDenoisingDiversityImage Colorization

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

Colorization

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections