MiDashengLM-Gen

Unified Audio Scene Generation via LLM-Driven Autoregressive Flow Matching

Overview

MiDashengLM-Gen is an end-to-end framework that uses a pre-trained Large Language Model and audio tokenizer as the backbone, combined with per-token conditional flow matching for autoregressive, variable-length mixed-audio scene generation. It generates coherent 16 kHz audio scenes that simultaneously blend speech, music, sound effects, and environmental acoustics from structured text descriptions.

Highlights

Architecture

Architecture
Left: training pipeline with flow matching loss. Right: autoregressive inference pipeline.
ComponentDetails
Audio TokenizerDashengTokenizer → 768-dim latents @ 25 Hz, downsampled to 5 Hz via audio projector
LLM BackboneQwen3-1.7B, fully fine-tuned
Flow Matching DiT16 layers, hidden dim 2048, 8 heads, MLP ratio 4.0
Inference10-step Euler ODE solver, CFG scale 2.0
Stop HeadLearned binary classifier for variable-length truncation
Output16 kHz mono WAV, auto-truncated

Input Format

Input uses structured multi-view captions with special tokens to describe different aspects of an audio scene. Use <|unknown|> for absent elements.

TokenDescriptionExample
<|caption|>Overall scene descriptionA comedian delivering a punchline followed by uproarious crowd laughter and an upbeat jazz band hit
<|asr|>Speech transcriptAnd that is why I never buy cheap luggage anymore!
<|speech|>Speaker characteristics (voice, emotion, style)expressive comedic male voice
<|sfx|>Sound effectsuproarious crowd laughter
<|music|>Music descriptionsudden upbeat jazz band sting
<|env|>Environment / ambienceintimate comedy club

Audio Demonstrations

Audio samples across seven categories, demonstrating the model's unified generation capabilities.

Citation

@article{sun2026midashenglmgen,
  title={MiDashengLM-Gen: Unified Audio Scene Generation via LLM-Driven Autoregressive Flow Matching},
  author={Sun, Xingwei and Dinkel, Heinrich and Li, Gang and Mei, Jiahao and Niu, Yadong and Han, Zerui and Jiang, Yuepeng and Zhou, Jiahao and Fan, Lichun and Luan, Jian},
  journal={arXiv preprint},
  year={2026}
}

Licensed under the Apache License 2.0.