Using deep neural network with small dataset to predict material defects

By Shuo Feng, Huiyu Zhou, Hongbiao Dong. Materials & Design (1980-2015), 2019

447 citations recorded by the source corpus.

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.

Subfields: Industrial and Manufacturing Engineering, Materials Chemistry, Mechanical Engineering

View Paper: https://www.semanticscholar.org/paper/837f34f5f08c0a03fdca09e1b17f702e0111baba

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

  1. Kamran Shaukat
  2. Ibrahim A. Hameed
  3. Suhuai Luo
  4. Min Xu
  5. V. Varadharajan

Citations in the corpus