Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
read the original abstract
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 arbitrary orientations. Our network uses filters built from spherical harmonics; due to the mathematical consequences of this filter choice, each layer accepts as input (and guarantees as output) scalars, vectors, and higher-order tensors, in the geometric sense of these terms. We demonstrate the capabilities of tensor field networks with tasks in geometry, physics, and chemistry.
This paper has not been read by Pith yet.
Forward citations
Cited by 30 Pith papers
-
Group-Algebraic Tensors: Provably-optimal Equivariant Learning and Physical Symmetry Discovery
A group-algebraic tensor framework delivers Eckart-Young optimal equivariant approximations and recovers physical selection rules from data alone via a Lean-formalized star_G algebra.
-
Discretizing Group-Convolutional Neural Networks for 3D Geometry in Feature Space
Feature-space sampling in GCNNs preserves 3D classification accuracy with coarse discretization, enabling precomputation and faster training of equivariant models.
-
Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks
An O(L^3) algorithm computes contracted Clebsch-Gordan tensor products for equivariant ML potentials using a structured angular grid and spherical Poisson bracket to handle parity-odd terms at fixed CP rank.
-
Chem-GMNet: A Sphere-Native Geometric Transformer for Molecular Property Prediction
Chem-GMNet uses sphere-native embeddings, DualSKA attention, and SH-FFN layers to match or beat ChemBERTa-2 on MoleculeNet tasks with fewer parameters and sometimes no pretraining.
-
Symmetry-Protected Basin Localization in Variational Quantum Eigensolvers
A symmetry-constrained preconditioner maps molecular geometry to VQE circuit parameters in the correlated ground-state basin, cutting initialization errors by 38x to 6250x on stretched molecules.
-
Diagnosing Spectral Ceilings in Equivariant Neural Force Fields
A new diagnostic reveals that L=2 equivariant force field backbones preserve frequencies up to l=4 but collapse at l=5 on aspirin, consistent with a finite-degree span theorem and controls.
-
Improving Molecular Force Fields with Minimal Temporal Information
FRAMES training with minimal temporal information from MD trajectory pairs improves energy and force prediction accuracy over Equiformer on MD17 and ISO17 benchmarks.
-
Physics-Constrained Self-Energy Warm Starts for Charge-Self-Consistent DFT+DMFT: Application to Iron at Core Conditions
Physics-constrained equivariant GNN warm starts cut DFT+DMFT self-consistency iterations by 2-4x across Fe, FeO and NiO, enabling an MLIP-based NVE coexistence simulation that yields a 6225 K melting temperature for h...
-
Atomistic Machine Learning with Irreducible Cartesian Natural Tensors
CarNet develops irreducible Cartesian natural tensors and an equivariant model that matches leading spherical-tensor performance for ML interatomic potentials and high-rank tensor predictions like elastic constants.
-
Unifying machine learning and quantum chemistry -- a deep neural network for molecular wavefunctions
Deep neural network predicts molecular wavefunctions in atomic orbital basis from which quantum properties are derived at force-field efficiency.
-
Tensor Channel Equivariant Graph Neural Networks for Molecular Polarizability Prediction
A tensor-channel equivariant GNN based on PaiNN propagates symmetric rank-2 tensor features during message passing and achieves lower full-tensor and anisotropic error than readout-only and MACE baselines on QM7-X geometries.
-
Neural Point-Forms
Neural point-forms are introduced as permutation-invariant neural layers that output learned form-comparison matrices for point clouds, with a claimed consistency proof under sampling and manifold assumptions and comp...
-
Compact SO(3) Equivariant Atomistic Foundation Models via Structural Pruning
Structural pruning of SO(3) equivariant atomistic models from large checkpoints yields 1.5-4x fewer parameters and 2.5-4x less pre-training compute than small models trained from scratch, while outperforming them on m...
-
PRIME: Protein Representation via Physics-Informed Multiscale Equivariant Hierarchies
PRIME is a multiscale graph framework that connects five physics-grounded protein structure levels with bidirectional operators and reports large gains on fold classification and reaction prediction benchmarks.
-
PRIME: Protein Representation via Physics-Informed Multiscale Equivariant Hierarchies
PRIME is a five-level hierarchical equivariant graph model for proteins that uses physics-informed deterministic operators to exchange information across scales and achieves state-of-the-art results on fold classifica...
-
Agentic Fusion of Large Atomic and Language Models to Accelerate Superconductor Discovery
An agentic framework fusing large atomic and language models rediscovers 66 known superconductors and guides experimental verification of four new ones with transition temperatures from 2.5 K to 6.5 K.
-
Metriplector: From Field Theory to Neural Architecture
Metriplector treats neural computation as coupled metriplectic field dynamics whose stress-energy tensor readout achieves competitive results on vision, control, Sudoku, language modeling, and pathfinding with small p...
-
Energy-Guided Generative Modeling for Low-Energy Molecular Structure Discovery
EnFlow integrates flow-based conformer generation with energy landscape modeling to enable joint ensemble generation and ground-state identification using only 1-2 ODE steps.
-
Bayesian E(3)-Equivariant Interatomic Potential with Iterative Restratification of Many-body Message Passing
Bayesian E(3)-equivariant MLPs with joint energy-force NLL loss achieve competitive accuracy while enabling uncertainty-guided active learning, OOD detection, and calibration.
-
Adaptive Canonicalization with Application to Invariant Anisotropic Geometric Networks
Adaptive canonicalization selects input canonical forms by maximizing network predictive confidence to yield continuous symmetry-preserving models with universal approximation for equivariant geometric networks.
-
Fast and flexible long-range models for atomistic machine learning
Introduces torch-pme and jax-pme libraries that embed Ewald-based long-range methods and purified descriptors into atomistic ML for accurate handling of non-local physical interactions.
-
Learning Color Equivariant Representations
Presents hue-, saturation-, luminance-equivariant GCNNs via a direct-image lifting layer that resolves invalid RGB issues in prior CEConv work and reports better OOD generalization plus sample efficiency.
-
TetraSphere: A Neural Descriptor for O(3)-Invariant Point Cloud Analysis
TetraSphere integrates a TetraTransform based on steerable spherical neurons into VN-DGCNN to produce an O(3)-equivariant descriptor that reports new SOTA results on rotated ScanObjectNN, ModelNet40 classification, an...
-
Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges
Geometric deep learning provides a unified mathematical framework based on grids, groups, graphs, geodesics, and gauges to explain and extend neural network architectures by incorporating physical regularities.
-
TacSE3: Equivariant SE(3) Motion Estimation from Low-Texture Visuotactile Images for In-Gripper Tracking and Compensation
TacSE3 estimates incremental SE(3) motion from paired visuotactile sensors by deriving planar translation from contact-centroid shifts and rotation from shear responses for in-gripper tracking.
-
A Systematic Survey and Benchmark of Deep Learning for Molecular Property Prediction in the Foundation Model Era
A systematic survey and benchmark of four deep learning paradigms for molecular property prediction that organizes the field, critiques current data practices, and outlines three future directions.
-
Machine-learning modeling of magnetization dynamics in quasi-equilibrium and driven metallic spin systems
Generalized ML force fields reproduce non-collinear magnetic orders on lattices and predict voltage-driven domain-wall motion in itinerant magnets using extensions to nonequilibrium torques.
-
Simultaneous Learning of Static and Dynamic Charges
Independent learning of static and dynamic charges in water systems achieves comparable accuracy to coupled models with environment-dependent screening but at lower computational cost.
-
Machine Learning Interatomic Potentials: Advancing Open-Source Software for Efficient and Scalable Molecular Simulation
mlip v2 is a new software release that integrates API redesign, e3j backend, eSEN model, improved charge modeling, and expanded simulation capabilities to support larger-scale molecular modeling.
-
Neuro-evolutionary stochastic architectures in gauge-covariant neural fields
Only the fully symmetry-constrained Ginibre U(1) evolutionary model in gauge-covariant neural fields robustly reaches a narrow near-marginal regime and reproduces the predicted low-frequency finite-width spectral behavior.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.