duduoliver / wavcaps

This reporsitory contains metadata of WavCaps dataset and codes for downstream tasks.

🌟 WavCaps can be downloaded through HuggingFace now!

WavCaps

WavCaps: A ChatGPT-Assisted Weakly-Labelled Audio Captioning Dataset for Audio-Language Multimodal Research. (arXiv)

This repository contains:

• Metadata of WavCaps dataset.
• Source code for related tasks: audio-language retrieval, automated audio captioning, and zero-shot audio classification.

Table of Contents

• WavCaps
  • Table of Contents
  • Introduction
  • Dataset
  • License
  • Code for related tasks
  • Citation

Introduction

The advancement of audio-language (AL) multimodal learning tasks has been significant in recent years, yet the limited size of existing audio-language datasets poses challenges for researchers due to the costly and time-consuming collection process. To address this data scarcity issue, we introduce WavCaps, the first large-scale weakly-labelled audio captioning dataset, comprising approximately 400k audio clips with paired captions. We sourced audio clips and their raw descriptions from web sources and a sound event detection dataset. However, the online-harvested raw descriptions are highly noisy and unsuitable for direct use in tasks such as automated audio captioning. To overcome this issue, we propose a three-stage processing pipeline for filtering noisy data and generating high-quality captions, where ChatGPT, a large language model, is leveraged to filter and transform raw descriptions automatically.

We conduct a comprehensive analysis of the characteristics of WavCaps dataset and evaluate it on multiple downstream audio-language multimodal learning tasks. The systems trained on WavCaps outperform previous state-of-the-art (SOTA) models by a significant margin. Our aspiration is for the WavCaps dataset we have proposed to facilitate research in audio-language multimodal learning and demonstrate the potential of utilizing ChatGPT to enhance academic research.

Dataset

WavCaps are sourced from three websites and a sound event detection dataset:

• FreeSound (262300)
• BBC Sound Effects (31201)
• SoundBible (1232)
• AudioSet Strongly-labelled Subset (108317)

ChatGPT is leveraged to process and transform raw-descriptions into captions.

We release WavCaps dataset on HuggingFace. Each data source is released with a json file and corresponding wavefroms.

License

Only academic uses are allowed for WavCaps dataset. By downloading audio clips through the links provided in the json files, you agree that you will use the audios for research purposes only. For credits for audio clips from FreeSound, please refer to its own page.

For detailed license information, please refer to: FreeSound, BBC Sound Effects, SoundBible

The models we provided are created under a UK data copyright exemption for non-commercial research.

Code for related tasks

We provide codes and pre-trained models for audio-language retrieval, automated audio captioning, and zero-shot audio classification.

• Retrieval
• Captioning
• Zero-shot Audio Classification
• Text-to-Sound Generation
• Models

Citation

Please cite our paper as below if you use the WavCaps dataset.

@article{mei2023WavCaps,
  title={Wav{C}aps: A {ChatGPT}-Assisted Weakly-Labelled Audio Captioning Dataset for Audio-Language Multimodal Research},
  author={Xinhao Mei and Chutong Meng and Haohe Liu and Qiuqiang Kong and Tom Ko and Chengqi Zhao and Mark D. Plumbley and Yuexian Zou and Wenwu Wang},
  journal={arXiv preprint arXiv:2303.17395},
  year={2023}
}