Next Article in Journal
Presenting a Multispectral Image Sensor for Quantification of Total Polyphenols in Low-Temperature Stressed Tomato Seedlings Using Hyperspectral Imaging
Previous Article in Journal
Manoeuvre Target Tracking in Wireless Sensor Networks Using Convolutional Bi-Directional Long Short-Term Memory Neural Networks and Extended Kalman Filtering
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
This is an early access version, the complete PDF, HTML, and XML versions will be available soon.
Article

Bearing-DETR: A Lightweight Deep Learning Model for Bearing Defect Detection Based on RT-DETR

1
School of Energy and Mining Engineering, Shandong University of Science and Technology, Qingdao 266590, China
2
School of Information Science and Engineering, Linyi University, Linyi 276002, China
3
School of Automation, Wuhan University of Technology, Wuhan 430070, China
*
Author to whom correspondence should be addressed.
Sensors 2024, 24(13), 4262; https://doi.org/10.3390/s24134262
Submission received: 29 May 2024 / Revised: 19 June 2024 / Accepted: 26 June 2024 / Published: 30 June 2024
(This article belongs to the Special Issue Sensors and Machine-Learning Based Signal Processing)

Abstract

Detecting bearing defects accurately and efficiently is critical for industrial safety and efficiency. This paper introduces Bearing-DETR, a deep learning model optimised using the Real-Time Detection Transformer (RT-DETR) architecture. Enhanced with Dysample Dynamic Upsampling, Efficient Model Optimization (EMO) with Meta-Mobile Blocks (MMB), and Deformable Large Kernel Attention (D-LKA), Bearing-DETR offers significant improvements in defect detection while maintaining a lightweight framework suitable for low-resource devices. Validated on a dataset from a chemical plant, Bearing-DETR outperformed the standard RT-DETR, achieving a mean average precision (mAP) of 94.3% at IoU = 0.5 and 57.5% at IoU = 0.5–0.95. It also reduced floating-point operations (FLOPs) to 8.2 G and parameters to 3.2 M, underscoring its enhanced efficiency and reduced computational demands. These results demonstrate the potential of Bearing-DETR to transform maintenance strategies and quality control across manufacturing environments, emphasising adaptability and impact on sustainability and operational costs.
Keywords: bearing defect detection; real-time detection transformer (RT-DETR); lightweight model; industrial efficiency bearing defect detection; real-time detection transformer (RT-DETR); lightweight model; industrial efficiency

Share and Cite

MDPI and ACS Style

Liu, M.; Wang, H.; Du, L.; Ji, F.; Zhang, M. Bearing-DETR: A Lightweight Deep Learning Model for Bearing Defect Detection Based on RT-DETR. Sensors 2024, 24, 4262. https://doi.org/10.3390/s24134262

AMA Style

Liu M, Wang H, Du L, Ji F, Zhang M. Bearing-DETR: A Lightweight Deep Learning Model for Bearing Defect Detection Based on RT-DETR. Sensors. 2024; 24(13):4262. https://doi.org/10.3390/s24134262

Chicago/Turabian Style

Liu, Minggao, Haifeng Wang, Luyao Du, Fangsong Ji, and Ming Zhang. 2024. "Bearing-DETR: A Lightweight Deep Learning Model for Bearing Defect Detection Based on RT-DETR" Sensors 24, no. 13: 4262. https://doi.org/10.3390/s24134262

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop