CHILI: Chemically-Informed Large-scale Inorganic Nanomaterials Dataset for Advancing Graph Machine Learning

Abstract

Advances in graph machine learning (ML) have been driven by applications in chemistry as graphs have remained the most expressive representations of molecules. While early graph ML methods focused primarily on small organic molecules, recently, the scope of graph ML has expanded to include inorganic materials. Modelling the periodicity and symmetry of inorganic crystalline materials poses unique challenges, which existing graph ML methods are unable to address. Moving to inorganic nanomaterials increases complexity as the scale of number of nodes within each graph can be broad ($10$ to $10^5$). The bulk of existing graph ML focuses on characterising molecules and materials by predicting target properties with graphs as input. However, the most exciting applications of graph ML will be in their generative capabilities, which is currently not at par with other domains such as images or text. We invite the graph ML community to address these open challenges by presenting two new chemically-informed large-scale inorganic (CHILI) nanomaterials datasets: A medium-scale dataset (with overall >6M nodes, >49M edges) of mono-metallic oxide nanomaterials generated from 12 selected crystal types (CHILI-3K) and a large-scale dataset (with overall >183M nodes, >1.2B edges) of nanomaterials generated from experimentally determined crystal structures (CHILI-100K). We define 11 property prediction tasks and 6 structure prediction tasks, which are of special interest for nanomaterial research. We benchmark the performance of a wide array of baseline methods and use these benchmarking results to highlight areas which need future work. To the best of our knowledge, CHILI-3K and CHILI-100K are the first open-source nanomaterial datasets of this scale -- both on the individual graph level and of the dataset as a whole -- and the only nanomaterials datasets with high structural and elemental diversity.

16 pages, 15 figures, 8 tables. Dataset is available at https://github.com/UlrikFriisJensen/CHILI

References in corpus (17)

• Deep Learning in Neural Networks: An Overview
• A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials
• The Open Catalyst 2020 (OC20) Dataset and Community Challenges
• Atomistic Line Graph Neural Network for Improved Materials Property Predictions
• Open Graph Benchmark: Datasets for Machine Learning on Graphs
• Predicting materials properties without crystal structure: Deep representation learning from stoichiometry
• The Open Catalyst 2022 (OC22) Dataset and Challenges for Oxide Electrocatalysts
• Improving Graph Neural Network Expressivity via Subgraph Isomorphism Counting
• Global Attention based Graph Convolutional Neural Networks for Improved Materials Property Prediction
• A foundation model for atomistic materials chemistry
• Global Self-Attention as a Replacement for Graph Convolution
• Graph Neural Networks for Molecules
• MatterGen: a generative model for inorganic materials design
• JARVIS-Leaderboard: A Large Scale Benchmark of Materials Design Methods
• High-accuracy thermodynamic properties to the melting point from ab initio calculations aided by machine-learning potentials
• IRNet: A General Purpose Deep Residual Regression Framework for Materials Discovery
• IntelliGraphs: Datasets for Benchmarking Knowledge Graph Generation