Online Knowledge Distillation for Machine Health Prognosis Considering Edge Deployment

Publisher:
Institute of Electrical and Electronics Engineers (IEEE)
Publication Type:
Journal Article
Citation:
IEEE Internet of Things Journal, 2024, PP, (99), pp. 1-1
Issue Date:
2024-01-01
Filename Description Size
1730949.pdfPublished version4.74 MB
Adobe PDF
Full metadata record
Complex neural networks with deep structures are beneficial for solving problems such as fault classification and health prediction of industrial equipment due to their powerful feature extraction capabilities. Unfortunately, corresponding complex models designed based on deep learning algorithms require huge computational and memory resources, making them difficult to achieve effective edge deployment. In order to solve this difficulty with practical industrial significance, this paper proposes an online knowledge distillation framework for machine health prognosis. Within this framework, the learned knowledge of complex networks can be distilled to simple networks that can be deployed on edge devices in sites. Specifically, the response-based knowledge distillation module, feature-based knowledge distillation module, and relation-based knowledge distillation module are respectively designed to achieve effective information transmission from different levels. Furthermore, the inherent differences between simple and complex networks have been fully considered for their impact on the efficiency of knowledge distillation, and an adaptive mutual learning strategy has been contrapuntally proposed to address this limitation. Multiple online knowledge distillation experiments were conducted on two different sets of run-to-failure datasets of mechanical key components with different pairs of complex and simple networks to verify the effectiveness of the proposed framework. The experimental results show that the simple student-networks can effectively improve prediction performance after receiving knowledge distillation from the complex teacher-networks, providing a new solution for machine health prognosis under the premise of edge deployment.
Please use this identifier to cite or link to this item: