Search papers, labs, and topics across Lattice.
This paper introduces UniSpace, a unified visual representation framework that leverages a pretrained semantic Vision Transformer (ViT) to enhance multimodal understanding and image generation. By employing a novel technique called Patch Reparameterization, the authors retain fine-grained visual details that are typically lost in traditional semantic encoders, thus improving pixel reconstruction for sensitive tasks. The resulting 8B Mixture-of-Transformer-Experts model demonstrates significant advancements in text-to-image generation and instruction-based image editing within a single visual space, eliminating the need for separate pathways like VAEs.
A unified visual representation can achieve high-fidelity image reconstruction and multimodal understanding without the need for separate generative pathways.
Semantic vision encoders have become a central visual interface for multimodal understanding and semantic conditioning in image generation. However, their final tokens discard fine-grained visual details, leading to poor pixel reconstruction and limiting their use in reconstruction-sensitive tasks such as image generation and editing. In this work, we ask whether understanding, generation, and editing can be modeled in a single visual representation space built from a pretrained semantic ViT. We show that the frozen Transformer blocks of a semantic ViT are not intrinsically unable to preserve visual details. Instead, the original patch parameterization drives the representation toward semantic abstraction, making fine-grained information difficult to recover from the final tokens. Based on this observation, we introduce Patch Reparameterization, which preserves the original semantic pathway while adding a reconstruction-aware patch embedding that provides fine-grained visual information to the same frozen ViT blocks. The resulting unified representation preserves multimodal understanding while enabling high-fidelity image reconstruction and a favorable reconstruction--generation trade-off. We further scale this representation into UniSpace, an 8B Mixture-of-Transformer-Experts model that performs understanding, generation, and editing in the same visual space without a separate VAE pathway. System-level evaluations demonstrate practical text-to-image generation and instruction-based image editing, showing that a reparameterized pretrained ViT can serve as a unified visual interface for scalable multimodal modeling.
