English
Related papers

Related papers: Budgeted Embedding Table For Recommender Systems

200 papers

Recommender systems are one of the most successful applications of machine learning and data science. They are successful in a wide variety of application domains, including e-commerce, media streaming content, email marketing, and…

Information Retrieval · Computer Science 2023-04-04 Juan Pablo Equihua , Maged Ali , Henrik Nordmark , Berthold Lausen

Embedding-based neural retrieval (EBR) is an effective search retrieval method in product search for tackling the vocabulary gap between customer search queries and products. The initial launch of our EBR system at Walmart yielded…

Embeddings, low-dimensional vector representation of objects, are fundamental in building modern machine learning systems. In industrial settings, there is usually an embedding team that trains an embedding model to solve intended tasks…

Machine Learning · Statistics 2022-06-08 Weihua Hu , Rajas Bansal , Kaidi Cao , Nikhil Rao , Karthik Subbian , Jure Leskovec

Recommender systems often struggle with data sparsity and cold-start scenarios, limiting their ability to provide accurate suggestions for new or infrequent users. This paper presents a Graph Attention Network (GAT) based Collaborative…

Information Retrieval · Computer Science 2025-10-31 Danial Ebrat , Sepideh Ahmadian , Luis Rueda

Although exact term match between queries and documents is the dominant method to perform first-stage retrieval, we propose a different approach, called RepBERT, to represent documents and queries with fixed-length contextualized…

Information Retrieval · Computer Science 2020-07-21 Jingtao Zhan , Jiaxin Mao , Yiqun Liu , Min Zhang , Shaoping Ma

A transaction-based recommender system (TBRS) aims to predict the next item by modeling dependencies in transactional data. Generally, two kinds of dependencies considered are intra-transaction dependency and inter-transaction dependency.…

Information Retrieval · Computer Science 2020-06-09 Shoujin Wang , Longbing Cao , Liang Hu , Shlomo Berkovsky , Xiaoshui Huang , Lin Xiao , Wenpeng Lu

Representation learning-based recommendation models play a dominant role among recommendation techniques. However, most of the existing methods assume both historical interactions and embedding dimensions are independent of each other, and…

Information Retrieval · Computer Science 2023-03-14 Zhuoyi Lin , Lei Feng , Xingzhi Guo , Yu Zhang , Rui Yin , Chee Keong Kwoh , Chi Xu

Retrieval, the initial stage of a recommendation system, is tasked with down-selecting items from a pool of tens of millions of candidates to a few thousands. Embedding Based Retrieval (EBR) has been a typical choice for this problem,…

With the increasing demands on e-commerce platforms, numerous user action history is emerging. Those enriched action records are vital to understand users' interests and intents. Recently, prior works for user behavior prediction mainly…

Information Retrieval · Computer Science 2022-02-15 Ruijie Wang , Zheng Li , Danqing Zhang , Qingyu Yin , Tong Zhao , Bing Yin , Tarek Abdelzaher

Sequential recommendation aims to predict users' future interactions by modeling collaborative filtering (CF) signals from historical behaviors of similar users or items. Traditional sequential recommenders predominantly rely on ID-based…

Information Retrieval · Computer Science 2025-06-30 Yingzhi He , Xiaohao Liu , An Zhang , Yunshan Ma , Tat-Seng Chua

Current recommender systems employ large-sized embedding tables with uniform dimensions for all features, leading to overfitting, high computational cost, and suboptimal generalizing performance. Many techniques aim to solve this issue by…

Information Retrieval · Computer Science 2023-09-18 Yakun Yu , Shi-ang Qi , Jiuding Yang , Liyao Jiang , Di Niu

This paper proposes a new neural architecture for collaborative ranking with implicit feedback. Our model, LRML (\textit{Latent Relational Metric Learning}) is a novel metric learning approach for recommendation. More specifically, instead…

Artificial Intelligence · Computer Science 2018-02-14 Yi Tay , Anh Tuan Luu , Siu Cheung Hui

The review-based recommender systems are commonly utilized to measure users preferences towards different items. In this paper, we focus on addressing three main problems existing in the review-based methods. Firstly, these methods suffer…

Information Retrieval · Computer Science 2020-12-14 Yuexin Wu , Tianyu Gao , Sihao Wang , Zhongmin Xiong

Deep learning for recommendation data is one of the most pervasive and challenging AI workload in recent times. State-of-the-art recommendation models are one of the largest models matching the likes of GPT-3 and Switch Transformer.…

Information Retrieval · Computer Science 2022-01-25 Aditya Desai , Li Chou , Anshumali Shrivastava

As users often express their preferences with binary behavior data~(implicit feedback), such as clicking items or buying products, implicit feedback based Collaborative Filtering~(CF) models predict the top ranked items a user might like by…

Information Retrieval · Computer Science 2021-05-27 Lei Chen , Le Wu , Kun Zhang , Richang Hong , Meng Wang

Although multi-interest recommenders have achieved significant progress in the matching stage, our research reveals that existing models tend to exhibit an under-clustered item embedding space, which leads to a low discernibility between…

Information Retrieval · Computer Science 2023-11-30 Yaokun Liu , Xiaowang Zhang , Minghui Zou , Zhiyong Feng

Embedding-based neural retrieval is a prevalent approach to address the semantic gap problem which often arises in product search on tail queries. In contrast, popular queries typically lack context and have a broad intent where additional…

Information Retrieval · Computer Science 2024-09-26 Rishikesh Jha , Siddharth Subramaniyam , Ethan Benjamin , Thrivikrama Taula

We introduce the payload optimization method for federated recommender systems (FRS). In federated learning (FL), the global model payload that is moved between the server and users depends on the number of items to recommend. The model…

Machine Learning · Computer Science 2021-07-29 Farwa K. Khan , Adrian Flanagan , Kuan E. Tan , Zareen Alamgir , Muhammad Ammad-Ud-Din

We present an information-theoretic framework to learn fixed-dimensional embeddings for tasks in reinforcement learning. We leverage the idea that two tasks are similar if observing an agent's performance on one task reduces our uncertainty…

Machine Learning · Computer Science 2024-05-10 Mridul Mahajan , Georgios Tzannetos , Goran Radanovic , Adish Singla

Interactive recommender systems can dynamically adapt to user feedback, but often suffer from content homogeneity and filter bubble effects due to overfitting short-term user preferences. While recent efforts aim to improve content…

Information Retrieval · Computer Science 2026-05-12 Chongjun Xia , Yanchun Peng , Xianzhi Wang