arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.22904cs.CVcs.AIcs.LG

超球面自编码器用于高保真图像重建与生成

Hyperspherical Autoencoder for High-Fidelity Image Reconstruction and Generation

KAIST AI(韩国科学技术院人工智能实验室)

Hun Chang, Byunghee Cha, Jong Chul Ye

AI总结:

本文提出超球面自编码器,通过方向特征对齐和层次卷积补丁嵌入模块,提升图像重建和生成的保真度,实验结果显示其在gFID、rFID和PSNR指标上均优于现有方法。

摘要:

近期研究探讨了使用预训练视觉基础模型(如DINO)进行生成自编码器,显示出强大的生成性能。然而,现有方法常因高频细节丢失而重建保真度有限。本文提出超球面自编码器(HAE),通过方向特征对齐和层次卷积补丁嵌入模块,在语义表示与像素级重建之间建立桥梁。进一步观察到SSL表示本质上位于超球面,采用黎曼流匹配训练扩散变换器(DiT)直接在球面潜在流形上进行训练。实验表明,我们的球面感知DiT在gFID、rFID和PSNR指标上均表现出色,验证了球面感知方法的优势。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