Knowledge distillation for multi-depth-model-fusion recommendation algorithm

PLoS One. 2022 Oct 25;17(10):e0275955. doi: 10.1371/journal.pone.0275955. eCollection 2022.

Abstract

Recommendation algorithms save a lot of valuable time for people to get the information they are interested in. However, the feature calculation and extraction process of each machine learning or deep learning recommendation algorithm are different, so how to obtain various features with different dimensions, i.e., how to integrate the advantages of each model and improve the model inference efficiency, becomes the focus of this paper. In this paper, a better deep learning model is obtained by integrating several cutting-edge deep learning models. Meanwhile, to make the integrated learning model converge better and faster, the parameters of the integrated module are initialized, constraints are imposed, and a new activation function is designed for better integration of the sub-models. Finally, the integrated large model is distilled for knowledge distillation, which greatly reduces the number of model parameters and improves the model inference efficiency.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Humans
  • Machine Learning*

Grants and funding

This paper was supported by the National Key Research and Development Program of China under Grant 2020YFB1713300. In addtion, the recipient is Proffsor Shaobo Li. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.