PreprintCited by 1,3033D Shape Modeling and AnalysisNeural Networks and ApplicationsTopological and Geometric Data Analysis

The DOI matches its DataCite record; 5 of the 5 references it cites were found in a registry or an index.

Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds

Open access

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.

Overview

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

DataCite · 10.48550/arxiv.1802.08219

Cites

5 of 5 references verified0 citation errors · 0 not checkable

Statistics

Verification

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • The structures and relaxation trajectories present a large and diverse dataset to enable training of learned, equivariant interatomic potentials with unprecedented accuracy and zero-shot generalization.

    “The structures and relaxation trajectories present a large and diverse dataset to enable training of learned, equivariant interatomic potentials30,31 with unprecedented accuracy and zero-shot generalization.”

    From Scaling deep learning for materials discovery (2023) · cites this paper as [31]

    Section 5.3 describes an equivariant neural network trained on molecular structures, supporting the citation’s use as an example of a learned equivariant model for atomic systems. Its demonstrated task is replacing a missing atom, not learning an interatomic potential; Section 6 presents learning molecular dynamics as future work. The citation is attached to “learned, equivariant interatomic potentials,” not to the sentence’s claims about the citing dataset or its prospective accuracy. (LLM, conf=0.85)

    Experimental — accuracy not yet measured

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Citations

Cites

Flagged first, then by citations

