HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

OQM9HK: A Large-Scale Graph Dataset for Machine Learning in Materials Science

{Takenori Yamamoto}

OQM9HK: A Large-Scale Graph Dataset for Machine Learning in Materials Science

Abstract

We introduce a large-scale dataset of quantum-mechanically calculated properties of crystalline materials for graph representation learning that contains approximately 900k entries (OQM9HK). This dataset is constructed on the basis of the Open Quantum Materials Database (OQMD) v1.5 containing more than one million entries, and is the successor to the OQMD v1.2 dataset containing approximately 600k entries (OQM6HK). We develop the graph creation algorithm to produce a binary edge-labeled (BEL) graph representing a crystalline material. The BEL graph has higher representability of crystal structure than the edge-unlabeled ones. In materials property prediction tasks, crystal graph neural networks trained on the BEL graph dataset perform better than ones on the other graph datasets. The OQM9HK graph dataset is available at the Zenodo repository, https://doi.org/10.5281/zenodo.7124330

Benchmarks

BenchmarkMethodologyMetrics
formation-energy-on-oqm9hkCGNN Trio Ensemble
MAE: 0.03658
formation-energy-on-oqm9hkCGNN
MAE: 0.04249 ± 0.00037
formation-energy-on-oqm9hkCGNN Full Ensemble
MAE: 0.03433

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
OQM9HK: A Large-Scale Graph Dataset for Machine Learning in Materials Science | Papers | HyperAI