🌙

一种通过知识蒸馏的混合模型压缩方法用于预测增材制造中的能耗

A hybrid model compression approach via knowledge distillation for predicting energy consumption in additive manufacturing

International Journal of Production Research · 2023
被引 12
ABS 3

中文导读

提出一种混合模型压缩方法,通过知识蒸馏和教师助理架构降低增材制造能耗预测模型的复杂度,在减少计算成本的同时保持性能,实验显示该方法均方根误差、平均绝对误差和训练时间均最低。

Abstract

Recently, additive manufacturing (AM) has received increased attention due to its high energy consumption. By extracting hidden information or highly representative features from energy-relevant data, knowledge distillation (KD) reduces predictive model complexity and computational load. By using almost predetermined and fixed models, the distillation process restricts students and teachers from learning from one model to another. To reduce computational costs while maintaining acceptable performance, a teacher assistant (TA) was added to the teacher-student architecture. Firstly, a teacher ensemble was combined with three baseline models to enhance accuracy. In the second step, a teacher ensemble (TA) was formed to bridge the capacity gap between the ensemble and the simplified model. As a result, the complexity of the student model was reduced. Using geometry-based features derived from layer-wise image data, a KD-based predictive model was developed to evaluate the feasibility and effectiveness of two independently trained student models. In comparison with independently trained student models, the performance of the proposed method has the lowest RMSE, MAE, and training time.

增材制造能耗预测知识蒸馏模型压缩机器学习