Learning a Representation for Cover Song Identification Using Convolutional Neural Network

TL;DR

Proposes a CNN-based approach with multi-scale and dilated convolutions, achieving state-of-the-art cover song identification with high robustness and efficiency.

cs.MM 🔴 Advanced 2019-11-01 37 views
Zhesong Yu Xiaoshuo Xu Xiaoou Chen Deshun Yang
Music IR Deep Learning CNN Cover Song Recognition Data Augmentation

Key Findings

Methodology

This paper introduces a specialized CNN architecture for cover song identification, integrating multi-scale convolution kernels (like sizes 12 and 13) and dilated convolutions to enlarge receptive fields. The network avoids frequency dimension downsampling to preserve key invariance, employing global pooling for variable-length input handling. Training uses a classification approach, grouping different versions of the same song, with data augmentation simulating tempo changes (speed variations from 0.7 to 1.3). The feature vectors extracted are compared via cosine similarity for retrieval. Experiments on datasets such as SHS100K, Covers80, and Mazurkas demonstrate superior performance over existing methods, with MAP improvements exceeding 40%. The model achieves rapid inference (~3.68 ms per query), suitable for large-scale real-world applications.

Key Results

  • On SHS100K, MAP reaches 0.655, outperforming the previous best of 0.465 by 41%. On Covers80, MAP is 0.840, surpassing 0.744. Mazurkas results show MAP at 0.933, better than DTW and NCD. Query time is approximately 3.68 ms, confirming real-time capability. Ablation studies show that maintaining high resolution in frequency domain enhances key-invariance recognition, especially for transpositions of one or two semitones.
  • Incorporating dilated convolutions enlarges the receptive field, capturing long-range melodic features crucial for transposition invariance. Multi-scale kernels improve adaptability to diverse frequency patterns. Data augmentation with tempo variation enhances robustness against rhythm changes, validated by ablation experiments. The architecture's avoidance of frequency downsampling preserves detailed spectral information, leading to higher accuracy.
  • Results across multiple datasets demonstrate the model's strong generalization. Ablation confirms the importance of high-frequency resolution. The approach balances efficiency and accuracy, making it feasible for large-scale music retrieval systems. Its simplicity and speed outperform traditional alignment methods like DTW, enabling real-time applications.

Significance

This work advances music content recognition by shifting from handcrafted features and alignment algorithms to an end-to-end deep learning framework. It effectively addresses long-standing challenges such as key transposition and tempo variation, providing a scalable, accurate, and fast solution. The approach's robustness and efficiency make it highly suitable for commercial deployment in copyright management, music recommendation, and large-scale music database search, marking a significant step toward intelligent music understanding.

Technical Contribution

The paper introduces a novel CNN architecture combining multi-scale convolution kernels and dilated convolutions, avoiding frequency dimension downsampling to maintain key invariance. It innovatively applies global pooling for variable-length inputs and employs data augmentation to simulate tempo changes, significantly improving robustness. The model’s design balances complexity and speed, enabling high-precision retrieval in large datasets. Extensive experiments validate its superiority over existing methods, establishing new benchmarks.

Novelty

This is the first work to incorporate specific kernel sizes (12 and 13) aligned with musical semitones for long-range melodic feature extraction, combined with dilated convolutions to expand receptive fields. Unlike prior approaches relying on handcrafted features or alignment algorithms, this end-to-end CNN learns key-invariant representations directly from spectral data. The data augmentation strategy further enhances robustness against tempo variations, representing a comprehensive innovation in music retrieval.

Limitations

  • The model's robustness diminishes under extreme pitch shifts or complex rhythmic transformations beyond the simulated tempo variations. It requires large annotated datasets for training, limiting applicability in low-resource scenarios. Although fast, the model still faces challenges in real-time deployment on very large music collections without further optimization. Future work should explore unsupervised learning and domain adaptation to address these issues.

Future Work

Future directions include integrating attention mechanisms to focus on salient melodic segments, extending the model to handle more complex musical transformations, and exploring unsupervised or semi-supervised training to reduce dependency on labeled data. Additionally, optimizing model architecture for deployment on resource-constrained devices and expanding to cross-modal retrieval (e.g., lyrics, video) are promising avenues.

AI Executive Summary

Cover song identification has long been a challenging task in music information retrieval, hindered by complex variations such as key transposition, tempo changes, and structural differences. Traditional methods relied heavily on handcrafted features and alignment algorithms like dynamic programming, which, while effective, are computationally intensive and less scalable for large datasets. Recent advances in deep learning, particularly convolutional neural networks (CNNs), have opened new avenues for end-to-end feature learning, promising higher accuracy and efficiency.

This paper introduces a novel CNN architecture tailored for cover song recognition. The model employs multi-scale convolution kernels—specifically sizes 12 and 13—to align with musical semitones, capturing long-range melodic features crucial for transposition invariance. To further enhance the model’s ability to handle rhythm and tempo variations, dilated convolutions are integrated, enlarging the receptive field without losing spectral resolution. Importantly, the architecture avoids downsampling in the frequency domain, preserving spectral details essential for accurate key recognition.

Training involves a classification approach where different versions of the same song are grouped into one class, with data augmentation simulating tempo changes by varying playback speed from 0.7 to 1.3. This strategy significantly improves the model’s robustness against tempo variations. The extracted features are fixed-length vectors, obtained via global pooling, enabling fast similarity computation using cosine metrics.

Experimental results on datasets such as SHS100K, Covers80, and Mazurkas demonstrate that the proposed method outperforms existing state-of-the-art approaches, with MAP improvements exceeding 40%. The inference speed is remarkably fast, around 3.68 milliseconds per query, making it suitable for real-time large-scale music retrieval. These advances mark a significant step forward in automating and scaling cover song recognition, with broad implications for copyright enforcement, music recommendation, and digital music library management.

Despite its strengths, the approach faces limitations in handling extreme pitch shifts and rhythmic complexities beyond the simulated tempo changes. Future work aims to incorporate attention mechanisms, unsupervised learning, and cross-modal features to further enhance robustness and applicability. Overall, this research sets a new benchmark for efficient, accurate, and scalable cover song identification, paving the way for smarter music content understanding systems.

Deep Dive

Abstract

Cover song identification represents a challenging task in the field of Music Information Retrieval (MIR) due to complex musical variations between query tracks and cover versions. Previous works typically utilize hand-crafted features and alignment algorithms for the task. More recently, further breakthroughs are achieved employing neural network approaches. In this paper, we propose a novel Convolutional Neural Network (CNN) architecture based on the characteristics of the cover song task. We first train the network through classification strategies; the network is then used to extract music representation for cover song identification. A scheme is designed to train robust models against tempo changes. Experimental results show that our approach outperforms state-of-the-art methods on all public datasets, improving the performance especially on the large dataset.

cs.MM cs.LG cs.SD eess.AS