English
Related papers

Related papers: Relaxed Equivariant Graph Neural Networks

200 papers

Equivariant Graph Neural Networks (GNNs) have achieved remarkable success across diverse scientific applications. However, existing approaches face critical efficiency challenges when scaling to large geometric graphs and suffer significant…

Machine Learning · Computer Science 2025-06-25 Yuelin Zhang , Jiacheng Cen , Jiaqi Han , Wenbing Huang

The ability of convolutional neural networks (CNNs) to recognize objects regardless of their position in the image is due to the translation-equivariance of the convolutional operation. Group-equivariant CNNs transfer this equivariance to…

Computer Vision and Pattern Recognition · Computer Science 2023-04-13 Thomas Wimmer , Vladimir Golkov , Hoai Nam Dang , Moritz Zaiss , Andreas Maier , Daniel Cremers

Incorporating equivariance as an inductive bias into deep learning architectures to take advantage of the data symmetry has been successful in multiple applications, such as chemistry and dynamical systems. In particular, roto-translations…

Machine Learning · Computer Science 2026-01-06 Ahmed A. Elhag , T. Konstantin Rusch , Francesco Di Giovanni , Michael Bronstein

We propose a framework to understand the unprecedented performance and robustness of deep neural networks using field theory. Correlations between the weights within the same layer can be described by symmetries in that layer, and networks…

Computation · Statistics 2017-10-18 Ricky Fok , Aijun An , Xiaogang Wang

We test the efficiency of applying Geometric Deep Learning to the problems in low-dimensional topology in a certain simple setting. Specifically, we consider the class of 3-manifolds described by plumbing graphs and use Graph Neural…

Geometric Topology · Mathematics 2023-07-31 Pavel Putrov , Song Jin Ri

Neural networks for point clouds, which respect their natural invariance to permutation and rigid motion, have enjoyed recent success in modeling geometric phenomena, from molecular dynamics to recommender systems. Yet, to date, no model…

Machine Learning · Computer Science 2024-04-10 Snir Hordan , Tal Amir , Steven J. Gortler , Nadav Dym

Most equivariant neural networks rely on a single global symmetry, limiting their use in domains where symmetries are instead local. We introduce Torsor CNNs, a framework for learning on graphs with local symmetries encoded as edge…

Machine Learning · Computer Science 2025-10-28 Daiyuan Li , Shreya Arya , Robert Ghrist

Graph neural networks that model 3D data, such as point clouds or atoms, are typically desired to be $SO(3)$ equivariant, i.e., equivariant to 3D rotations. Unfortunately equivariant convolutions, which are a fundamental operation for…

Machine Learning · Computer Science 2023-06-16 Saro Passaro , C. Lawrence Zitnick

We introduce tensor field neural networks, which are locally equivariant to 3D rotations, translations, and permutations of points at every layer. 3D rotation equivariance removes the need for data augmentation to identify features in…

Machine Learning · Computer Science 2018-05-22 Nathaniel Thomas , Tess Smidt , Steven Kearnes , Lusann Yang , Li Li , Kai Kohlhoff , Patrick Riley

Message-passing neural networks (MPNNs) are the leading architecture for deep learning on graph-structured data, in large part due to their simplicity and scalability. Unfortunately, it was shown that these architectures are limited in…

Shape assembly aims to reassemble parts (or fragments) into a complete object, which is a common task in our daily life. Different from the semantic part assembly (e.g., assembling a chair's semantic parts like legs into a whole chair),…

Computer Vision and Pattern Recognition · Computer Science 2023-12-19 Ruihai Wu , Chenrui Tie , Yushi Du , Yan Zhao , Hao Dong

In this draft paper, we introduce a novel architecture for graph networks which is equivariant to the Euclidean group in $n$-dimensions. The model is designed to work with graph networks in their general form and can be shown to include…

Machine Learning · Computer Science 2021-05-24 Emma Slade , Francesco Farina

Existing equivariant neural networks require prior knowledge of the symmetry group and discretization for continuous groups. We propose to work with Lie algebras (infinitesimal generators) instead of Lie groups. Our model, the Lie algebra…

Machine Learning · Computer Science 2021-11-03 Nima Dehmamy , Robin Walters , Yanchen Liu , Dashun Wang , Rose Yu

Variational quantum algorithms are the leading candidate for advantage on near-term quantum hardware. When training a parametrized quantum circuit in this setting to solve a specific problem, the choice of ansatz is one of the most…

Quantum Physics · Physics 2023-04-25 Andrea Skolik , Michele Cattelan , Sheir Yarkoni , Thomas Bäck , Vedran Dunjko

Deep neural network architectures often consist of repetitive structural elements. We introduce an approach that reveals these patterns and can be broadly applied to the study of deep learning. Similarly to how a power strip helps untangle…

Statistical Mechanics · Physics 2025-07-03 Donghee Lee , Hye-Sung Lee , Jaeok Yi

Optimization over trained machine learning models has applications including: verification, minimizing neural acquisition functions, and integrating a trained surrogate into a larger decision-making problem. This paper formulates and solves…

Optimization and Control · Mathematics 2023-10-13 Shiqiang Zhang , Juan S. Campos , Christian Feldmann , David Walz , Frederik Sandfort , Miriam Mathea , Calvin Tsay , Ruth Misener

We present a geometric neural network-based tracking controller for systems evolving on matrix Lie groups under unknown dynamics, actuator faults, and bounded disturbances. Leveraging the left-invariance of the tangent bundle of matrix Lie…

Systems and Control · Electrical Eng. & Systems 2025-05-09 Robin Chhabra , Farzaneh Abdollahi

Nonlinear parametric systems have been widely used in modeling nonlinear dynamics in science and engineering. Bifurcation analysis of these nonlinear systems on the parameter space are usually used to study the solution structure such as…

Numerical Analysis · Mathematics 2022-01-26 Wenrui Hao , Chunyue Zheng

This paper proposes an equivariant neural network that takes data in any semi-simple Lie algebra as input. The corresponding group acts on the Lie algebra as adjoint operations, making our proposed network adjoint-equivariant. Our framework…

Machine Learning · Computer Science 2024-06-10 Tzu-Yuan Lin , Minghan Zhu , Maani Ghaffari

3D Convolutional Neural Networks are sensitive to transformations applied to their input. This is a problem because a voxelized version of a 3D object, and its rotated clone, will look unrelated to each other after passing through to the…

Computer Vision and Pattern Recognition · Computer Science 2018-04-13 Daniel Worrall , Gabriel Brostow