npj Computational Materials (Sep 2021)

Deep learning framework for material design space exploration using active transfer learning and data augmentation

  • Yongtae Kim,
  • Youngsoo Kim,
  • Charles Yang,
  • Kundo Park,
  • Grace X. Gu,
  • Seunghwa Ryu

DOI
https://doi.org/10.1038/s41524-021-00609-2
Journal volume & issue
Vol. 7, no. 1
pp. 1 – 7

Abstract

Read online

Abstract Neural network-based generative models have been actively investigated as an inverse design method for finding novel materials in a vast design space. However, the applicability of conventional generative models is limited because they cannot access data outside the range of training sets. Advanced generative models that were devised to overcome the limitation also suffer from the weak predictive power on the unseen domain. In this study, we propose a deep neural network-based forward design approach that enables an efficient search for superior materials far beyond the domain of the initial training set. This approach compensates for the weak predictive power of neural networks on an unseen domain through gradual updates of the neural network with active transfer learning and data augmentation methods. We demonstrate the potential of our framework with a grid composite optimization problem that has an astronomical number of possible design configurations. Results show that our proposed framework can provide excellent designs close to the global optima, even with the addition of a very small dataset corresponding to less than 0.5% of the initial training dataset size.