Recent 3D content generation pipelines commonly employ Variational Autoencoders (VAEs) to encode shapes into compact latent representations for diffusion-based generation. However, the widely adopted uniform point sampling strategy in Shape VAE training often leads to a significant loss of geometric details, limiting the quality of shape reconstruction and downstream generation tasks. We present Dora-VAE, a novel approach that enhances VAE reconstruction through our proposed sharp edge sampling strategy and a dual cross-attention mechanism. By identifying and prioritizing regions with high geometric complexity during training, our method significantly improves the preservation of fine-grained shape features. Such sampling strategy and the dual attention mechanism enable the VAE to focus on crucial geometric details that are typically missed by uniform sampling approaches. To systematically evaluate VAE reconstruction quality, we additionally propose Dora-bench, a benchmark that quantifies shape complexity through the density of sharp edges, introducing a new metric focused on reconstruction accuracy at these salient geometric features. Extensive experiments on the Dora-bench demonstrate that Dora-VAE achieves comparable reconstruction quality to the state-of-the-art dense XCube-VAE while requiring a latent space at least 8$\times$ smaller (1,280 vs. > 10,000 codes).


    Access

    Download


    Export, share and cite



    Title :

    Dora: Sampling and Benchmarking for 3D Shape Variational Auto-Encoders


    Contributors:
    Chen, Rui (author) / Zhang, Jianfeng (author) / Liang, Yixun (author) / Luo, Guan (author) / Li, Weiyu (author) / Liu, Jiarui (author) / Li, Xiu (author) / Long, Xiaoxiao (author) / Feng, Jiashi (author) / Tan, Ping (author)

    Publication date :

    2024


    Remarks:

    Accepted by CVPR 2025. Project page: https://aruichen.github.io/Dora/


    Type of media :

    Preprint


    Type of material :

    Electronic Resource


    Language :

    English




    Consistency Regularization for Variational Auto-Encoders

    Sinha, Samarth / Dieng, Adji B. | ArXiv | 2021

    Free access

    Variational auto-encoders with Student's t-prior

    Abiri, Najmeh / Ohlsson, Mattias | ArXiv | 2020

    Free access

    Conditional Image Generation by Conditioning Variational Auto-Encoders

    Harvey, William / Naderiparizi, Saeid / Wood, Frank | ArXiv | 2021

    Free access


    Adversarial Training of Variational Auto-encoders for High Fidelity Image Generation

    Khan, Salman H. / Hayat, Munawar / Barnes, Nick | ArXiv | 2018

    Free access