Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

On Metric Learning for Audio-Text Cross-Modal Retrieval

About

Audio-text retrieval aims at retrieving a target audio clip or caption from a pool of candidates given a query in another modality. Solving such cross-modal retrieval task is challenging because it not only requires learning robust feature representations for both modalities, but also requires capturing the fine-grained alignment between these two modalities. Existing cross-modal retrieval models are mostly optimized by metric learning objectives as both of them attempt to map data to an embedding space, where similar data are close together and dissimilar data are far apart. Unlike other cross-modal retrieval tasks such as image-text and video-text retrievals, audio-text retrieval is still an unexplored task. In this work, we aim to study the impact of different metric learning objectives on the audio-text retrieval task. We present an extensive evaluation of popular metric learning objectives on the AudioCaps and Clotho datasets. We demonstrate that NT-Xent loss adapted from self-supervised learning shows stable performance across different datasets and training settings, and outperforms the popular triplet-based losses. Our code is available at https://github.com/XinhaoMei/audio-text_retrieval.

Xinhao Mei, Xubo Liu, Jianyuan Sun, Mark D. Plumbley, Wenwu Wang• 2022

Related benchmarks

TaskDatasetResultRank
Text-to-Audio RetrievalAudioCaps (test)
Recall@139.4
145
Audio CaptioningAudioCaps (test)
CIDEr63.5
140
Audio-to-Text RetrievalClotho (test)
R@112.8
78
Text-to-Audio RetrievalClotho (test)
R@110.1
62
Cross-modal retrievalClotho (test)
R@116.2
29
Cross-modal retrievalAudioCaps (test)
R@137.9
23
Text-to-Audio RetrievalClotho 1K 1.0 (test)
R@114.4
10
Text-to-Audio RetrievalAudioCaps 1K 1.0 (test)
Recall@133.9
10
Audio-to-Text RetrievalClotho 1K 1.0 (test)
R@116.2
8
Audio-to-Text RetrievalAudioCaps 1K 1.0 (test)
R@139.4
8
Showing 10 of 10 rows

Other info

Follow for update