An Evolutionary One-Shot Neural Architecture Search Method based on Single Path Cells towards Physics-Informed Fault Diagnosis

Fault diagnosis is important for the complex equipment, and physics-informed fault diagnosis has become a development trend. While, physics-informed fault diagnosis is hard to realize, unless several problems have been addressed, and one of the limitations is searching for the best architecture, which influences the performance greatly. Neural architecture search (NAS) has been a research hotspot. However, limited by the computing resources and the deviation of supernet prediction, NAS might miss the best architecture, and impedes the application of NAS in physics-informed fault diagnosis greatly. Thus, this paper proposes an evolutionary one-shot NAS method based on Single Path Cells (SPC-NAS) for physics-informed fault diagnosis. The proposed method develops a new supernet based on single path cells, to reduce the computing resources and improve the reusability. And, an improved supernet training method is introduced to reduce the deviation between the one-shot model prediction and the stand-alone model accuracy. Finally, an evolutionary search strategy with constraint is developed to find the best architecture. The experimental results show that the proposed method can automatically find the best architecture for different tasks, which achieved an accuracy of 100% in CWRU dataset with only 0.292M parameters. All the results indicate that the proposed method can address the limitation to search the best architecture and show the potential for physics-informed fault diagnosis.