Efficient Personalized Text-to-image Generation by Leveraging Textual Subspace

TL;DR

This paper introduces an efficient personalized text-to-image generation method that leverages a textual subspace, improving reconstruction fidelity, prompt adaptability, and training efficiency by exploiting the self-expressiveness property.

Contribution

It proposes a novel approach to explore a textual subspace with a basis selection strategy, enhancing personalization and robustness in text-to-image generation.

Findings

Improved alignment with novel textual prompts.

Enhanced robustness to initial word choice.

Faster convergence and training efficiency.

Abstract

Personalized text-to-image generation has attracted unprecedented attention in the recent few years due to its unique capability of generating highly-personalized images via using the input concept dataset and novel textual prompt. However, previous methods solely focus on the performance of the reconstruction task, degrading its ability to combine with different textual prompt. Besides, optimizing in the high-dimensional embedding space usually leads to unnecessary time-consuming training process and slow convergence. To address these issues, we propose an efficient method to explore the target embedding in a textual subspace, drawing inspiration from the self-expressiveness property. Additionally, we propose an efficient selection strategy for determining the basis vectors of the textual subspace. The experimental evaluations demonstrate that the learned embedding can not only faithfully reconstruct input image, but also significantly improves its alignment with novel input textual prompt. Furthermore, we observe that optimizing in the textual subspace leads to an significant improvement of the robustness to the initial word, relaxing the constraint that requires users to input the most relevant initial word. Our method opens the door to more efficient representation learning for personalized text-to-image generation.