Adaptive Spiral Layers for Efficient 3D Representation Learning on Meshes
Adaptive Spiral Layers for Efficient 3D Representation Learning on Meshes
复制标题
DOI:
10.1109/iccv51070.2023.01344
复制
发表时间:
2023-10
期刊:
影响因子:
--
通讯作者:
F. Babiloni;Matteo Maggioni;T. Tanay;Jiankang Deng;A. Leonardis;S. Zafeiriou
中科院分区:
文献类型:
--
作者:
F. Babiloni;Matteo Maggioni;T. Tanay;Jiankang Deng;A. Leonardis;S. Zafeiriou
The success of deep learning models on structured data has generated significant interest in extending their application to non-Euclidean domains. In this work, we introduce a novel intrinsic operator suitable for representation learning on 3D meshes. Our operator is specifically tailored to adapt its behavior to the irregular structure of the underlying graph and effectively utilize its long-range dependencies, while at the same time ensuring computational efficiency and ease of optimization. In particular, inspired by the framework of Spiral Convolution, which extracts and transforms the vertices in the 3D mesh following a local spiral ordering, we propose a general operator that dynamically adjusts the length of the spiral trajectory and the parameters of the transformation for each processed vertex and mesh. Then, we use polyadic decomposition to factorize its dense weight tensor into a sequence of lighter linear layers that separately process features and vertices information, hence significantly reducing the computational complexity without introducing any stringent inductive biases. Notably, we leverage dynamic gating to achieve spatial adaptivity and induce global reasoning with constant time complexity benefitting from an efficient dynamic pooling mechanism based on Summed-Area-tables. Used as a drop-in replacement on existing architectures for shape correspondence our operator significantly improves the performance-efficiency trade-off, and in 3D shape generation with morphable models achieves state-of-the-art performance with a three-fold reduction in the number of parameters required. Project page: https://github.com/Fb2221/DFC