HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Evaluation of CNN-based Automatic Music Tagging Models

Minz Won; Andres Ferraro; Dmitry Bogdanov; Xavier Serra

Evaluation of CNN-based Automatic Music Tagging Models

Abstract

Recent advances in deep learning accelerated the development of content-based automatic music tagging systems. Music information retrieval (MIR) researchers proposed various architecture designs, mainly based on convolutional neural networks (CNNs), that achieve state-of-the-art results in this multi-label binary classification task. However, due to the differences in experimental setups followed by researchers, such as using different dataset splits and software versions for evaluation, it is difficult to compare the proposed architectures directly with each other. To facilitate further research, in this paper we conduct a consistent evaluation of different music tagging models on three datasets (MagnaTagATune, Million Song Dataset, and MTG-Jamendo) and provide reference results using common evaluation metrics (ROC-AUC and PR-AUC). Furthermore, all the models are evaluated with perturbed inputs to investigate the generalization capabilities concerning time stretch, pitch shift, dynamic range compression, and addition of white noise. For reproducibility, we provide the PyTorch implementations with the pre-trained models.

Code Repositories

HephaestusProject/pytorch-FCN
pytorch
Mentioned in GitHub
cgaroufis/msspt
tf
Mentioned in GitHub
minzwon/tag-based-music-retrieval
pytorch
Mentioned in GitHub
minzwon/sota-music-tagging-models
Official
pytorch
Mentioned in GitHub
Dohppak/Music_DeepEmbedding_Extractor
pytorch
Mentioned in GitHub
pxaris/ccml
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
music-auto-tagging-on-magnatagatune-cleanShort-chunk CNN + Res
PR-AUC: 46.14
ROC-AUC: 91.29

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
Evaluation of CNN-based Automatic Music Tagging Models | Papers | HyperAI