Search papers, labs, and topics across Lattice.
This paper introduces ChunkVAE, a novel sparse grid variational autoencoder that organizes 3D modeling around local chunks instead of a global latent volume, addressing the rapid memory growth associated with high-resolution voxel grids. By employing local learned operators and two complementary data operators鈥擝alanced Binary Object Partitioning and S-Curve weighted stitching鈥擟hunkVAE achieves efficient memory usage and improved reconstruction quality. The method demonstrates competitive performance across multiple benchmarks, showing that local compression can effectively scale geometry while preserving necessary global interfaces for downstream tasks.
Local chunk-based modeling in ChunkVAE reduces memory usage and computation time while enhancing 3D reconstruction quality, outperforming traditional global latent approaches.
Sparse voxel grids preserve the spatial structure needed for detailed 3D reconstruction, but their memory still grows rapidly with resolution as active surface cells increase. We introduce ChunkVAE, a sparse grid variational autoencoder organized around local chunks rather than a global latent volume. Local learned operators permit independently chosen encoder and decoder partitions and allow inference chunk sizes to differ from training. Two complementary data operators make this flexibility practical: Balanced Binary Object Partitioning distributes active cells while limiting replicated overlap, while S-Curve weighted stitching attenuates unreliable boundary features when assembling a global latent or reconstruction. Across three object benchmarks, ChunkVAE is competitive with or better than strong baselines from $512^3$ to $1536^3$; smaller chunks lower peak allocated memory and shorten per-chunk compute, enabling faster parallel inference. Stable stitched latents and improved image to 3D metrics indicate that local compression can scale geometry while retaining the global interface required downstream.
