English

Hyperspherical Autoencoder for High-Fidelity Image Reconstruction and Generation

Computer Vision and Pattern Recognition 2026-05-12 v2 Artificial Intelligence Machine Learning

Abstract

Recent studies have explored using pretrained Vision Foundation Models (VFMs) such as DINO for generative autoencoders, showing strong generative performance. Unfortunately, existing approaches often suffer from limited reconstruction fidelity due to the loss of high-frequency details. In this work, we present the \textbf{\em Hyperspherical Autoencoder (HAE)}, a framework that bridges semantic representation and pixel-level reconstruction. Our key insight is that while semantic information in contrastive representations is primarily directional, enforcing strict magnitude matching hinders the preservation of fine-grained details. To address this, we introduce a {\em Directional Feature Alignment} objective that enforces semantic consistency while allowing flexible feature magnitudes for detail retention, alongside a {\em Hierarchical Convolutional Patch Embedding} module to enhance local structure preservation. Furthermore, observing that SSL-based representations intrinsically lie on a hypersphere, we employ {\em Riemannian Flow Matching} to train a Diffusion Transformer (DiT) directly on this spherical latent manifold. Notably, our manifold-aware DiT exhibits highly efficient convergence, achieving an exceptional gFID of \textbf{1.96} alongside a reconstruction rFID of \textbf{0.78} and a PSNR of \textbf{25.2} dB, validating the advantages of our manifold-aware approach.

Keywords

Cite

@article{arxiv.2601.22904,
  title  = {Hyperspherical Autoencoder for High-Fidelity Image Reconstruction and Generation},
  author = {Hun Chang and Byunghee Cha and Jong Chul Ye},
  journal= {arXiv preprint arXiv:2601.22904},
  year   = {2026}
}

Comments

22 pages, and 20 figures

R2 v1 2026-07-01T09:27:40.531Z