Cross-property deep transfer learning framework for enhanced predictive analytics on small materials data

Vishu Gupta, Kamal Choudhary, Francesca Tavazza, Carelyn Campbell, Wei keng Liao, Alok Choudhary, Ankit Agrawal*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

81 Scopus citations

Abstract

Artificial intelligence (AI) and machine learning (ML) have been increasingly used in materials science to build predictive models and accelerate discovery. For selected properties, availability of large databases has also facilitated application of deep learning (DL) and transfer learning (TL). However, unavailability of large datasets for a majority of properties prohibits widespread application of DL/TL. We present a cross-property deep-transfer-learning framework that leverages models trained on large datasets to build models on small datasets of different properties. We test the proposed framework on 39 computational and two experimental datasets and find that the TL models with only elemental fractions as input outperform ML/DL models trained from scratch even when they are allowed to use physical attributes as input, for 27/39 (≈ 69%) computational and both the experimental datasets. We believe that the proposed framework can be widely useful to tackle the small data challenge in applying AI/ML in materials science.

Original languageEnglish (US)
Article number6595
JournalNature communications
Volume12
Issue number1
DOIs
StatePublished - Dec 2021

Funding

This work was performed under the following financial assistance award 70NANB19H005 from U.S. Department of Commerce, National Institute of Standards and Technology as part of the Center for Hierarchical Materials Design (CHiMaD). Partial support is also acknowledged from Department of Energy (DOE) awards DESC0014330, DE-SC0019358, and DE-SC0021399.

ASJC Scopus subject areas

  • General Physics and Astronomy
  • General Chemistry
  • General Biochemistry, Genetics and Molecular Biology

Fingerprint

Dive into the research topics of 'Cross-property deep transfer learning framework for enhanced predictive analytics on small materials data'. Together they form a unique fingerprint.

Cite this