tsinghua-ee/WAVE-7B

Model

WAVE: Learning Unified & Versatile Audio-Visual Embeddings with Multimodal LLM

6

10 commits

3 repos using this model

1 linked in READMEs

updated Jul 22, 2026

See the code

README

WAVE: Learning Unified & Versatile Audio-Visual Embeddings with Multimodal LLM


πŸ”₯ News

  • [2026/02/11]: πŸš€ We released the official code and model checkpoints!
  • [2026/02/06]: πŸŽ‰ WAVE has been accepted to ICLR 2026 as an Oral Presentation!

πŸ“– Abstract

We introduce WAVE (Unified & Versatile Audio-Visual Embeddings), the first LLM-based embedding model that creates a unified representation space for text, audio, silent video, and synchronized audio-visual modalities.

Unlike previous methods that use separate encoders, WAVE is built upon Qwen2.5-Omni and employs a novel hierarchical feature fusion strategy. It features a dual-encoder design for audio (Speech + Environmental Sounds) and is trained via a joint multi-modal, multi-task approach.

Key Capabilities:

  1. Any-to-Any Retrieval: Seamlessly retrieve across text, audio, and video modalities.
  2. Prompt-Aware Embeddings: Generate embeddings tailored to user instructions (e.g., specific questions), significantly boosting performance in multimodal QA.
  3. SOTA Performance: WAVE achieves state-of-the-art results on the MMEB-v2-video benchmark and excels in audio/audio-visual retrieval tasks.
WAVE Architecture
Overview of WAVE Architecture. It supports text-only, visual-only, audio-only, and audio-visual inputs.

πŸ† Performance

WAVE achieves SOTA performance on MMEB-v2-video, outperforming industrial-grade models like Seed-1.6-Embedding.

ModelOverallCLSQARETMRET
LamRA 7B35.039.342.624.332.8
GME 7B38.437.450.428.437.0
CAFe 7B42.435.858.734.439.5
Seed-1.6-Embedding55.355.060.951.353.5
WAVE 7B (Ours)59.957.872.554.750.8

WAVE also demonstrates strong performance across audio and audio-visual modalities.

DatasetAudioCapsClothoMMAUMMARVGGSoundMusicCaps
TaskA-RETA-RETA-QAA-QAAV-RETAV-RET
ReferenceWavCapsWavCapsQwen2.5-OmniQwen2.5-OmniEncoder-onlyEncoder-only
Ref. Value42.221.571.556.710.38.6
WAVE 7B44.225.676.668.125.020.4

πŸ› οΈ Installation

For key library versions, please refer to requirements.txt. Install via:

pip install -r requirements.txt


πŸ€– Model Zoo

We provide the model checkpoint on Hugging Face.

ModelParametersBackboneDownload
WAVE-7B7BQwen2.5-OmniπŸ€— HuggingFace

πŸš€ Quick Start

To evaluate WAVE on your own datasets or benchmarks, follow these steps:

  1. Data Preparation: Organize your data into the sample JSON format. Refer to the templates provided in the scripts/ directory, such as:

    • scripts/ret_av2t.json (video/audio to text retrieval)
    • scripts/ret_mc.json (QA)
    • scripts/ret_v2a.json (audio-visual retrieval)
  2. Inference: Modify scripts/direct_test.sh, then run the script:

    bash scripts/direct_test.sh
    

    The results will be automatically saved in the output/test directory.


πŸ“ Citation

If you find WAVE useful for your research, please consider citing our paper:

