paper-with-me

홈 › Papers

Diffusion Model Compression for Image-to-Image Translation

2024-01-31 · Geonung Kim, Beomsu Kim, Eunhyeok Park, Sunghyun Cho

As recent advances in large-scale Text-to-Image (T2I) diffusion models have yielded remarkable high-quality image generation, diverse downstream Image-to-Image (I2I) applications have emerged. Despite the impressive results achieved by these I2I models, their practical utility is hampered by their large model size and the computational burden of the iterative denoising process. In this paper, we propose a novel compression method tailored for diffusion-based I2I models. Based on the observations that the image conditions of I2I models already provide rich information on image structures, and that the time steps with a larger impact tend to be biased, we develop surprisingly simple yet effective approaches for reducing the model size and latency. We validate the effectiveness of our method on three representative I2I tasks: InstructPix2Pix for image editing, StableSR for image restoration, and ControlNet for image-conditional image generation. Our approach achieves satisfactory output quality with 39.2%, 56.4% and 39.2% reduction in model footprint, as well as 81.4%, 68.7% and 31.1% decrease in latency to InstructPix2Pix, StableSR and ControlNet, respectively.

📄 PDF Abstract BibTeX arXiv:2401.17547

Code (0)

등록된 구현이 없습니다.

Tasks

Conditional Image GenerationDenoisingImage GenerationImage RestorationImage-to-Image TranslationmodelModel CompressionTranslation

Methods 이 논문이 사용한 방법론

Diffusion Diffusion models generate samples by gradually removing noise from a signal, and their training objective can be expressed as a reweighted variational lower-bound…

Similar Papers 제목 키워드 기반

DiffHarmony: Latent Diffusion Model Meets Image Harmonization

2024-04-09 · Pengfei Zhou, Fangxiang Feng, Xiaojie Wang

Image harmonization, which involves adjusting the foreground of a composite image to attain a unified visual consistency with the background, can be conceptualized as an image-to-image translation task. Diffusion models …

Image CompressionImage GenerationImage HarmonizationImage-to-Image Translation+2

PPDM: Pixel Puzzling Diffusion Model for Speed and Memory Efficient Volumetric Medical Image Translation

2026-06-13 · Tianqi Chen, Jun Hou, Yinchi Zhou, James S. Duncan 외 arxiv

Diffusion models have demonstrated superior fidelity for medical image-to-image translation, but their extension to high-resolution 3D volumes is severely constrained by prohibitive computational cost and GPU memory requ…

Image-to-Image Translation

A Novel Framework for Image-to-image Translation and Image Compression

2021-11-25 · Fei Yang, Yaxing Wang, Luis Herranz, Yongmei Cheng 외

Data-driven paradigms using machine learning are becoming ubiquitous in image processing and communications. In particular, image-to-image (I2I) translation is a generic and widely used approach to image processing probl…

Image CompressionImage GenerationImage RestorationImage-to-Image Translation+3

A Noise Constrained Diffusion (NC-Diffusion) Framework for High Fidelity Image Compression

2026-04-08 · Zhenyu Du, Yanbo Gao, Shuai Li, Yiyang Li 외 arxiv

With the great success of diffusion models in image generation, diffusion-based image compression is attracting increasing interests. However, due to the random noise introduced in the diffusion learning, they usually pr…

Image CompressionImage Generation

Addressing Detail Bottlenecks in Latent Diffusion for RGB-to-SWIR Image Translation

2026-06-18 · Kaili Wang, Martin Dimitrievski, Jose Maria Salvador, Ben Stoffelen 외 arxiv

Latent diffusion models (LDMs) enable efficient image-to-image translation but discard fine spatial details during compression, degrading downstream perception tasks. We identify two bottlenecks: the autoencoder, which l…

Image-to-Image Translation