Align-VAE: Aligning Attention and Geometry in VecSet VAEs for 3D Reconstruction

Huaizhi Qu, Mingxin Yang, Xintong Han +2

Set-based variational autoencoders (VecSet VAEs) represent 3D shapes as compact sets of latent tokens. Although each token originates from a query at a sampled surface location, wh…