Pre-trained network-based transfer learning: A small-sample machine learning approach to nuclear power plant classification problem
- University of Pittsburgh, PA (United States); OSTI
- University of Pittsburgh, PA (United States)
Some research topics belonging to classification problems in the nuclear industry, such as fault diagnosis and accident identification, can be solved by feature extraction and subsequent application of statistical machine learning classifiers. Recently, deep neural network-based methods with automatic feature extraction and high accuracy have gained wide attention. They usually require large-scale training data, however, plant fault or accident data are scarce or difficult to obtain. Here this paper proposes a convolutional network (CNN)-based transfer learning method to solve this problem. The network's shallow layer is derived from a pre-trained CNN based on the ImageNet database to automatically extract features, and the deep layer is customized to match the classification problem. Data in non-image formats are converted to image formats and subsequently used to train the network. Case studies of rotating machines fault diagnosis show that the proposed method requires only limited training data to achieve high accuracy.
- Research Organization:
- University of Pittsburgh, PA (United States)
- Sponsoring Organization:
- USDOE Office of Nuclear Energy (NE); University of Pittsburgh
- Grant/Contract Number:
- NE0008909
- OSTI ID:
- 1976818
- Journal Information:
- Annals of Nuclear Energy, Journal Name: Annals of Nuclear Energy Journal Issue: C Vol. 175; ISSN 0306-4549
- Publisher:
- ElsevierCopyright Statement
- Country of Publication:
- United States
- Language:
- English
Similar Records
Sparse coding of pathology slides compared to transfer learning with deep neural networks
Comparing Deep Learning Approaches for Understanding Genotype × Phenotype Interactions in Biomass Sorghum