Abstract:The burgeoning field of the Industrial Internet of Things (IIoT) necessitates advanced fault diagnosis methods capable of navigating the dual challenges of high-predictive accuracy and the constraints of edge computing environments. Our study introduces Wave-ConvNeXt, a novel fault diagnosis model that seamlessly integrates the state-of-the-art ConvNeXt architecture with Wavelet Transform. This innovative model stands out for its lightweight design yet delivers exceptional accuracy in fault diagnosis. In Wave-ConvNeXt, we re-engineer the ConvNeXt model for IIoT applications by adopting 1-D convolution, tailored for processing high-frequency, nonperiodic inputs. This adaptation is complemented by replacing the traditional "patchify" layer with a Wavelet transform layer, which simplifies input signals into subsignals, thereby easing learning complexities and diminishing the dependence on elaborate deep architectures. Further enhancing this model, we incorporate a squeeze-and-excitation module, enriching its ability to prioritize channelwise feature relevance, akin to self-attention mechanisms. This integration is rigorously validated through an ablation study. Wave-ConvNeXt epitomizes a holistic approach, enabling an end-to-end optimization of feature learning and fault classification. Our empirical analysis on two real-world IIoT data sets demonstrates Wave-ConvNeXt's superiority over existing models. It not only elevates prediction accuracy but also significantly curtails computational complexity. Additionally, our exploration into the impact of various mother wavelets reveals the effectiveness of using wavelet basis functions with smaller support, bolstering diagnostic precision. The source code of Wave-ConvNeXt is available at https://github.com/leviszhang/waveConvNeXt.

Wave-ConvNeXt: an Efficient and Precise Fault Diagnosis Method for IIoT Leveraging Tailored ConvNeXt and Wavelet Transform

Fusion Innovation: Multi-Scale Dilated Collaborative Model of ConvNeXt and MSDA for Fault Diagnosis

IWSNs with On-sensor Data Processing for Machine Fault Diagnosis

A Robust Wavelet-integrated Residual Network for Fault Diagnosis of Machines with Adversarial Training.

WaveletKernelNet: An Interpretable Deep Neural Network for Industrial Intelligent Diagnosis

Bearing Intelligent Fault Diagnosis in the Industrial Internet of Things Context: A Lightweight Convolutional Neural Network

WPConvNet: an Interpretable Wavelet Packet Kernel-Constrained Convolutional Network for Noise-Robust Fault Diagnosis

WavFormer: an Interpretable Wavelet-Constrained Transformer for Industrial Acoustics Diagnosis

Compound fault diagnosis for industrial robots based on dual-transformer networks

A newly-designed fault diagnostic method for transformers via improved empirical wavelet transform and kernel extreme learning machine

Rotating Machinery Fault Diagnosis Based on a Novel Lightweight Convolutional Neural Network

Comprehensive Intelligent Diagnosis for Mechanical and Insulation Faults of Power Equipment in the Power Internet of Things Context

A Novel Hierarchical Vision Transformer and Wavelet Time–Frequency Based on Multi-Source Information Fusion for Intelligent Fault Diagnosis

Gearbox Compound Fault Diagnosis in Edge-IoT Based on Legendre Multiwavelet Transform and Convolutional Neural Network

Dislocated Time Series Convolutional Neural Architecture: an Intelligent Fault Diagnosis Approach for Electric Machine

LightFD: Real-Time Fault Diagnosis with Edge Intelligence for Power Transformers

Highly imbalanced fault diagnosis of mechanical systems based on wavelet packet distortion and convolutional neural networks

Wavelet transform for rotary machine fault diagnosis:10 years revisited

MFSFormer: A Novel End-to-End Rotating Machinery Fault Diagnosis Framework for Noise and Small Samples

Data-driven Fault Diagnosis Method Based on the Conversion of Erosion Operation Signals into Images and Convolutional Neural Network

A wavelet convolutional capsule network with modified super resolution generative adversarial network for fault diagnosis and classification