Capsule graph networks for accurate and interpretable crystalline materials property prediction

用于准确且可解释地预测晶体材料性质的胶囊图网络

阅读:2

Abstract

Accurate and interpretable modeling of crystalline materials is essential for understanding the structure-property relationships in materials critical in accelerating materials discovery. While recent graph neural networks (GNNs) have achieved high predictive accuracy, they often struggle to provide physical interpretability and fail to explicitly model the hierarchical and symmetrical nature of crystals. In this work, we introduce Capsule Graph Networks with E(3)-Equivariance (CGN-e3), a novel deep learning framework that integrates equivariant message passing with capsule networks to capture both geometric symmetries and hierarchical motif structures. CGN-e3 leverages E(3)-equivariant message passing to learn physically consistent features and organize them into capsule representations that can disentangle local motifs, such as polyhedral environments, and connects them to global properties. We validate the effectiveness of our framework on bandgap and formation energy prediction, as well as material classification using Materials Project and Matbench datasets. Our model achieves a MAE of 0.054 eV/atom and 0.379 eV on formation energy and bandgap prediction, respectively, outperforming CGCNN and matching the performance of MEGNet on the same dataset, while also providing insightful interpretations of the learned capsule representations.Scientific contribution: We present the first integration of E(3)-equivariant graph neural networks with capsule networks for modeling crystalline materials. This unified architecture captures both the fundamental physical symmetries of 3D space; rotation, translation, reflection and the intrinsic hierarchical part-whole relationships e.g., atoms to polyhedra to extended motifs found in crystal structures. The framework provides an unsupervised pathway for interpretable motif discovery. The dynamic routing-by-agreement mechanism identifies and aggregates structurally significant local environments such as the TiO6 octahedra into higher-order graph-level capsules. This process yields human-intelligible insights by explicitly quantifying the contribution of specific structural motifs to target material properties, moving beyond "black-box" predictions. We validate our framework on key property prediction tasks and provide capsule-level interpretation of the results.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。