Improved rank discrepancy-aware sampling based on bidirectional distillation for top-K recommender system

Article type: Research Article

Authors: Du, Weidong^{; *}

Affiliations: School of Mechnical Engineering, SouthEast University, Nanjing, PR, China

Correspondence: [*] Corresponding author. Weidong Du, School of Mechnical Engineering, SouthEast University, Nanjing, PR, China. E-mail: [email protected].

Abstract: Nowadays, the model compression method of knowledge distillation has drawn great attentions in Recommender systems (RS). The strategy of bidirectional distillation performs the bidirectional learning for both the teacher and the student models such that these two models can collaboratively improve with each other. However, this strategy cannot effectively exploit representation capabilities of each item and lack of the interpretability for the importance of items. Thus, how to develop an effective sampling scheme is still valuable for us to further study and explore. In this paper, we propose an improved rank discrepancy-aware item sampling strategy to enhance the performance of bidirectional distillation learning. Specifically, by employing the distillation loss, we train the teacher and student models to reflect the fact that a user has partiality for the unobserved items. Then, we propose the improved rank discrepancy-aware sampling strategy based on feedback learning mechanism to transfer just the useful information which can effectively enhance each other. The key part of the multiple distillation training aims to select valuable items which can be re-distilled in the network for training. The proposed technique can effectively solve the problem of high ambiguity in nature for recommender system. Experimental results on several real-world recommender system datasets well demonstrate that the improved bidirectional distillation strategy shows better performance.

Keywords: Bidirectional distillation, student-teacher learning, rank discrepancy aware items selection, recommender system

DOI: 10.3233/JIFS-222063

Journal: Journal of Intelligent & Fuzzy Systems, vol. 44, no. 3, pp. 4197-4206, 2023