Epistemic Uncertainty-aware Recommendation Systems via Bayesian Deep Ensemble Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cheraghi, Radin, Mahfoozi, Amir Mohammad, Zolfaghari, Sepehr, Shabani, Mohammadshayan, Ramezani, Maryam, Rabiee, Hamid R.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910911751520256
author Cheraghi, Radin
Mahfoozi, Amir Mohammad
Zolfaghari, Sepehr
Shabani, Mohammadshayan
Ramezani, Maryam
Rabiee, Hamid R.
author_facet Cheraghi, Radin
Mahfoozi, Amir Mohammad
Zolfaghari, Sepehr
Shabani, Mohammadshayan
Ramezani, Maryam
Rabiee, Hamid R.
contents Recommending items to users has long been a fundamental task, and studies have tried to improve it ever since. Most well-known models commonly employ representation learning to map users and items into a unified embedding space for matching assessment. These approaches have primary limitations, especially when dealing with explicit feedback and sparse data contexts. Two primary limitations are their proneness to overfitting and failure to incorporate epistemic uncertainty in predictions. To address these problems, we propose a novel Bayesian Deep Ensemble Collaborative Filtering method named BDECF. To improve model generalization and quality, we utilize Bayesian Neural Networks, which incorporate uncertainty within their weight parameters. In addition, we introduce a new interpretable non-linear matching approach for the user and item embeddings, leveraging the advantages of the attention mechanism. Furthermore, we endorse the implementation of an ensemble-based supermodel to generate more robust and reliable predictions, resulting in a more complete model. Empirical evaluation through extensive experiments and ablation studies across a range of publicly accessible real-world datasets with differing sparsity characteristics confirms our proposed method's effectiveness and the importance of its components.
format Preprint
id arxiv_https___arxiv_org_abs_2504_10753
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Epistemic Uncertainty-aware Recommendation Systems via Bayesian Deep Ensemble Learning
Cheraghi, Radin
Mahfoozi, Amir Mohammad
Zolfaghari, Sepehr
Shabani, Mohammadshayan
Ramezani, Maryam
Rabiee, Hamid R.
Information Retrieval
Artificial Intelligence
Machine Learning
Recommending items to users has long been a fundamental task, and studies have tried to improve it ever since. Most well-known models commonly employ representation learning to map users and items into a unified embedding space for matching assessment. These approaches have primary limitations, especially when dealing with explicit feedback and sparse data contexts. Two primary limitations are their proneness to overfitting and failure to incorporate epistemic uncertainty in predictions. To address these problems, we propose a novel Bayesian Deep Ensemble Collaborative Filtering method named BDECF. To improve model generalization and quality, we utilize Bayesian Neural Networks, which incorporate uncertainty within their weight parameters. In addition, we introduce a new interpretable non-linear matching approach for the user and item embeddings, leveraging the advantages of the attention mechanism. Furthermore, we endorse the implementation of an ensemble-based supermodel to generate more robust and reliable predictions, resulting in a more complete model. Empirical evaluation through extensive experiments and ablation studies across a range of publicly accessible real-world datasets with differing sparsity characteristics confirms our proposed method's effectiveness and the importance of its components.
title Epistemic Uncertainty-aware Recommendation Systems via Bayesian Deep Ensemble Learning
topic Information Retrieval
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2504.10753