Using deep neural network with small dataset to predict material defects

Shuo Feng, Huiyu Zhou, Hongbiao Dong*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

129 Citations (Scopus)
3 Downloads (Pure)

Abstract

Deep neural network (DNN) exhibits state-of-the-art performance in many fields including microstructure recognition where big dataset is used in training. However, DNN trained by conventional methods with small datasets commonly shows worse performance than traditional machine learning methods, e.g. shallow neural network and support vector machine. This inherent limitation prevented the wide adoption of DNN in material study because collecting and assembling big dataset in material science is a challenge. In this study, we attempted to predict solidification defects by DNN regression with a small dataset that contains 487 data points. It is found that a pre-trained and fine-tuned DNN shows better generalization performance over shallow neural network, support vector machine, and DNN trained by conventional methods. The trained DNN transforms scattered experimental data points into a map of high accuracy in high-dimensional chemistry and processing parameters space. Though DNN with big datasets is the optimal solution, DNN with small datasets and pre-training can be a reasonable choice when big datasets are unavailable in material study.

Original languageEnglish
Pages (from-to)300-310
Number of pages11
JournalMaterials and Design
Volume162
DOIs
Publication statusPublished - 15 Jan 2019
Externally publishedYes

Bibliographical note

Funding Information:
Shuo Feng wishes to acknowledge EPSRC CDT (Grant No: EP/L016206/1 ) in Innovative Metal Processing for providing a PhD studentship for this study.

Funding Information:
Shuo Feng wishes to acknowledge EPSRC CDT (Grant No: EP/L016206/1) in Innovative Metal Processing for providing a PhD studentship for this study.

Publisher Copyright:
© 2018

Copyright:
Copyright 2019 Elsevier B.V., All rights reserved.

Keywords

  • Deep neural network
  • Defects
  • Pre-training
  • Small dataset

ASJC Scopus subject areas

  • Materials Science(all)
  • Mechanics of Materials
  • Mechanical Engineering

Fingerprint

Dive into the research topics of 'Using deep neural network with small dataset to predict material defects'. Together they form a unique fingerprint.

Cite this