Extending One-Step Image Generation from Class Labels to Text via Discriminative Text Representation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhao, Chenxi, Zhu, Chen, Feng, Xiaokun, Hao, Aiming, Zhu, Jiashu, Lei, Jiachen, Wu, Jiahong, Chu, Xiangxiang, Yang, Jufeng
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918456793759744
author Zhao, Chenxi
Zhu, Chen
Feng, Xiaokun
Hao, Aiming
Zhu, Jiashu
Lei, Jiachen
Wu, Jiahong
Chu, Xiangxiang
Yang, Jufeng
author_facet Zhao, Chenxi
Zhu, Chen
Feng, Xiaokun
Hao, Aiming
Zhu, Jiashu
Lei, Jiachen
Wu, Jiahong
Chu, Xiangxiang
Yang, Jufeng
contents Few-step generation has been a long-standing goal, with recent one-step generation methods exemplified by MeanFlow achieving remarkable results. Existing research on MeanFlow primarily focuses on class-to-image generation. However, an intuitive yet unexplored direction is to extend the condition from fixed class labels to flexible text inputs, enabling richer content creation. Compared to the limited class labels, text conditions pose greater challenges to the model's understanding capability, necessitating the effective integration of powerful text encoders into the MeanFlow framework. Surprisingly, although incorporating text conditions appears straightforward, we find that integrating powerful LLM-based text encoders using conventional training strategies results in unsatisfactory performance. To uncover the underlying cause, we conduct detailed analyses and reveal that, due to the extremely limited number of refinement steps in the MeanFlow generation, such as only one step, the text feature representations are required to possess sufficiently high discriminability. This also explains why discrete and easily distinguishable class features perform well within the MeanFlow framework. Guided by these insights, we leverage a powerful LLM-based text encoder validated to possess the required semantic properties and adapt the MeanFlow generation process to this framework, resulting in efficient text-conditioned synthesis for the first time. Furthermore, we validate our approach on the widely used diffusion model, demonstrating significant generation performance improvements. We hope this work provides a general and practical reference for future research on text-conditioned MeanFlow generation. The code is available at https://github.com/AMAP-ML/EMF.
format Preprint
id arxiv_https___arxiv_org_abs_2604_18168
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Extending One-Step Image Generation from Class Labels to Text via Discriminative Text Representation
Zhao, Chenxi
Zhu, Chen
Feng, Xiaokun
Hao, Aiming
Zhu, Jiashu
Lei, Jiachen
Wu, Jiahong
Chu, Xiangxiang
Yang, Jufeng
Computer Vision and Pattern Recognition
Few-step generation has been a long-standing goal, with recent one-step generation methods exemplified by MeanFlow achieving remarkable results. Existing research on MeanFlow primarily focuses on class-to-image generation. However, an intuitive yet unexplored direction is to extend the condition from fixed class labels to flexible text inputs, enabling richer content creation. Compared to the limited class labels, text conditions pose greater challenges to the model's understanding capability, necessitating the effective integration of powerful text encoders into the MeanFlow framework. Surprisingly, although incorporating text conditions appears straightforward, we find that integrating powerful LLM-based text encoders using conventional training strategies results in unsatisfactory performance. To uncover the underlying cause, we conduct detailed analyses and reveal that, due to the extremely limited number of refinement steps in the MeanFlow generation, such as only one step, the text feature representations are required to possess sufficiently high discriminability. This also explains why discrete and easily distinguishable class features perform well within the MeanFlow framework. Guided by these insights, we leverage a powerful LLM-based text encoder validated to possess the required semantic properties and adapt the MeanFlow generation process to this framework, resulting in efficient text-conditioned synthesis for the first time. Furthermore, we validate our approach on the widely used diffusion model, demonstrating significant generation performance improvements. We hope this work provides a general and practical reference for future research on text-conditioned MeanFlow generation. The code is available at https://github.com/AMAP-ML/EMF.
title Extending One-Step Image Generation from Class Labels to Text via Discriminative Text Representation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2604.18168