Employing Graph Neural Networks for Predicting Electrode Average Voltages and Screening High-Voltage Sodium Cathode Materials

Abstract

For many years, humans have been relentlessly focused on enhancing battery longevity and boosting energy storage capacities. The performance and durability of a battery depend significantly on the material used for its electrodes. In this context, merging machine learning with density functional theory (DFT) calculations has emerged as a pivotal approach to advancing the exploration of battery crystal structures. We present a new method that combines a graph convolutional neural network (GNN) with a Transformer convolutional layer, which we call Transformer-GNN. To underscore its efficacy, we benchmarked Transformer-GNN against three established statistical machine learning models: Support Vector Machine, Random Forest, and XGBoost. We also developed a standard GNN, which we refer to as Basic-GNN. Additionally, we compared Basic-GNN with Transformer-GNN to highlight the improvements brought about by incorporating the Transformer convolutional layer. The Transformer-GNN model outperforms the other models, achieving the highest R² value of 0.82 and the lowest mean squared error of 0.3161. Our findings demonstrate that the Transformer-GNN can profoundly understand battery crystal structures, thus forging the path toward more sophisticated and durable battery systems. Leveraging the GNN model’s voltage predictions in tandem with the capacity data sourced from the database, we screened and pinpointed Na(NiO₂)₂ as a high-voltage (higher than 5 V), high-capacity sodium cathode material. We conducted DFT calculations on Na(NiO₂)₂ and revealed the migration mechanism of the Na ions.