4.4k citations · 4.6k across the 16 of their papers we have counts for
7 papers · 1 filter
Private Learning with Public Features
Walid Krichene, Nicolas Mayoraz, Steffen Rendle +3
We study a class of private learning problems in which the data is a join of private and public features. This is often the case in private personalization tasks such as recommenda…
ALX: Large Scale Matrix Factorization on TPUs
Harsh Mehta, Steffen Rendle, Walid Krichene +1
We present ALX, an open-source library for distributed matrix factorization using Alternating Least Squares, written in JAX. Our design allows for efficient use of the TPU architec…
iALS++: Speeding up Matrix Factorization with Subspace Optimization
Steffen Rendle, Walid Krichene, Li Zhang +1
iALS is a popular algorithm for learning matrix factorization models from implicit feedback with alternating least squares. This algorithm was invented over a decade ago but still…
Private Alternating Least Squares: Practical Private Matrix Completion with Tighter Rates
Steve Chien, Prateek Jain, Walid Krichene +4
We study the problem of differentially private (DP) matrix completion under user-level privacy. We design a joint differentially private variant of the popular Alternating-Least-Sq…
Zero-Shot Heterogeneous Transfer Learning from Recommender Systems to Cold-Start Search Retrieval
Tao Wu, Ellie Ka-In Chio, Heng-Tze Cheng +15
Many recent advances in neural information retrieval models, which predict top-K items given a query, learn directly from a large training set of (query, item) pairs. However, they…
Superbloom: Bloom filter meets Transformer
John Anderson, Qingqing Huang, Walid Krichene +2
We extend the idea of word pieces in natural language models to machine learning tasks on opaque ids. This is achieved by applying hash functions to map each id to multiple hash to…