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 accessAbstract
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.
Integrity signals
Registry
Cites
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.
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
Quantum chemistry structures and properties of 134 kilo molecules
Ramakrishnan, R., Dral, P. O., Rupp, M., & von Lilienfeld, O. A. (2014) · Scientific Data · doi:10.1038/sdata.2014.22 · Cited by 1,966
ANI-1: an extensible neural network potential with DFT accuracy at force field computational cost
Smith, J. S., Isayev, O., & Roitberg, A. E. (2017) · Chemical Science · doi:10.1039/c6sc05720a · Cited by 1,921
Quantum-chemical insights from deep tensor neural networks
Schütt, K. T., Arbabzadah, F., Chmiela, S., Müller, K. R., & Tkatchenko, A. (2017) · Nature Communications · doi:10.1038/ncomms13890 · Cited by 1,445
Prediction Errors of Molecular Machine Learning Models Lower than Hybrid DFT Error
Faber, F. A., Hutchison, L., Huang, B., Gilmer, J., Schoenholz, S. S., Dahl, G. E., Vinyals, O., Kearnes, S., Riley, P. F., & von Lilienfeld, O. A. (2017) · Journal of Chemical Theory and Computation · doi:10.1021/acs.jctc.7b00577 · Cited by 651
Self-citation3D deep convolutional neural networks for amino acid environment similarity analysis
Torng, W., & Altman, R. B. (2017) · BMC Bioinformatics · doi:10.1186/s12859-017-1702-0 · Cited by 142
Flagged first, then by citations
Research works printed but not linked: 25
- 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.
- M. Reisert and H. Burkhardt. Spherical tensor calculus for local adaptive filtering. Tensors in Image Processing and Computer Vision, pages 153-178, 2009.
- Y. Zhou, Q. Ye, Q. Qiu, and J. Jiao. Oriented Response Networks. ArXiv e-prints, January 2017.
- 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
- J. Li, Z. Yang, H. Liu, and D. Cai. Deep Rotation Equivariant Network. ArXiv e-prints, May 2017.
- T. S. Cohen and M. Welling. Group equivariant convolutional networks. In Proceedings of the International Conference on Machine Learning (ICML), 2016.
- 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.
- R. Kondor, H. Truong Son, H. Pan, B. Anderson, and S. Trivedi. Covariant compositional networks for learning graphs. ArXiv e-prints, January 2018.
- 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.
- 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
- 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
- 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.
- 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
- 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.
- Curran Associates, Inc., 2017. URL http://papers.nips.cc/paper/6935-spherical- convolutions-and-their-application-in-molecular-modelling.pdf.
- T. S. Cohen and M. Welling. Steerable CNNs. In International Conference on Learning Representations (ICLR), 2017.
- 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
- R. Kondor. N -body networks: A covariant hierarchical neural network architecture for learning atomic potentials. arXiv preprint arXiv:1803.01588, 2018. · arXiv
- 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
- R. Goodman and N. R. Wallach. Representations and invariants of the classical groups, volume 68. Cambridge University Press, 1998.
- R. Gilmore. Lie groups, physics, and geometry: An introduction for physicists, engineers and chemists. Cambridge University Press, 2008.
- D. J. Griffiths. Introduction to quantum mechanics. Cambridge University Press, 2017.
- 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
- 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
- L. D. Landau and E. M. Lifshitz. Mechanics, volume 1 of Course of Theoretical Physics. Elsevier, 1976.
Cited by
Sorted by citations
Scientific discovery in the age of artificial intelligence.
Wang, H., Fu, T., Du, Y., Gao, W., Huang, K., Liu, Z., Chandak, P., Liu, S., Van Katwyk, P., Deac, A., et al. (2023) · Nature · doi:10.1038/s41586-023-06221-2 · Cited by 1,823
Scaling deep learning for materials discovery
Merchant, A., Batzner, S., Schoenholz, S. S., Aykol, M., Cheon, G., & Cubuk, E. D. (2023) · Nature · doi:10.1038/s41586-023-06735-9 · Cited by 1,527
Combining Machine Learning and Computational Chemistry for Predictive Insights Into Chemical Systems
Keith, J. A., Vassilev-Galindo, V., Cheng, B., Chmiela, S., Gastegger, M., Müller, K.-R., & Tkatchenko, A. (2021) · Chemical Reviews · doi:10.1021/acs.chemrev.1c00107 · Cited by 849
Deep learning for molecular design—a review of the state of the art
Elton, D. C., Boukouvalas, Z., Fuge, M. D., & Chung, P. W. (2019) · Molecular Systems Design & Engineering · doi:10.1039/c9me00039a · Cited by 683
A review of large language models and autonomous agents in chemistry
Ramos, M. C., Collison, C. J., & White, A. D. (2025) · Chemical Science · doi:10.1039/d4sc03921a · Cited by 274
DPA-2: a large atomic model as a multi-task learner
Zhang, D., Liu, X., Zhang, X., Zhang, C., Cai, C., Bi, H., Du, Y., Qin, X., Peng, A., Huang, J., et al. (2023) · arXiv · doi:10.48550/arxiv.2312.15492 · Cited by 155
Uni-Mol: A Universal 3D Molecular Representation Learning Framework
Zhou, G., Gao, Z., Ding, Q., Zheng, H., Xu, H., Wei, Z., Zhang, L., & Ke, G. (2022) · doi:10.26434/chemrxiv-2022-jjm0j · Cited by 81
Protein structure prediction by AlphaFold 2: are attention and symmetries all you need?
Bouatta, N., Sorger, P., & AlQuraishi, M. (2021) · Acta Crystallographica Section D Structural Biology · doi:10.1107/s2059798321007531 · Cited by 69
Data-Driven Studies of Li-Ion-Battery Materials
Kauwe, S. K., Rhone, T. D., & Sparks, T. D. (2019) · Crystals · doi:10.3390/cryst9010054 · Cited by 57
Protein sequence‐to‐structure learning: Is this the end(‐to‐end revolution)?
Laine, E., Eismann, S., Elofsson, A., & Grudinin, S. (2021) · Proteins: Structure, Function, and Bioinformatics · doi:10.1002/prot.26235 · Cited by 44
“In particular, advances in the treatment of language [14] and of 3D geometry [15, 16, 17, 18] by deep learning architectures have further benefited the field of protein structure and function prediction.”
BackgroundInfluentialThorn, A. (2022) · Current Opinion in Structural Biology · doi:10.1016/j.sbi.2022.102368 · Cited by 16
Interpretable representation learning for 3D multi-piece intracellular structures using point clouds
Vasan, R., Ferrante, A. J., Borensztejn, A., Frick, C. L., Garrison, P., Gaudreault, N., Mogre, S. S., Mohammed, F. S., Morris, B., Pires, G. G., et al. (2025) · Nature Methods · doi:10.1038/s41592-025-02729-9 · Cited by 7
Nguyen, N.-Q., & Kang, J. (2025) · Journal of Chemical Information and Modeling · doi:10.1021/acs.jcim.5c00773 · Cited by 7
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)
- arXiv
- 1802.08219
- 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
- green
- arxiv.org
- 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