University of Leicester
Browse
cleaned manuscript Application of deep transfer learning to.docx (8.02 MB)

Application of deep transfer learning to predicting crystal structures of inorganic substances

Download (8.02 MB)
journal contribution
posted on 2021-04-28, 16:18 authored by S Feng, Huiyu Zhou, H Dong
Convolutional neural network (CNN) consists of shallow learning machine and automatic feature extractor. The feature extractor of a well-trained CNN on a big dataset can be reused in related tasks with small datasets. This technique is called deep transfer learning which not only bypasses manual feature engineering but also improves the generalization of new models. In this study, we attempted to predict crystal structures of inorganic substances, a challenge for material science, with CNN and transfer learning. CNNs were trained on a big dataset of 228 k compounds from open quantum materials database (OQMD). The feature extractors of the well-trained CNNs were reused for extracting features on a phase prototypes dataset (containing 17 k inorganic substances and involving 170 crystal structures) and two high-entropy alloy datasets. The extracted features were then fed into random forest classifier as input. High classification accuracy (above 0.9) was achieved in three datasets. The visualization of the extracted features proved the effectiveness of the transferable feature extractors. This method can be easily adopted in quickly building machine learning models of good performance without resorting to time-consuming manual feature engineering routes.

History

Citation

Computational Materials Science, Volume 195, July 2021, 110476

Author affiliation

Department of Engineering

Version

  • AM (Accepted Manuscript)

Published in

Computational Materials Science

Volume

195

Publisher

Elsevier

issn

0927-0256

Acceptance date

2021-03-29

Copyright date

2021

Available date

2022-04-12

Language

en

Usage metrics

    University of Leicester Publications

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC