Open this publication in new window or tab >>Show others...
2024 (English)In: Proceedings of the 41st International Conference on Machine Learning / [ed] Salakhutdinov, Ruslan and Kolter, Zico and Heller, Katherine and Weller, Adrian and Oliver, Nuria and Scarlett, Jonathan and Berkenkamp, Felix, PMLR , 2024, Vol. 235, p. 35324-35339Conference paper, Poster (with or without abstract) (Refereed)
Abstract [en]
In this paper, we utilize hyperspheres and regular n-simplexes and propose an approach to learning deep features equivariant under the transformations of nD reflections and rotations, encompassed by the powerful group of O(n). Namely, we propose O(n)-equivariant neurons with spherical decision surfaces that generalize to any dimension n, which we call Deep Equivariant Hyperspheres. We demonstrate how to combine them in a network that directly operates on the basis of the input points and propose an invariant operator based on the relation between two points and a sphere, which as we show, turns out to be a Gram matrix. Using synthetic and real-world data in nD, we experimentally verify our theoretical contributions and find that our approach is superior to the competing methods for O(n)-equivariant benchmark datasets (classification and regression), demonstrating a favorable speed/performance trade-off. The code is available on GitHub.
Place, publisher, year, edition, pages
PMLR, 2024
Series
Proceedings of Machine Learning Research, ISSN 2640-3498 ; 235
National Category
Computer Vision and Learning Systems Engineering and Technology
Identifiers
urn:nbn:se:liu:diva-206460 (URN)
Conference
41st International Conference on Machine Learning, Vienna, Austria, 21-27 July 2024
2024-08-142024-08-142026-04-29