English

Machine Learning for Electrode Materials: Property Prediction via Composition

Materials Science 2026-04-15 v2

Abstract

In this work, we benchmark three leading Machine Learning (ML) frameworks-MODNet, CrabNet, and a random forest model based on Magpie feature-for predicting properties of battery electrode materials using the Materials Project Battery Explorer dataset. We evaluate these models based on predictive accuracy, visualize numerical features using two-dimensional embeddings, and quantify performance using standard metrics. Our results demonstrate that CrabNet consistently outperforms the other models across all tests. To validate these findings, we employ robust statistical methods: bootstrap resampling and two cross-validation (CV) strategies (leave one cluster out and stratified 5-fold CV), comparing each model against a control baseline. In addition, we apply unsupervised clustering on MODNet-derived features using t-SNE and DBSCAN, revealing coherent material groupings without prior labels. This analysis confirms the robustness of the evaluated models and underscores the potential of ML-driven approaches for accelerating the electrode materials discovery. However, our study also identifies practical limitations and quantifies challenges associated with integrating ML models into materials science workflows. Despite these constraints, our findings suggest that ML models are highly effective for early-stage compositional screening in the battery industry. This work provides a foundation for future research on ML applications in materials discovery.

Keywords

Cite

@article{arxiv.2603.07805,
  title  = {Machine Learning for Electrode Materials: Property Prediction via Composition},
  author = {Hao Wu and Cameron Hargreaves and Arpit Mishra and Gian-Marco Rignanese},
  journal= {arXiv preprint arXiv:2603.07805},
  year   = {2026}
}

Comments

28 pages, 12 figures

R2 v1 2026-07-01T11:09:25.608Z