Peng, Sitian, Jiang, Jingwen, Ou, Qiaofeng, Xiong, Bangshu, Liu, Chang, and Rao, Zhibo
Morphological filters are widely used in data pre-processing, whereas the existing morphological filters are not optimal in terms of performance, since they are designed by combining the basic morphological operators together artificially. Hence, we proposed an automatic morphological neural network design method based on differentiable architecture search (DARTS), which is a neural architecture search (NAS) method. Firstly, the mechanism is revealed that the morphological dilation with flat structural elements behaves the same as Maxpooling without down-sampling. Secondly, morphological dilation and erosion neural network layers were defined. Thirdly, using DARTS and the search space with morphological neural network layers, end-to-end morphological neural network was implemented. Finally, the morphological neural network was successfully applied in bearing fault diagnosis. Experiments on the three datasets show that the proposed method improved the feature extraction capabilities of neural networks, then achieved end-to-end bearing fault diagnosis, and the diagnosis accuracy was improved by more than 2.0%. • For the first time, the morphological operator is embedded in the neural network, which improves the signal feature extraction and suppresses the background noise. • For the first time, the DARTS algorithm is applied to search for the optimal morphological operator combination, which avoids the dependence of artificial experience and improves efficiency. • The morphological neural network obtained based on DARTS was applied in bearing fault diagnosis, and improved the diagnosis accuracy. [ABSTRACT FROM AUTHOR]