Tangent Bundle Convolutional Learning: from Manifolds to Cellular Sheaves and Back
Claudio Battiloro, Zhiyang Wang, Hans Riess, Paolo Di Lorenzo, Alejandro Ribeiro
TL;DR
This work develops tangent bundle convolution to process vector-field signals on Riemannian manifolds by leveraging the Connection Laplacian Δ. It defines tangent bundle filters and Tangent Bundle Neural Networks (TNNs), derives their spectral representation, and provides a principled space-time discretization via orthogonal cellular sheaves, yielding DD-TNNs with a formal convergence guarantee to the continuous model. The approach unifies manifold, graph, and time-domain filtering for vector fields, and is validated on torus denoising, wind-field reconstruction/forecasting, and manifold classification, demonstrating clear benefits from incorporating tangent-bundle geometry. By linking tangent bundle processing to cellular sheaves and algebraic signal processing, the paper offers a theoretically grounded, scalable framework with potential implications for geometric deep learning and physics-informed networks.
Abstract
In this work we introduce a convolution operation over the tangent bundle of Riemann manifolds in terms of exponentials of the Connection Laplacian operator. We define tangent bundle filters and tangent bundle neural networks (TNNs) based on this convolution operation, which are novel continuous architectures operating on tangent bundle signals, i.e. vector fields over the manifolds. Tangent bundle filters admit a spectral representation that generalizes the ones of scalar manifold filters, graph filters and standard convolutional filters in continuous time. We then introduce a discretization procedure, both in the space and time domains, to make TNNs implementable, showing that their discrete counterpart is a novel principled variant of the very recently introduced sheaf neural networks. We formally prove that this discretized architecture converges to the underlying continuous TNN. Finally, we numerically evaluate the effectiveness of the proposed architecture on various learning tasks, both on synthetic and real data.
