Differentiable neural architecture search for domain adaptation in fault diagnosis

In recent years, the application of deep transfer learning in fault diagnosis has gained considerable traction. Notably, the utilization of domain adaptation methods to address the challenges of unsupervised transfer learning has emerged as a focal point of research inquiry. However, prevailing investigations primarily emphasize the construction of loss functions, leaving the design of network structures reliant on manual intervention. To overcome this limitation, a novel approach employing differentiable neural architecture search is proposed. This method facilitates the automatic exploration of network structures that are well-suited for domain adaptation in fault diagnosis. The proposed method explores two key components within the Inception search space: the arrangement of Inception blocks and the feature mask. By stacking Inception blocks, a comprehensive network architecture is constructed, while the feature mask selectively incorporates domain-relevant features. Moreover, to enhance search efficiency, simultaneous exploration and adaptation of multiple target domains are undertaken, thereby expanding the potential application domains of domain adaptation. The effectiveness of the proposed method is evaluated using 3 public datasets through a series of experiments, which demonstrate that this method exhibits a remarkable capacity for generating networks of satisfied performance.

Automated summary

In recent years, deep transfer learning has been extensively used in fault diagnosis. Research has focused on utilizing domain adaptation methods in unsupervised transfer learning. However, there has been a lack of emphasis on the design of network structures. To address this, a novel approach that uses differentiable neural architecture search for automatic exploration of network structures suitable for fault diagnosis is proposed. The method explores key components within the Inception search space, resulting in the construction of comprehensive network architectures that selectively incorporate domain-relevant features. The effectiveness of the proposed method is demonstrated through experiments using public datasets, showing its remarkable capacity in generating networks with satisfactory performance.