[Paper Review] DeePAW: A universal machine learning model for orbital-free ab initio calculations

Tianhao Su, Shunbo Hu+6 more|arXiv (Cornell University)|Mar 19, 2026

Machine Learning in Materials Science0 citations

TL;DR

DeePAW presents a universal ML model for orbital-free DFT calculations that achieves broad element coverage, diverse structure applicability, and high accuracy without extra fine-tuning, enabled by SE(3)-equivariant double massage passing neuron networks. It also predicts formation energies to enable multiscale materials modeling.

ABSTRACT

Developing universal machine learning models for ab initio calculations is the frontier of materials cutting edge research in the new era of artificial intelligence. Here, we present the Deep Augment Way model (DeePAW) that is a universal machine learning (ML) model for orbital-free (OF) ab initio calculations, based on the density functional theory (DFT). DeePAW is currently the best OFDFT ML model according to the three criterions, 1) covering the largest number of elements, 2) having the widest application capability to diverse crystal structures, and 3) achieving the highest prediction accuracy without further fine-tuning. These scientific merits and innovations of DeePAW are stemmed from the novel SE(3)-equivariant double massage passing neuron networks. Besides predicting electron density distributions, DeePAW predicts formation energies of crystals as well and therefore paves an efficient avenue for multiscale materials modeling beyond conventional electronic structure calculation methods.

Motivation & Objective

• Advance universal machine learning models for orbital-free ab initio calculations within DFT frameworks.
• Increase elemental coverage and structural versatility of OFDFT models.
• Achieve high prediction accuracy without additional fine-tuning.
• Enable prediction of formation energies for crystals to support multiscale materials modeling.

Proposed method

• Propose the Deep Augment Way (DeePAW) model as a universal ML framework for orbital-free DFT.
• Utilize SE(3)-equivariant double massage passing neuron networks to handle symmetry-aware learning.
• Predict electron density distributions and crystal formation energies.
• Demonstrate broad element coverage and applicability to diverse crystal structures.
• Aim for high accuracy without requiring task-specific fine-tuning.

Experimental results

Research questions

• RQ1 Can a single ML model accurately perform orbital-free ab initio calculations across many elements and crystal structures without fine-tuning?
• RQ2 How well does DeePAW predict electron density distributions and crystal formation energies within OFDFT?
• RQ3 What advantages do SE(3)-equivariant networks provide for orbital-free learning in materials science?
• RQ4 To what extent can the model support multiscale materials modeling beyond conventional electronic structure methods.

Key findings

• DeePAW achieves broad elemental coverage compared to existing OFDFT ML models.
• The model exhibits wide applicability to diverse crystal structures.
• It attains high prediction accuracy without the need for additional fine-tuning.
• DeePAW predicts formation energies in addition to electron densities, enabling multiscale materials modeling.