Guided sampling-based evolutionary deep neural network for intelligent fault diagnosis

• Published in: Engineering applications of artificial intelligence Vol. 128; p. 107498
• Main Authors: Sharma, Arun K., Verma, Nishchal K.
• Format: Journal Article
• Language: English
• Published: Elsevier Ltd 01.02.2024
• Subjects: Deep neural network; Intelligent fault diagnosis; Neural architecture search; Non-dominated sorting algorithm; Policy gradient; Deep neural network; Non-dominated sorting algorithm; Policy gradient; Neural architecture search; Intelligent fault diagnosis
• ISSN: 0952-1976; 1873-6769
• DOI: 10.1016/j.engappai.2023.107498

The selection of model architecture and hyperparameters has a significant impact on the diagnostic performance of most deep learning models. Because training and evaluating the various architectures of deep learning models is a time-consuming procedure, manual selection of model architecture becomes infeasible. Therefore, we have proposed a novel framework for evolutionary deep neural networks that uses a policy gradient to guide the evolution of the DNN architecture towards maximum diagnostic accuracy. We have formulated a policy gradient-based controller that generates an action to sample the new model architecture at every generation so that optimality is obtained quickly. The fitness of the best model obtained is used as a reward to update the policy parameters. Also, the best model obtained is transferred to the next generation for quick model evaluation in the NSGA-II evolutionary framework. Thus, the algorithm gets the benefits of fast non-dominated sorting as well as quick model evaluation. The effectiveness of the proposed framework has been validated on three datasets: the Air Compressor dataset, the Case Western Reserve University dataset, and the Paderborn University dataset.