Abstract:Machine learning method has been widely applied in industrial fault diagnosis, especially the deep learning method. In the field of industrial fault diagnosis, deep learning is mostly used to extract features of vibration signals to achieve end-to-end fault diagnosis systems. Due to the complexity and variety of actual industrial datasets, some deep learning models are designed to be complicated. However, designing neural network architectures requires rich professional knowledge, experience, and a large number of experiments, increasing the difficulty of developing deep learning models. Fortunately, Neural Architecture Search (NAS), a branch of Automated Machine Learning (AutoML), is developing rapidly. Given a search space, NAS can search for networks that perform better than manually designed. In this paper, a one-shot NAS method for fault diagnosis is proposed. The one-shot model is a supernet that contains all candidate networks in a given search space. The supernet is trained to evaluate the actual performance of candidate networks by measuring the difference between its output probability and the true labels. According to the prediction of supernet, the networks with excellent performance can be searched, using some common search methods such as random search or evolutionary algorithm. Finally, the searched network is trained by reusing the weights of the supernet. To evaluate the proposed method, two search spaces are designed, ResNet and Inception search spaces, to search on PHM 2009 Data Challenge gearbox dataset. The state-of-the-art results are obtained, and accuracies of searched ResNet-A and Inception-A are 84.11% and 83.81%, which are 3.29% and 10.88% higher than Reinforcement Learning based NAS.

Differentiable neural architecture search augmented with pruning and multi-objective optimization for time-efficient intelligent fault diagnosis of machinery

Differentiable Architecture Search for Aeroengine Bevel Gear Fault Diagnosis

Differentiable Sampling Based Efficient Architecture Search for Automatic Fault Diagnosis

Bayesian Differentiable Architecture Search for Efficient Domain Matching Fault Diagnosis

Neural Architecture Search For Fault Diagnosis

Automated Model Generation for Machinery Fault Diagnosis Based on Reinforcement Learning and Neural Architecture Search

A new differentiable architecture search method for optimizing convolutional neural networks in the digital twin of intelligent robotic grasping

One-shot Neural Architecture Search for Fault Diagnosis Using Vibration Signals

A Neural Architecture Search Method with Multi-Dimensional Correlation Representation for Fault Diagnosis

Latency-Aware Differentiable Neural Architecture Search

Wavelet-driven differentiable architecture search for planetary gear fault diagnosis

Iterative-AMC: a novel model compression and structure optimization method in mechanical system safety monitoring

Graph neural network architecture search for rotating machinery fault diagnosis based on reinforcement learning

Towards Improving the Consistency, Efficiency, and Flexibility of Differentiable Neural Architecture Search

Neural Architecture Search for Multi-Sensor Information Fusion-Based Intelligent Fault Diagnosis

Enhanced Differentiable Architecture Search Based on Asymptotic Regularization

RL-DARTS: Differentiable neural architecture search via reinforcement-learning-based meta-optimizer

A neural architecture search method based on gradient descent for remaining useful life estimation

Partial Order Pruning: for Best Speed/Accuracy Trade-off in Neural Architecture Search

Automatic Routability Predictor Development Using Neural Architecture Search

Neural-Architecture-Search-Based Multiobjective Cognitive Automation System