International Business Machines Corporation
MOBILE AI

Abstract:

A machine learning model can be optimized for deployment on a device based on hardware specifications of the device. An existing model is acquired and pruned to reduce hardware resource consumption of the model. The pruned model is then trained based on training data. The pruned model is also trained based on a collection of "teacher" models. Performance of the trained model is then evaluated and compared to performance requirements, which can be based on the hardware specifications of a device.