Goto

Collaborating Authors

 Khoeini, Arash


Improving OOD Generalization of Pre-trained Encoders via Aligned Embedding-Space Ensembles

arXiv.org Artificial Intelligence

The quality of self-supervised pre-trained embeddings on out-of-distribution (OOD) data is poor without fine-tuning. A straightforward and simple approach to improving the generalization of pre-trained representation to OOD data is the use of deep ensembles. However, obtaining an effective ensemble in the embedding space with only unlabeled data remains an unsolved problem. We first perform a theoretical analysis that reveals the relationship between individual hyperspherical embedding spaces in an ensemble. We then design a principled method to align these embedding spaces in an unsupervised manner. Experimental results on the MNIST dataset show that our embedding-space ensemble method improves pre-trained embedding quality on in-distribution and OOD data compared to single encoders.


GEMRank: Global Entity Embedding For Collaborative Filtering

arXiv.org Machine Learning

Abstract--Recently, word embedding algorithms have been applied to map the entities of recommender systems, such as users and items, to new feature spaces using textual elementcontext relations among them. Unlike many other domains, this approach has not achieved a desired performance in collaborative filtering problems, probably due to unavailability of appropriate textual data. In this paper we propose a new recommendation framework, called GEMRank that can be applied when the useritem matrix is the sole available souce of information. It uses the concept of profile co-occurrence for defining relations among entities and applies a factorization method for embedding the users and items. GEMRank then feeds the extracted representations to a neural network model to predict user-item like/dislike relations which the final recommendations are made based on. We evaluated GEMRank in an extensive set of experiments against state of the art recommendation methods. The results show that GEMRank significantly outperforms the baseline algorithms in a variety of data sets with different degrees of density. Recommendation Systems help users to find relevant items based on their preferences. Many prominent recommendation systems are using Collaborative Filtering (CF) for making recommendations ( [1]).