Abstract
Text-to-image users often provide concise and underspecified prompts, whereas generative models benefit from detailed textual conditions for reliable instruction following. Existing systems bridge this gap with Prompt Enhancers (PEs) that rewrite raw prompts at inference time, introducing additional latency and leaving prompt elaboration external to the generator. We instead view enhanced prompts as privileged training information and ask whether their benefits can be internalized. We propose Prompt-Enhanced On-Policy Self-Distillation (PE-OPSD) for text-to-image flow-matching models. During training, a raw-prompt student follows its own generation trajectory, while an enhanced-prompt teacher provides vector-field targets at the states visited by the student. This on-policy supervision distills the behavior induced by enhanced prompts into the raw-prompt student without requiring additional text--image pairs. At inference, both the PE and teacher are removed, and the student generates directly from raw prompts. Across multiple model families, PEs, and benchmarks, PE-OPSD achieves the strongest aggregate prompt fidelity among the evaluated baselines, yields positive aggregate visual appeal gains, and retains the base-model inference efficiency.
Keywords
Subject
Publication details
- Journal
- Not available
- Open access
- Green open access
Cite this article
APA 7
Lin, M., Cai, C., Xu, L., Ma, C., Wei, Y., & Han, L. (2026). PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation. https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation
MLA 9
Lin, Mingfeng, et al. "PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation." https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation.
Chicago (author–date)
Lin, Mingfeng, Chengfei Cai, Lin Xu, Chengqian Ma, Yuxiang Wei, and Liang Han. 2026. "PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation." https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation.
Harvard
Lin, M., Cai, C., Xu, L., Ma, C., Wei, Y. and Han, L. (2026) 'PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation', Available at: https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation.
Vancouver
Lin M, Cai C, Xu L, Ma C, Wei Y, Han L. PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation. https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation
IEEE
M. Lin, C. Cai, L. Xu, C. Ma, Y. Wei, and L. Han, "PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation," https://omanscience.com/en/articles/pe-opsd-internalizing-prompt-enhancement-into-flow-matching-models-via-on-policy-self-distillation.