Accurate piezoelectric tensor prediction with equivariant attention tensor graph neural network
Source abstract
Abstract
The piezoelectric materials enable the mutual conversion between mechanical and electrical energy, which drive a multi-billion dollar industry through their applications as sensors, actuators, and energy harvesters. The third-rank piezoelectric tensor is the core matrices for piezoelectric materials and their devices. However, the high costs of obtaining full piezoelectric tensor data through either experimental or computational methods make a significant challenge. Here, we propose an equivariant attention tensor graph neural network (EATGNN) that can identify crystal symmetry and remain independent of the reference frame, ultimately enabling the accurate prediction of the complete third-rank piezoelectric tensor. Especially, we perform an irreducible decomposition of the piezoelectric tensor into four irreducible representations to efficiently reserve the symmetry under group transformation operations. Our results further demonstrate that this model performs well in both bulk and two-dimensional materials. Finally, combining EATGNN with first-principles calculations, we discovered several potential high-performance piezoelectric materials.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Machine Learning in Materials Science · Thermal properties of materials · Advanced Sensor and Energy Harvesting Materials
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (44)
Show source-provided references
1. Uchino, K. in Advanced Piezoelectric Materials, edited by Uchino, K. (Woodhead Publishing, 2010), pp. 1–85.10.1533/9781845699758.1 ↗
2. Borlido, P. et al. Exchange-correlation functionals for band gaps of solids: benchmark, reparametrization and machine learning. Npj Comput. Mater. 6, 96 (2020).10.1038/s41524-020-00360-0 ↗
3. Yang, Z. et al. Scalable crystal structure relaxation using an iteration-free deep generative model with uncertainty quantification. Nat. Commun. 15, 8148 (2024).10.1038/s41467-024-52378-3 ↗
4. Teichert, G. H., Natarajan, A. R., Van der Ven, A. & Garikipati, K. Machine learning materials physics: integrable deep neural networks enable scale bridging by learning free energy functions. Comput. Meth. Appl. Mech. Eng. 353, 201–216 (2019).10.1016/j.cma.2019.05.019 ↗
5. Chen, L. et al. Frequency-dependent dielectric constant prediction of polymers using machine learning. Npj Comput. Mater. 6, 61 (2020).10.1038/s41524-020-0333-6 ↗
6. Ward, L. et al. Matminer: an open source toolkit for materials data mining. Comput. Mater. Sci. 152, 60–69 (2018).10.1016/j.commatsci.2018.05.018 ↗
7. Choudhary, K., Decost, B. & Tavazza, F. Machine learning with force-field-inspired descriptors for materials: fast screening and mapping energy landscape. Phys. Rev. Mater. 2, 83801 (2018).10.1103/PhysRevMaterials.2.083801 ↗
8. Xie, T. & Grossman, J. C. Crystal graph convolutional neural networks for an accurate and interpretable prediction of material properties. Phys. Rev. Lett. 120, 145301 (2018).10.1103/PhysRevLett.120.145301 ↗
9. Choudhary, K. & Decost, B. Atomistic line graph neural network for improved materials property predictions. Npj Comput. Mater. 7, 185 (2021).10.1038/s41524-021-00650-1 ↗
10. Liao, Y. & Smidt, T. Equiformer: equivariant graph attention transformer for 3d atomistic graphs. In Proc. International Conference on Learning Representations (2023).
11. Hu, J. & Song, Y. Piezoelectric modulus prediction using machine learning and graph neural networks. Chem. Phys. Lett. 791, 139359 (2022).10.1016/j.cplett.2022.139359 ↗
12. Poudel, S., Thapa, R., Basnet, R., Timofiejczuk, A. & Kunwar, A. Piezotensornet: crystallography informed multi-scale hierarchical machine learning model for rapid piezoelectric performance finetuning. Appl. Energy 361, 122901 (2024).10.1016/j.apenergy.2024.122901 ↗
13. Li, X. et al. Machine learning accelerated discovery of functional mxenes with giant piezoelectric coefficients. Acs Appl. Mater. Interfaces 16, 12731–12743 (2024).10.1021/acsami.3c14610 ↗
14. Liu, J., Liu, S., Yang, J. & Liu, L. Electric auxetic effect in piezoelectrics. Phys. Rev. Lett. 125, 197601 (2020).10.1103/PhysRevLett.125.197601 ↗
15. Benveniste, Y. & Dvorak, G. J. Uniform fields and universal relations in piezoelectric composites. J. Mech. Phys. Solids 40, 1295–1312 (1992).10.1016/0022-5096(92)90016-U ↗
16. Zhong, Y., Yu, H., Gong, X. & Xiang, H. A general tensor prediction framework based on graph neural networks. J. Phys. Chem. Lett. 14, 6339–6348 (2023).10.1021/acs.jpclett.3c01200 ↗
17. Fuchs, F., Worrall, D., Fischer, V. & Welling, M. Se (3)-transformers: 3d roto-translation equivariant attention networks. Adv. Neural Inf. Process. Syst. 33, 1970–1981 (2020).
18. Wen, M., Horton, M. K., Munro, J. M., Huck, P. & Persson, K. A. An equivariant graph neural network for the elasticity tensors of all seven crystal systems. Digit. Discov. 3, 869–882 (2024).10.1039/D3DD00233K ↗
19. Wang, J. et al. E(n)-equivariant cartesian tensor message passing interatomic potential. Nat. Commun. 15, 7607 (2024).10.1038/s41467-024-51886-6 ↗
20. Geiger, M. & Smidt, T. E3nn: euclidean neural networks. Arxiv Preprint Arxiv:2207.09453 (2022).
21. Jain, A. et al. Commentary: the materials project: a materials genome approach to accelerating materials innovation. Apl. Mater. 1, 11002 (2013).10.1063/1.4812323 ↗
22. Ong, S. P. et al. Python materials genomics (pymatgen): a robust, open-source python library for materials analysis. Comput. Mater. Sci. 68, 314–319 (2013).10.1016/j.commatsci.2012.10.028 ↗
23. Ward, L., Agrawal, A., Choudhary, A. & Wolverton, C. A general-purpose machine learning framework for predicting properties of inorganic materials. Npj Comput. Mater. 2, 16028 (2016).10.1038/npjcompumats.2016.28 ↗
24. Choudhary, K. et al. The joint automated repository for various integrated simulations (jarvis) for data-driven materials design. Npj Comput. Mater. 6, 173 (2020).10.1038/s41524-020-00440-1 ↗
25. Pedregosa, F. et al. Scikit-learn: machine learning in python. J. Mach. Learn. Res. 12, 2825–2830 (2011).
26. Kholkin, A. L., Pertsev, N. A. & Goltsev, A. V. in Piezoelectric and acoustic materials for transducer applications (Springer, 2008), pp. 17-38.10.1007/978-0-387-76540-2_2 ↗
27. Zou, W., Tang, C. & Pan, E. Symmetry types of the piezoelectric tensor and their identification. Proc. R. Soc. a: Math., Phys. Eng. Sci. 469, 20120755 (2013).10.1098/rspa.2012.0755 ↗
28. Chen, Y., Ming, Z., Qi, L. & Zou, W. A polynomially irreducible functional basis of hemitropic invariants of piezoelectric tensors. Arxiv Preprint Arxiv:1901.01701 (2019).
29. Itin, Y. & Reches, S. Decomposition of third-order constitutive tensors. Math. Mech. Solids 27, 222–249 (2022).10.1177/10812865211016530 ↗
30. Choudhary, K. et al. High-throughput density functional perturbation theory and machine learning predictions of infrared, piezoelectric, and dielectric responses. Npj Comput. Mater. 6, 64 (2020).10.1038/s41524-020-0337-2 ↗
31. Chen, T. & Guestrin, C. Xgboost: a scalable tree boosting system. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, 785-7942016).10.1145/2939672.2939785 ↗
32. Wang, Z. & Dong, S. Large in-plane negative piezoelectricity and giant nonlinear optical susceptibility in elementary ferroelectric monolayers. Phys. Rev. B 108, 235423 (2023).10.1103/PhysRevB.108.235423 ↗
33. Haastrup, S. et al. The computational 2d materials database: high-throughput modeling and discovery of atomically thin crystals. 2D Mater. 5, 42002 (2018).10.1088/2053-1583/aacfc1 ↗
34. Gjerding, M. N. et al. Recent progress of the computational 2d materials database (c2db). 2D Mater. 8, 44002 (2021).10.1088/2053-1583/ac1059 ↗
35. Breiman, L. Random forests. Mach. Learn. 45, 5–32 (2001).10.1023/A:1010933404324 ↗
36. Zhou, J. et al. 2dmatpedia, an open computational database of two-dimensional materials from top-down and bottom-up approaches. Sci. Data 6, 86 (2019).10.1038/s41597-019-0097-3 ↗
37. Cohen, R. E., Krakauer, H. & Sághi-Szabó, G. First-principles study of piezoelectricity in pbtio3. Phys. Rev. Lett. 80, 4321–4324 (1998).10.1103/PhysRevLett.80.4321 ↗
38. Kresse, G. & Furthmüller, J. Efficient iterative schemes for ab initio total-energy calculations using a plane-wave basis set. Phys. Rev. B 54, 11169 (1996).10.1103/PhysRevB.54.11169 ↗
39. Larsen, A. H. et al. The atomic simulation environment—a python library for working with atoms. J. Phys.: Condens. Matter 29, 273002 (2017).
40. Perdew, J. P., Burke, K. & Ernzerhof, M. Generalized gradient approximation made simple. Phys. Rev. Lett. 77, 3865 (1996).10.1103/PhysRevLett.77.3865 ↗
41. Gonze, X. & Lee, C. Dynamical matrices, born effective charges, dielectric permittivity tensors, and interatomic force constants from density-functional perturbation theory. Phys. Rev. B 55, 10355 (1997).10.1103/PhysRevB.55.10355 ↗
42. Hendrycks, D. & Gimpel, K. Gaussian error linear units (gelus). Arxiv Preprint Arxiv:1606.08415 (2023).
43. Weiler, M., Geiger, M., Welling, M., Boomsma, W. & Cohen, T. S. 3d steerable cnns: learning rotationally equivariant features in volumetric data. 31, 10402–10413 (2018).
44. Loshchilov, I. & Hutter, F. Decoupled weight decay regularization. In Proceedings of the International Conference on Learning Representations (2017).