An Advanced Convolutional Neural Network for Bearing Fault Diagnosis under Limited Data

2509.11053v1 cs.LG, cs.AI, cs.CE 2025-09-17
Авторы:

Shengke Sun, Shuzhen Han, Ziqian Luan, Xinghao Qin, Jiao Yin, Zhanshan Zhao, Jinli Cao, Hua Wang

Резюме на русском

## Контекст Проблемы диагностики дефектов в подшипниках широко распространены в машиностроении и технологиях. Эти дефекты могут привести к серьезным последствиям, включая аварии и необходимость в дорогостоящих ремонтах. Большинство текущих систем диагностики используют глубокие нейронные сети, но эти методы требуют больших объемов данных для эффективного обучения. Однако в реальных условиях общего доступа к высококачественным метканным данным часто ограничено из-за высоких затрат или беспрецедентных конфиденциальности. Этот факт ставит под угрозу эффективность существующих систем диагностики. Многие методы глубокого обучения, такие как традиционные согласованные нейронные сети, сталкиваются с проблемами избыточностью и неэффективностью при обработке недостаточного количества данных. Для решения этих проблем необходимо разработать более устойчивые и эффективные алгоритмы для диагностики дефектов в подшипниках, которые могут работать с ограниченными данными. ## Метод Мы предлагаем новую архитектуру свёрточной нейронной сети, названную "DAC-FCF", которая основывается на двух основных компонентах: 1. **Генерирующая сеть GAN**: Мы используем новую версию генерирующей сети GAN, названную **Conditional Consistent Latent Representation and Reconstruction GAN (CCLR-GAN)**. Эта сеть генерирует более разнообразные и качественные данные для обучения, используя условные представления для поддержки контекста данных. 2. **Обучение с помощью механизма контрастного обучения**: Мы вводим новый механизм, использующий **контрастное обучение** для моделирования отношений между доступными данными. Это позволяет сети учиться более эффективно даже при недостаточном количестве данных. 3. **1D Fourier Convolution Neural Network (1D-FCNN)**: Мы также предлагаем использовать 1D Fourier Convolution, которая может достаточно хорошо извлекать глобальные межсигнальные отношения в звуковых сигналах. Это позволяет сети более эффективно ориентироваться на глобальные модели дефектов в данных. ## Результаты Мы провели эксперименты на двух датасетах: **CWRU (Case Western Reserve University)** и нашем собственном тестовом штабе. На CWRU мы достигли улучшения до 32% в сравнении с базовыми методами. На тестовом штабе — до 10%. Помимо этого, мы провели анализ абляции, который показал значительное влияние каждого отдельного компонента на результат. Эти результаты демонстрируют указанную выше систему DAC-FCF как эффективную и надежную альтернативу существующим методам, особенно в условиях нехватки данных. ## Значимость Алгоритм DAC-FCF может быть применён в различных областях, таких как

Abstract

In the area of bearing fault diagnosis, deep learning (DL) methods have been widely used recently. However, due to the high cost or privacy concerns, high-quality labeled data are scarce in real world scenarios. While few-shot learning has shown promise in addressing data scarcity, existing methods still face significant limitations in this domain. Traditional data augmentation techniques often suffer from mode collapse and generate low-quality samples that fail to capture the diversity of bearing fault patterns. Moreover, conventional convolutional neural networks (CNNs) with local receptive fields makes them inadequate for extracting global features from complex vibration signals. Additionally, existing methods fail to model the intricate relationships between limited training samples. To solve these problems, we propose an advanced data augmentation and contrastive fourier convolution framework (DAC-FCF) for bearing fault diagnosis under limited data. Firstly, a novel conditional consistent latent representation and reconstruction generative adversarial network (CCLR-GAN) is proposed to generate more diverse data. Secondly, a contrastive learning based joint optimization mechanism is utilized to better model the relations between the available training data. Finally, we propose a 1D fourier convolution neural network (1D-FCNN) to achieve a global-aware of the input data. Experiments demonstrate that DAC-FCF achieves significant improvements, outperforming baselines by up to 32\% on case western reserve university (CWRU) dataset and 10\% on a self-collected test bench. Extensive ablation experiments prove the effectiveness of the proposed components. Thus, the proposed DAC-FCF offers a promising solution for bearing fault diagnosis under limited data.

Ссылки и действия