EAAI Journal 2026 Journal Article
A prediction method for micro-motor rotor unbalance based on the InceptionV3- Convolutional block attention module model
- Haoyan Zhang
- Yang An
- Jiarong Fan
- Huimin Wang
- Dongxia Zheng
- Shuai Wang
- Guoqiang Wang
Rotor unbalance is a significant cause of motor failure in machinery, making accurate estimation of unbalance values essential. Due to their lightweight and compact design, micro-motors pose greater challenges in precise unbalance measurement. To address this issue, a rotor unbalance prediction model for micro-motors is proposed, which integrates data fusion and an attention mechanism. First, two sets of vibration signals are converted into images using the Gramian Angular Field (GAF) method and fused to construct the unbalance dataset. Data augmentation is then employed to enhance the model's generalization ability. Subsequently, an unbalance prediction model is developed based on the InceptionV3 network with a Convolutional Block Attention Module (CBAM), where the attention mechanism enhances feature extraction and transfer learning improves training efficiency and prediction accuracy. Finally, the improved model is combined with a probabilistic mapping approach to estimate the unbalance mass. Experimental results show that the proposed method achieves a prediction accuracy of 94% for both unbalance magnitude and phase. This approach not only improves the accuracy of rotor unbalance estimation in micro-motors but also provides a reference for unbalance detection in other types of machinery.