Video-to-Video
ThinkSound / README.md
nielsr's picture
nielsr HF Staff
Improve model card: add pipeline tag, links, abstract, features, overview, and quick start
b649f60 verified
|
raw
history blame
5.4 kB
---
license: apache-2.0
pipeline_tag: other
---
# 🎶 ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing
This repository contains the weights of the model presented in the paper [ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing](https://arxiv.org/abs/2506.21448).
**ThinkSound** is a unified Any2Audio generation framework with flow matching guided by Chain-of-Thought (CoT) reasoning. It enables multimodal audio generation and editing: generate or edit audio from video, text, and audio, powered by step-by-step reasoning from Multimodal Large Language Models (MLLMs).
[\ud83d\udcda Paper (arXiv)](https://arxiv.org/abs/2506.21448) | [\ud83d\udcda Paper (Hugging Face)](https://huggingface.co/papers/2506.21448) | [\ud83d\udcbb Code](https://github.com/liuhuadai/ThinkSound) | [\ud83c\udf10 Project Page](https://thinksound-project.github.io/) | [\ud83e\udd17 Hugging Face Space Demo](https://huggingface.co/spaces/FunAudioLLM/ThinkSound)
<div align="center">
<img src="https://raw.githubusercontent.com/liuhuadai/ThinkSound/main/assets/figs/fig1_teaser.png" alt="ThinkSound Teaser" width="100%"/>
</div>
## Abstract
While end-to-end video-to-audio generation has greatly improved, producing high-fidelity audio that authentically captures the nuances of visual content remains challenging. Like professionals in the creative industries, such generation requires sophisticated reasoning about items such as visual dynamics, acoustic environments, and temporal relationships. We present ThinkSound, a novel framework that leverages Chain-of-Thought (CoT) reasoning to enable stepwise, interactive audio generation and editing for videos. Our approach decomposes the process into three complementary stages: foundational foley generation that creates semantically coherent soundscapes, interactive object-centric refinement through precise user interactions, and targeted editing guided by natural language instructions. At each stage, a multimodal large language model generates contextually aligned CoT reasoning that guides a unified audio foundation model. Furthermore, we introduce AudioCoT, a comprehensive dataset with structured reasoning annotations that establishes connections between visual content, textual descriptions, and sound synthesis. Experiments demonstrate that ThinkSound achieves state-of-the-art performance in video-to-audio generation across both audio metrics and CoT metrics and excels in out-of-distribution Movie Gen Audio benchmark.
## \ud83d\ude80 Features
- **Any2Audio**: Generate audio from arbitrary modalities — video, text, audio, or their combinations.
- **Video-to-Audio SOTA**: Achieves state-of-the-art results on multiple V2A benchmarks.
- **CoT-Driven Reasoning**: Chain-of-Thought reasoning for compositional and controllable audio generation via MLLMs.
- **Interactive Object-centric Editing**: Refine or edit specific sound events by clicking on visual objects or using text instructions.
- **Unified Framework**: One foundation model supports generation, editing, and interactive workflow.
## \u2728 Method Overview
ThinkSound decomposes audio generation and editing into three interactive stages, all guided by MLLM-based Chain-of-Thought (CoT) reasoning:
1. **Foley Generation:** Generate foundational, semantically and temporally aligned soundscapes from video.
2. **Object-Centric Refinement:** Refine or add sounds for user-specified objects via clicks or regions in the video.
3. **Targeted Audio Editing:** Modify generated audio using high-level natural language instructions.
<div align="center">
<img src="https://raw.githubusercontent.com/liuhuadai/ThinkSound/main/assets/figs/fig3_model.png" alt="ThinkSound Overview" width="100%"/>
</div>
## \u26a1 Quick Start
### Environment Preparation
```bash
git clone https://github.com/liuhuadai/ThinkSound.git
cd ThinkSound
pip install -r requirements.txt
conda install -y -c conda-forge 'ffmpeg<7'
# Download pretrained weights from Hugging Face:
# git lfs install
# git clone https://huggingface.co/liuhuadai/ThinkSound
# Move downloaded weights (e.g., "ThinkSound" folder containing model files) to ckpts/ directory
```
### Run the script
```bash
chmod +x scripts/demo.sh
./scripts/demo.sh <video_path> <title> <CoT description>
```
### Web Interface Usage
For an interactive experience, launch the Gradio web interface:
```bash
python app.py
```
## \ud83d\udcc4 License
This project is released under the [Apache 2.0 License](https://github.com/liuhuadai/ThinkSound/blob/main/LICENSE).
> **Note:**
> The code, models, and dataset are **for research and educational purposes only**.
> **Commercial use is NOT permitted.**
>
> For commercial licensing, please contact the authors.
## \ud83d\udcd6 Citation
If you find ThinkSound useful in your research or work, please cite our paper:
```bibtex
@misc{liu2025thinksoundchainofthoughtreasoningmultimodal,
title={ThinkSound: Chain-of-Thought Reasoning in Multimodal Large Language Models for Audio Generation and Editing},
author={Huadai Liu and Jialei Wang and Kaicheng Luo and Wen Wang and Qian Chen and Zhou Zhao and Wei Xue},
year={2025},
eprint={2506.21448},
archivePrefix={arXiv},
primaryClass={eess.AS},
url={https://arxiv.org/abs/2506.21448},
}
```