Research works printed but not linked: 25

  1. D. E. Worrall, S. J. Garbin, D. Turmukhambetov, and G. J. Brostow. Harmonic networks: Deep translation and rotation equivariance. In The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), July 2017.
  2. M. Reisert and H. Burkhardt. Spherical tensor calculus for local adaptive filtering. Tensors in Image Processing and Computer Vision, pages 153-178, 2009.
  3. Y. Zhou, Q. Ye, Q. Qiu, and J. Jiao. Oriented Response Networks. ArXiv e-prints, January 2017.
  4. D. M. Gonzalez, M. Volpi, N. Komodakis, and D. Tuia. Rotation equivariant vector field networks. CoRR, abs/1612.09346, 2016. URL http://arxiv.org/abs/1612.09346. · arXiv
  5. J. Li, Z. Yang, H. Liu, and D. Cai. Deep Rotation Equivariant Network. ArXiv e-prints, May 2017.
  6. T. S. Cohen and M. Welling. Group equivariant convolutional networks. In Proceedings of the International Conference on Machine Learning (ICML), 2016.
  7. T. Cohen, M. Geiger, J. Köhler, and M. Welling. Spherical CNNs. International Conference on Learning Representations, 2018. URL https://openreview.net/forum?id=Hkbd5xZRb.
  8. R. Kondor, H. Truong Son, H. Pan, B. Anderson, and S. Trivedi. Covariant compositional networks for learning graphs. ArXiv e-prints, January 2018.
  9. J. Gilmer, S. S. Schoenholz, P. F. Riley, O. Vinyals, and G. E. Dahl. Neural message passing for quantum chemistry. In Doina Precup and Yee Whye Teh, editors, Proceedings of the 34th International Conference on Machine Learning, volume 70 of Proceedings of Machine Learning Research, pages 1263-1272, International Convention Centre, Sydney, Australia, 06-11 Aug 2017. PMLR. URL http://proceedings.mlr.press/v70/gilmer17a.html.
  10. C. R. Qi, H. Su, K. Mo, and L. J. Guibas. PointNet: Deep learning on point sets for 3D classification and segmentation. arXiv preprint arXiv:1612.00593, 2016. · arXiv
  11. C. R. Qi, L. Yi, H. Su, and L. J. Guibas. PointNet++: Deep hierarchical feature learning on point sets in a metric space. arXiv preprint arXiv:1706.02413, 2017. · arXiv
  12. K. T. Schütt, P.-J. Kindermans, H. E. Sauceda, S. Chmiela, A. Tkatchenko, and K.-R. Müller. SchNet: A continuous-filter convolutional neural network for modeling quantum interactions. ArXiv e-prints, June 2017.
  13. I. Wallach, M. Dzamba, and A. Heifets. AtomNet: A deep convolutional neural network for bioactivity prediction in structure-based drug discovery. CoRR, abs/1510.02855, 2015. URL http://arxiv.org/abs/1510.02855. · arXiv
  14. W. Boomsma and J. Frellsen. Spherical convolutions and their application in molecular mod- elling. In I. Guyon, U. V. Luxburg, S. Bengio, H. Wallach, R. Fergus, S. Vishwanathan, and R. Garnett, editors, Advances in Neural Information Processing Systems 30, pages 3436-3446.
  15. Curran Associates, Inc., 2017. URL http://papers.nips.cc/paper/6935-spherical- convolutions-and-their-application-in-molecular-modelling.pdf.
  16. T. S. Cohen and M. Welling. Steerable CNNs. In International Conference on Learning Representations (ICLR), 2017.
  17. R. Kondor and S. Trivedi. On the generalization of equivariance and convolution in neural networks to the action of compact groups. arXiv preprint arXiv:1802.03690, 2018. · arXiv
  18. R. Kondor. N -body networks: A covariant hierarchical neural network architecture for learning atomic potentials. arXiv preprint arXiv:1803.01588, 2018. · arXiv
  19. T. S. Cohen, M. Geiger, and M. Weiler. Intertwiners between induced representations (with applications to the theory of equivariant neural networks). arXiv preprint arXiv:1803.10743, 2018. · arXiv
  20. R. Goodman and N. R. Wallach. Representations and invariants of the classical groups, volume 68. Cambridge University Press, 1998.
  21. R. Gilmore. Lie groups, physics, and geometry: An introduction for physicists, engineers and chemists. Cambridge University Press, 2008.
  22. D. J. Griffiths. Introduction to quantum mechanics. Cambridge University Press, 2017.
  23. S. Chmiela, A. Tkatchenko, H. E. Sauceda, I. Poltavsky, K. T. Schütt, and K.-R. Müller. Machine learning of accurate energy-conserving molecular force fields. Science Advances, 3(5), 2017. doi: 10.1126/sciadv.1603015. URL http://advances.sciencemag.org/content/3/5/ e1603015. · DOI
  24. Z. Wu, S. Song, A. Khosla, F. Yu, L. Zhang, X. Tang, and J. Xiao. 3d shapenets: A deep representation for volumetric shapes. In IEEE Conference on Computer Vision and Pattern Recognition, CVPR 2015, Boston, MA, USA, June 7-12, 2015, pages 1912-1920, 2015. doi: 10.1109/CVPR.2015.7298801. URL https://doi.org/10.1109/CVPR.2015.7298801. · DOI
  25. L. D. Landau and E. M. Lifshitz. Mechanics, volume 1 of Course of Theoretical Physics. Elsevier, 1976.

Cited by

Sorted by citations

Metadata

Authors
Thomas, N., Smidt, T., Kearnes, S., Yang, L., Li, L., Kohlhoff, K., & Riley, P.
Affiliations
  • Stanford University (US)
  • Lawrence Berkeley National Laboratory (US)
  • Google (United States) (US)
Venue
arXiv
Publisher
arXiv
Published
2018
ISSN
2331-8422 (ISSN-L)
OpenAlex
W2788775653
Cited by
1,303
References
5
Citations by year
2026: 243 · 2025: 216 · 2024: 228 · 2023: 221 · 2022: 158 · 2021: 113 · 2020: 77 · 2019: 37 · 2018: 8
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Computer Graphics and Computer-Aided Design › 3D Shape Modeling and Analysis
  • Physical Sciences › Computer Science › Artificial Intelligence › Neural Networks and Applications
  • Physical Sciences › Computer Science › Computational Theory and Mathematics › Topological and Geometric Data Analysis
Keywords
rotation equivariance, translation equivariance, permutation equivariance, 3D point clouds, spherical harmonics, equivariant neural networks, higher-order tensors, data augmentation, solid geometry

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review