Diffusion Adaptive Text Embedding for Text-to-Image Diffusion Models

HeeSun Bae (KAIST) · Byeonghu Na (KAIST) · Il-chul Moon (Korea Advanced Institute of Science and Technology) · Se Jung Kwon (NAVER Cloud) · Minsang Park (Korea Advanced Institute of Science & Technology) · Gyuwon Sim (Korea Advanced Institute of Science & Technology) · Donghyeok Shin (Korea Advanced Institute of Science and Technology) · Mina Kang (KAIST) · Wanmo Kang (KAIST)
alignmentdiffusion adaptive text embeddingdiffusion samplingdynamic updatesempirical resultsgenerative processmulti-concept generationoptimization problempre-trained text encoderpreferencereverse-diffused imagestext embeddingstext-guided image editingtext-to-image diffusion modelstheoretical analysisupdate rule

Text-to-image diffusion models rely on text embeddings from a pre-trained text encoder, but these embeddings remain fixed across all diffusion timesteps, limiting their adaptability to the generative process. We propose Diffusion Adaptive Text Embedding (DATE), which dynamically updates text embeddings at each diffusion timestep based on intermediate perturbed data. We formulate an optimization problem and derive an update rule that refines the text embeddings at each sampling step to improve alignment and preference between the mean predicted image and the text. This allows DATE to dynamically adapts the text conditions to the reverse-diffused images throughout diffusion sampling without requiring additional model training. Through theoretical analysis and empirical results, we show that DATE maintains the generative capability of the model while providing superior text-image alignment over fixed text embeddings across various tasks, including multi-concept generation and text-guided image editing. Our code is available at https://github.com/aailab-kaist/DATE.