@inproceedings{tang2026wave,
    title={{WAVE}: Learning Unified \& Versatile Audio-Visual Embeddings with Multimodal {LLM}},
    author={Changli Tang and Qinfan Xiao and Ke Mei and Tianyi Wang and Fengyun Rao and Chao Zhang},
    booktitle={The Fourteenth International Conference on Learning Representations},
    year={2026},
    url={https://openreview.net/forum?id=MiV3WXDYJb}
}

License: Apache 2.0

custom_code
pytorch
qwen2_5_omni_thinker

tsinghua-ee/WAVE-7B

Model

WAVE: Learning Unified & Versatile Audio-Visual Embeddings with Multimodal LLM

6

10 commits

3 repos using this model

1 linked in READMEs

updated Jul 22, 2026

See the code

README

WAVE: Learning Unified & Versatile Audio-Visual Embeddings with Multimodal LLM


πŸ”₯ News

  • [2026/02/11]: πŸš€ We released the official code and model checkpoints!
  • [2026/02/06]: πŸŽ‰ WAVE has been accepted to ICLR 2026 as an Oral Presentation!

πŸ“– Abstract

We introduce WAVE (Unified & Versatile Audio-Visual Embeddings), the first LLM-based embedding model that creates a unified representation space for text, audio, silent video, and synchronized audio-visual modalities.

Unlike previous methods that use separate encoders, WAVE is built upon Qwen2.5-Omni and employs a novel hierarchical feature fusion strategy. It features a dual-encoder design for audio (Speech + Environmental Sounds) and is trained via a joint multi-modal, multi-task approach.

Key Capabilities:

  1. Any-to-Any Retrieval: Seamlessly retrieve across text, audio, and video modalities.
  2. Prompt-Aware Embeddings: Generate embeddings tailored to user instructions (e.g., specific questions), significantly boosting performance in multimodal QA.
  3. SOTA Performance: WAVE achieves state-of-the-art results on the MMEB-v2-video benchmark and excels in audio/audio-visual retrieval tasks.
WAVE Architecture
Overview of WAVE Architecture. It supports text-only, visual-only, audio-only, and audio-visual inputs.

πŸ† Performance

WAVE achieves SOTA performance on MMEB-v2-video, outperforming industrial-grade models like Seed-1.6-Embedding.

ModelOverallCLSQARETMRET
LamRA 7B35.039.342.624.332.8
GME 7B38.437.450.428.437.0
CAFe 7B42.435.858.734.439.5
Seed-1.6-Embedding55.355.060.951.353.5
WAVE 7B (Ours)59.957.872.554.750.8

WAVE also demonstrates strong performance across audio and audio-visual modalities.

DatasetAudioCapsClothoMMAUMMARVGGSoundMusicCaps
TaskA-RETA-RETA-QAA-QAAV-RETAV-RET
ReferenceWavCapsWavCapsQwen2.5-OmniQwen2.5-OmniEncoder-onlyEncoder-only
Ref. Value42.221.571.556.710.38.6
WAVE 7B44.225.676.668.125.020.4

πŸ› οΈ Installation

For key library versions, please refer to requirements.txt. Install via:

pip install -r requirements.txt


πŸ€– Model Zoo

We provide the model checkpoint on Hugging Face.

ModelParametersBackboneDownload
WAVE-7B7BQwen2.5-OmniπŸ€— HuggingFace

πŸš€ Quick Start

To evaluate WAVE on your own datasets or benchmarks, follow these steps:

  1. Data Preparation: Organize your data into the sample JSON format. Refer to the templates provided in the scripts/ directory, such as:

    • scripts/ret_av2t.json (video/audio to text retrieval)
    • scripts/ret_mc.json (QA)
    • scripts/ret_v2a.json (audio-visual retrieval)
  2. Inference: Modify scripts/direct_test.sh, then run the script:

    bash scripts/direct_test.sh
    

    The results will be automatically saved in the output/test directory.


πŸ“ Citation

If you find WAVE useful for your research, please consider citing our paper:

@inproceedings{tang2026wave,
    title={{WAVE}: Learning Unified \& Versatile Audio-Visual Embeddings with Multimodal {LLM}},
    author={Changli Tang and Qinfan Xiao and Ke Mei and Tianyi Wang and Fengyun Rao and Chao Zhang},
    booktitle={The Fourteenth International Conference on Learning Representations},
    year={2026},
    url={https://openreview.net/forum?id=MiV3WXDYJb}
}

License: Apache 2.0

custom_code
pytorch
qwen2_5_omni_thinker