Joint embedding-classifier learning for interpretable collaborative filtering

doi:10.1186/s12859-024-06026-8

Joint embedding-classifier learning for interpretable collaborative filtering

2025-01-22

0

OA

AI

DOI：10.1186/s12859-024-06026-8

Original

Original request for help

Save

Abstract

En 中文

BackgroundInterpretability is a topical question in recommender systems, especially in healthcare applications. An interpretable classifier quantifies the importance of each input feature for the predicted item-user association in a non-ambiguous fashion.ResultsWe introduce the novel Joint Embedding Learning-classifier for improved Interpretability (JELI). By combining the training of a structured collaborative-filtering classifier and an embedding learning task, JELI predicts new user-item associations based on jointly learned item and user embeddings while providing feature-wise importance scores. Therefore, JELI flexibly allows the introduction of priors on the connections between users, items, and features. In particular, JELI simultaneously (a) learns feature, item, and user embeddings; (b) predicts new item-user associations; (c) provides importance scores for each feature. Moreover, JELI instantiates a generic approach to training recommender systems by encoding generic graph-regularization constraints.ConclusionsFirst, we show that the joint training approach yields a gain in the predictive power of the downstream classifier. Second, JELI can recover feature-association dependencies. Finally, JELI induces a restriction in the number of parameters compared to baselines in synthetic and drug-repurposing data sets.

Keywords:

Drug repurposing

Interpretability

Gene expression

Collaborative filtering

Joint embedding-classifier learning for interpretable collaborative filtering

Abstract

Journal

Organization