arXiv:2503.16653cs.CV2025-03被引 13

混合全量与线性注意力,高效生成高质量3D网格

iFlame: Interleaving Full and Linear Attention for Efficient Mesh Generation

  • 交替使用全量与线性注意力,兼顾表达力与效率
  • 4张GPU训练2天,处理最多4000面的高分辨率网格
  • 缓存优化使推理速度翻倍,显存占用降为1/8

本文提出iFlame,一种基于Transformer的新型网格生成架构。尽管注意力模型在网格生成中表现优异,但其二次计算复杂度限制了高分辨率3D数据的可扩展性;而线性注意力虽计算成本低,却难以捕捉长程依赖。为此,我们设计了一种交错式自回归网格生成框架,融合线性注意力的高效性与全量注意力的表达能力。为进一步提升效率并利用网格结构特性,将该方法嵌入倒置瓶型架构,显著加速训练。通过缓存算法优化,推理速度近乎翻倍,KV缓存大小减少七分之六。在ShapeNet和Objaverse数据集上验证,该方法在仅用4张GPU、2天时间训练39,000条含最多4,000个面的数据时,生成质量媲美纯注意力模型。结果表明,该交错框架有效平衡了计算效率与生成性能,适用于实际网格生成任务。

原文摘要 · Abstract (English)

This paper propose iFlame, a novel transformer-based network architecture for mesh generation. While attention-based models have demonstrated remarkable performance in mesh generation, their quadratic computational complexity limits scalability, particularly for high-resolution 3D data. Conversely, linear attention mechanisms offer lower computational costs but often struggle to capture long-range dependencies, resulting in suboptimal outcomes. To address this trade-off, we propose an interleaving autoregressive mesh generation framework that combines the efficiency of linear attention with the expressive power of full attention mechanisms. To further enhance efficiency and leverage the inherent structure of mesh representations, we integrate this interleaving approach into an hourglass architecture, which significantly boosts efficiency. Our approach reduces training time while achieving performance comparable to pure attention-based models. To improve inference efficiency, we implemented a caching algorithm that almost doubles the speed and reduces the KV cache size by seven-eighths compared to the original Transformer. We evaluate our framework on ShapeNet and Objaverse, demonstrating its ability to generate high-quality 3D meshes efficiently. Our results indicate that the proposed interleaving framework effectively balances computational efficiency and generative performance, making it a practical solution for mesh generation. The training takes only 2 days with 4 GPUs on 39k data with a maximum of 4k faces on Objaverse.

3D生成注意力机制网格生成高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。