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.
Ссылки и действия
Дополнительные ресурсы: