Physics Aware Representation Learning on Electronic Charge Density for Materials Property Prediction

Abstract

The fundamental quantity governing the mechanical and thermodynamic properties of a crystalline solid is its electronic charge density. Yet, its direct use for the rapid prediction of materials properties remains challenging due to its high dimensionality. Here, we present a physics-informed deep learning framework that directly predicts mechanical and thermodynamic properties from the three-dimensional electronic charge density derived from density functional theory (DFT). The proposed approach first utilizes a three-dimensional convolutional autoencoder for unsupervised dimensionality reduction, compressing a high-resolution charge-density grid (128 x 128 x 128) into a compact latent representation (16 x 16 x 16 x 16) while preserving physically meaningful features, as confirmed by negligible reconstruction errors across diverse crystal systems. The compressed latent-space representation of charge density is then used by two different regression models for property prediction: Light Gradient Boosting Machine (LightGBM) and Attention-based 3D Convolutional Neural Networks (Att CNN), and their performance is compared. Combining composition-based descriptors (Material Agnostic Platform for Informatics and Exploration or MAGPIE) with electronic charge density data further improves the model accuracy. Using a dataset of about 6059 inorganic compounds spanning multiple crystal symmetries, the models achieve strong predictive performance for bulk modulus K (R2 = 0.94), Young's modulus E (R2 = 0.88), shear modulus G (R2 = 0.87), formation energy Eform (R2 = 0.96), and Debye temperature Θ (R2 = 0.89). This work establishes electronic charge density as a transferable, physics-grounded descriptor for materials property prediction, requiring ~ 1/25 the computational resources of full-fledged DFT calculations.

References in corpus (13)

• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials
• Insightful classification of crystal structures using deep learning
• Generative adversarial networks (GAN) based efficient sampling of chemical space for inverse design of inorganic materials
• High throughput quantitative metallography for complex microstructures using deep learning: A case study in ultrahigh carbon steel
• Efficient machine-learning model for fast assessment of elastic properties of high-entropy alloys
• Predicting Elastic Properties of Materials from Electronic Charge Density Using 3D Deep Convolutional Neural Networks
• Accelerating microstructure modelling via machine learning: a new method combining Autoencoder and ConvLSTM
• Reproducibility in Computational Materials Science: Lessons from 'A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials'
• Lean CNNs for mapping electron charge density fields to material properties
• Deep Learning Assisted Denoising of Experimental Micrographs
• Microstructural Studies Using Generative Adversarial Network (GAN): a Case Study
• Deep Learning-Driven Prediction of Microstructure Evolution via Latent Space Interpolation