English
Related papers

Related papers: SE(3) Equivariant Graph Neural Networks with Compl…

200 papers

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

Incorporating geometric invariance into neural networks enhances parameter efficiency but typically increases computational costs. This paper introduces new equivariant neural networks that preserve symmetry while maintaining a comparable…

Computer Vision and Pattern Recognition · Computer Science 2025-06-25 Georg Bökman , David Nordström , Fredrik Kahl

Accurately predicting the elastic properties of crystalline solids is vital for computational materials science. However, traditional atomistic scale ab initio approaches are computationally intensive, especially for studying complex…

Disordered Systems and Neural Networks · Physics 2023-11-13 Teerachote Pakornchote , Annop Ektarawong , Thiparat Chotibut

Equivariant deep learning architectures exploit symmetries in learning problems to improve the sample efficiency of neural-network-based models and their ability to generalise. However, when modelling real-world data, learning problems are…

Machine Learning · Statistics 2024-11-12 Matthew Ashman , Cristiana Diaconu , Adrian Weller , Wessel Bruinsma , Richard E. Turner

The task of shape space learning involves mapping a train set of shapes to and from a latent representation space with good generalization properties. Often, real-world collections of shapes have symmetries, which can be defined as…

Computer Vision and Pattern Recognition · Computer Science 2022-08-30 Matan Atzmon , Koki Nagano , Sanja Fidler , Sameh Khamis , Yaron Lipman

Graph-structured data jointly contain discrete topology and continuous geometry, which poses fundamental challenges for generative modeling due to heterogeneous distributions, incompatible noise dynamics, and the need for equivariant…

Machine Learning · Computer Science 2026-04-10 Rongjian Xu , Teng Pang , Zhiqiang Dong , Guoqiang Wu

Graph neural networks (GNNs) are commonly described as being permutation equivariant with respect to node relabeling in the graph. This symmetry of GNNs is often compared to the translation equivariance of Euclidean convolution neural…

Machine Learning · Statistics 2023-11-20 Ningyuan Huang , Ron Levie , Soledad Villar

Group equivariant neural networks have proven effective in modelling a wide range of tasks where the data lives in a classical geometric space and exhibits well-defined group symmetries. However, these networks are not suitable for learning…

Machine Learning · Computer Science 2025-05-26 Edward Pearce-Crump

Scene graphs have been proven to be useful for various scene understanding tasks due to their compact and explicit nature. However, existing approaches often neglect the importance of maintaining the symmetry-preserving property when…

Computer Vision and Pattern Recognition · Computer Science 2024-07-02 Quang P. M. Pham , Khoi T. N. Nguyen , Lan C. Ngo , Truong Do , Truong Son Hy

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

Equivariant Graph Neural Networks (GNNs) are essential for physically consistent molecular simulations but suffer from high computational costs and memory bottlenecks, especially with high-order representations. While low-bit quantization…

Machine Learning · Computer Science 2026-03-17 Haoyu Zhou , Ping Xue , Hao Zhang , Tianfan Fu

While (message-passing) graph neural networks have clear limitations in approximating permutation-equivariant functions over graphs or general relational data, more expressive, higher-order graph neural networks do not scale to large…

Machine Learning · Computer Science 2022-08-31 Christopher Morris , Gaurav Rattan , Sandra Kiefer , Siamak Ravanbakhsh

Many quantities we are interested in predicting are geometric tensors; we refer to this class of problems as geometric prediction. Attempts to perform geometric prediction in real-world scenarios have been limited to approximating them…

Machine Learning · Computer Science 2020-06-26 Raphael J. L. Townshend , Brent Townshend , Stephan Eismann , Ron O. Dror

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

Recent work has shown the utility of developing machine learning models that respect the structure and symmetries of eigenvectors. These works promote sign invariance, since for any eigenvector v the negation -v is also an eigenvector.…

Machine Learning · Computer Science 2023-12-06 Derek Lim , Joshua Robinson , Stefanie Jegelka , Haggai Maron

Analyzing scalar and vector fields on the sphere, such as temperature or wind speed and direction on Earth, is a difficult task. Models should respect both the rotational symmetries of the sphere and the inherent symmetries of the vector…

Machine Learning · Computer Science 2026-04-01 Francesco Ballerin , Nello Blaser , Erlend Grong

We introduce Equivariant Neural Field Expectation Maximization (EFEM), a simple, effective, and robust geometric algorithm that can segment objects in 3D scenes without annotations or training on scenes. We achieve such unsupervised…

Computer Vision and Pattern Recognition · Computer Science 2023-03-28 Jiahui Lei , Congyue Deng , Karl Schmeckpeper , Leonidas Guibas , Kostas Daniilidis

Equivariances provide useful inductive biases in neural network modeling, with the translation equivariance of convolutional neural networks being a canonical example. Equivariances can be embedded in architectures through weight-sharing…

Machine Learning · Computer Science 2022-11-15 Tycho F. A. van der Ouderaa , David W. Romero , Mark van der Wilk

Invariant models, one important class of geometric deep learning models, are capable of generating meaningful geometric representations by leveraging informative geometric features in point clouds. These models are characterized by their…

Machine Learning · Computer Science 2025-06-17 Zian Li , Xiyuan Wang , Shijia Kang , Muhan Zhang

We present E(3)-Pose, a novel fast pose estimation method that jointly and explicitly models rotation equivariance and object symmetry. Our work is motivated by the challenging problem of accounting for fetal head motion during a diagnostic…

Computer Vision and Pattern Recognition · Computer Science 2026-03-31 Ramya Muthukrishnan , Borjan Gagoski , Aryn Lee , P. Ellen Grant , Elfar Adalsteinsson , Benjamin Billot , Polina Golland