Published June 14, 2023 | Version 1.0

Enhancing MovieLens Dataset: Enriching Recommendations with Audio Information, Transcriptions, and Metadata

  • 1. Valencian Research Institute for Artificial Intelligence & Tyris Tech S.L
  • 2. Valencian Research Institute for Artificial Intelligence
  • 3. GREENIUS Research Group & Tyris Tech S.L

Description

Nowadays, there are lots of datasets available for training and experimentation in the field of recommender systems. Specifically, in the recommendation of audiovisual content, the MovieLens dataset is a prominent example. It is focused on the user-item relationship, providing actual interaction data between users and movies. However, although movies can be described with several characteristics, this dataset only offers limited information about the movie genres. 

In this work, we propose enriching the MovieLens dataset by incorporating metadata available on the web (such as cast, description, keywords, etc.) and movie trailers. By leveraging the trailers, we extract audio information and generate transcriptions for each trailer, introducing a crucial textual dimension to the dataset. The audio information was extracted by the waveform and frequency analysis, followed by the application of dimensionality reduction techniques. For the transcription generation, the deep learning model Whisper was used. Finally, metadata was obtained from TMDB, and the BERT model was applied to extract embeddings.

These additional attributes enrich the original dataset, providing deeper and more precise analysis. Then, the use of this extended and enhanced dataset could drive significant advancements in recommendation systems, enhancing user experiences by providing more relevant and tailored movie recommendations based on their tastes and preferences. 

Files

EMD.zip

Files (50.8 GB)

Name Size
md5:60e6a1b685b175a609d58b244d75f277
50.8 GB Preview Download
md5:555f0100e5000ca9cfc6dde4a839a9b9
5.8 kB Preview Download