Breaking the Likelihood-Quality Trade-off in Diffusion Models by Merging Pretrained Experts

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Esfandiari, Yasin, Bauer, Stefan, Stich, Sebastian U., Dittadi, Andrea
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911283970834432
author Esfandiari, Yasin
Bauer, Stefan
Stich, Sebastian U.
Dittadi, Andrea
author_facet Esfandiari, Yasin
Bauer, Stefan
Stich, Sebastian U.
Dittadi, Andrea
contents Diffusion models for image generation often exhibit a trade-off between perceptual sample quality and data likelihood: training objectives emphasizing high-noise denoising steps yield realistic images but poor likelihoods, whereas likelihood-oriented training overweights low-noise steps and harms visual fidelity. We introduce a simple plug-and-play sampling method that combines two pretrained diffusion experts by switching between them along the denoising trajectory. Specifically, we apply an image-quality expert at high noise levels to shape global structure, then switch to a likelihood expert at low noise levels to refine pixel statistics. The approach requires no retraining or fine-tuning -- only the choice of an intermediate switching step. On CIFAR-10 and ImageNet32, the merged model consistently matches or outperforms its base components, improving or preserving both likelihood and sample quality relative to each expert alone. These results demonstrate that expert switching across noise levels is an effective way to break the likelihood-quality trade-off in image diffusion models.
format Preprint
id arxiv_https___arxiv_org_abs_2511_19434
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Breaking the Likelihood-Quality Trade-off in Diffusion Models by Merging Pretrained Experts
Esfandiari, Yasin
Bauer, Stefan
Stich, Sebastian U.
Dittadi, Andrea
Computer Vision and Pattern Recognition
Machine Learning
Diffusion models for image generation often exhibit a trade-off between perceptual sample quality and data likelihood: training objectives emphasizing high-noise denoising steps yield realistic images but poor likelihoods, whereas likelihood-oriented training overweights low-noise steps and harms visual fidelity. We introduce a simple plug-and-play sampling method that combines two pretrained diffusion experts by switching between them along the denoising trajectory. Specifically, we apply an image-quality expert at high noise levels to shape global structure, then switch to a likelihood expert at low noise levels to refine pixel statistics. The approach requires no retraining or fine-tuning -- only the choice of an intermediate switching step. On CIFAR-10 and ImageNet32, the merged model consistently matches or outperforms its base components, improving or preserving both likelihood and sample quality relative to each expert alone. These results demonstrate that expert switching across noise levels is an effective way to break the likelihood-quality trade-off in image diffusion models.
title Breaking the Likelihood-Quality Trade-off in Diffusion Models by Merging Pretrained Experts
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2511.19434