Hyper-spherical Optimal Transport for Semantic Alignment in Text-to-3D End-to-end Generation.

Journal: IEEE transactions on visualization and computer graphics
Published Date:

Abstract

Recent CLIP-guided 3D generation methods have achieved promising results but struggle with generating faithful 3D shapes that conform with input text due to the gap between text and image embeddings. To this end, this paper proposes HOTS3D which makes the first attempt to effectively bridge this gap by aligning text features to the image features with spherical optimal transport (SOT). However, in high-dimensional situations, solving the SOT remains a challenge. To obtain the SOT map for high-dimensional features obtained from CLIP encoding of two modalities, we mathematically formulate and derive the solution based on Villani's theorem, which can directly align two hyper-sphere distributions without manifold exponential maps. Furthermore, we implement it by leveraging input convex neural networks (ICNNs) for the optimal Kantorovich potential. With the optimally mapped features, a diffusion-based generator is utilized to decode them into 3D shapes. Extensive quantitative and qualitative comparisons with state-of-the-art methods demonstrate the superiority of HOTS3D for text-to-3D generation, especially in the consistency with text semantics. The code will be publicly available.

Authors

  • Zezeng Li
  • Weimin Wang
    Institute of Health Management, Chinese People's Liberation Army (PLA) General Hospital, Beijing, China.
  • Yuming Zhao
  • Wenhai Li
  • Na Lei
    Dept of Soft and Tech, Dalian Univ of Tech, Dalian, PR China.
  • Xianfeng Gu
    Dept of Comp Sci, Stony Brook Univ, Stony Brook, USA.

Keywords

No keywords available for this article.