HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Transformers with multi-modal features and post-fusion context for e-commerce session-based recommendation

Gabriel de Souza P. Moreira Sara Rabhi Ronay Ak Md Yasin Kabir Even Oldridge

Transformers with multi-modal features and post-fusion context for e-commerce session-based recommendation

Abstract

Session-based recommendation is an important task for e-commerce services, where a large number of users browse anonymously or may have very distinct interests for different sessions. In this paper we present one of the winning solutions for the Recommendation task of the SIGIR 2021 Workshop on E-commerce Data Challenge. Our solution was inspired by NLP techniques and consists of an ensemble of two Transformer architectures - Transformer-XL and XLNet - trained with autoregressive and autoencoding approaches. To leverage most of the rich dataset made available for the competition, we describe how we prepared multi-model features by combining tabular events with textual and image vectors. We also present a model prediction analysis to better understand the effectiveness of our architectures for the session-based recommendation.

Benchmarks

BenchmarkMethodologyMetrics
product-recommendation-on-coveo-dataEnsemble (60 models)
F1: 0.0748
MRR: 0.2784

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Transformers with multi-modal features and post-fusion context for e-commerce session-based recommendation | Papers | HyperAI