Robust Latent Matters: Boosting Image Generation with Sampling Error Synthesis

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qiu, Kai, Li, Xiang, Kuen, Jason, Chen, Hao, Xu, Xiaohao, Gu, Jiuxiang, Luo, Yinyi, Raj, Bhiksha, Lin, Zhe, Savvides, Marios
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929762062041088
author Qiu, Kai
Li, Xiang
Kuen, Jason
Chen, Hao
Xu, Xiaohao
Gu, Jiuxiang
Luo, Yinyi
Raj, Bhiksha
Lin, Zhe
Savvides, Marios
author_facet Qiu, Kai
Li, Xiang
Kuen, Jason
Chen, Hao
Xu, Xiaohao
Gu, Jiuxiang
Luo, Yinyi
Raj, Bhiksha
Lin, Zhe
Savvides, Marios
contents Recent image generation schemes typically capture image distribution in a pre-constructed latent space relying on a frozen image tokenizer. Though the performance of tokenizer plays an essential role to the successful generation, its current evaluation metrics (e.g. rFID) fail to precisely assess the tokenizer and correlate its performance to the generation quality (e.g. gFID). In this paper, we comprehensively analyze the reason for the discrepancy of reconstruction and generation qualities in a discrete latent space, and, from which, we propose a novel plug-and-play tokenizer training scheme to facilitate latent space construction. Specifically, a latent perturbation approach is proposed to simulate sampling noises, i.e., the unexpected tokens sampled, from the generative process. With the latent perturbation, we further propose (1) a novel tokenizer evaluation metric, i.e., pFID, which successfully correlates the tokenizer performance to generation quality and (2) a plug-and-play tokenizer training scheme, which significantly enhances the robustness of tokenizer thus boosting the generation quality and convergence speed. Extensive benchmarking are conducted with 11 advanced discrete image tokenizers with 2 autoregressive generation models to validate our approach. The tokenizer trained with our proposed latent perturbation achieve a notable 1.60 gFID with classifier-free guidance (CFG) and 3.45 gFID without CFG with a $\sim$400M generator. Code: https://github.com/lxa9867/ImageFolder.
format Preprint
id arxiv_https___arxiv_org_abs_2503_08354
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Robust Latent Matters: Boosting Image Generation with Sampling Error Synthesis
Qiu, Kai
Li, Xiang
Kuen, Jason
Chen, Hao
Xu, Xiaohao
Gu, Jiuxiang
Luo, Yinyi
Raj, Bhiksha
Lin, Zhe
Savvides, Marios
Computer Vision and Pattern Recognition
Artificial Intelligence
Recent image generation schemes typically capture image distribution in a pre-constructed latent space relying on a frozen image tokenizer. Though the performance of tokenizer plays an essential role to the successful generation, its current evaluation metrics (e.g. rFID) fail to precisely assess the tokenizer and correlate its performance to the generation quality (e.g. gFID). In this paper, we comprehensively analyze the reason for the discrepancy of reconstruction and generation qualities in a discrete latent space, and, from which, we propose a novel plug-and-play tokenizer training scheme to facilitate latent space construction. Specifically, a latent perturbation approach is proposed to simulate sampling noises, i.e., the unexpected tokens sampled, from the generative process. With the latent perturbation, we further propose (1) a novel tokenizer evaluation metric, i.e., pFID, which successfully correlates the tokenizer performance to generation quality and (2) a plug-and-play tokenizer training scheme, which significantly enhances the robustness of tokenizer thus boosting the generation quality and convergence speed. Extensive benchmarking are conducted with 11 advanced discrete image tokenizers with 2 autoregressive generation models to validate our approach. The tokenizer trained with our proposed latent perturbation achieve a notable 1.60 gFID with classifier-free guidance (CFG) and 3.45 gFID without CFG with a $\sim$400M generator. Code: https://github.com/lxa9867/ImageFolder.
title Robust Latent Matters: Boosting Image Generation with Sampling Error Synthesis
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2503.08354