Text-to-Audio Generation using Instruction-Guided Latent Diffusion Model

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)

Tango 2: Improving Diffusion-based Text-to-Audio Generation using Direct Preference Optimization

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)

Tango 2: Improving Diffusion-based Text-to-Audio Generation using Direct Preference Optimization based Alignment

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)

Tango 2: Enhancing Diffusion-based Text-to-Audio Generation through Direct Preference Optimization-based Alignment

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)

Tango 2: Aligning Diffusion-based Text-to-Audio Generative Models through Direct Preference Optimization

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)

Tango 2: Aligning Diffusion-based Text-to-Audio Generations through Direct Preference Optimization

Navonil Majumder Singapore University of Technology and Design, Singapore Chia-Yu Hung Singapore University of Technology and Design, Singapore Deepanway Ghosal Singapore University of Technology and Design, Singapore Wei-Ning Hsu Meta AI, USA Rada Mihalcea University of Michigan, USA  and  Soujanya Poria Singapore University of Technology and Design, Singapore
(2024)
Abstract.

Generative multimodal content is increasingly prevalent in much of the content creation arena, as it has the potential to allow artists and media personnel to create pre-production mockups by quickly bringing their ideas to life. The generation of audio from text prompts is an important aspect of such processes in the music and film industry. Many of the recent diffusion-based text-to-audio models focus on training increasingly sophisticated diffusion models on a large set of datasets of prompt-audio pairs. These models do not explicitly focus on the presence of concepts or events and their temporal ordering in the output audio with respect to the input prompt. Our hypothesis is focusing on how these aspects of audio generation could improve audio generation performance in the presence of limited data. As such, in this work, using an existing text-to-audio model Tango, we synthetically create a preference dataset where each prompt has a winner audio output and some loser audio outputs for the diffusion model to learn from. The loser outputs, in theory, have some concepts from the prompt missing or in an incorrect order. We fine-tune the publicly available Tango text-to-audio model using diffusion-DPO (direct preference optimization) loss on our preference dataset and show that it leads to improved audio output over Tango and AudioLDM2, in terms of both automatic- and manual-evaluation metrics.

Multimodal AI, Text-to-Audio Generation, Diffusion Models, Large Language Models, Preference Optimization
journalyear: 2024ccs: Computing methodologies Natural language processingccs: Information systems Multimedia information systems

1. Introduction

Generative AI is increasingly turning into a mainstay of our daily lives, be it directly through using ChatGPT (OpenAI, 2023c), GPT-4 (OpenAI, 2023b) in an assistive capacity, or indirectly by consuming AI-generated memes, generated using models like StableDiffusion (Rombach et al., 2022), DALL-E 3 (OpenAI, 2023a; Betker et al., [n. d.]), on social media platforms. Nonetheless, there is a massive demand for AI-generated content across industries, especially in the multimedia sector. Quick creation of audio-visual content or prototypes would require an effective text-to-audio model along with text-to-image and -video models. Thus, improving the fidelity of such models with respect to the input prompts is paramount.

Recently, supervised fine-tuning-based direct preference optimization (Rafailov et al., 2023) (DPO) has emerged as a cheaper and more robust alternative to reinforcement learning with human feedback (RLHF) to align LLM responses with human preferences. This idea is subsequently adapted for diffusion models by Wallace et al. (2023) to align the denoised outputs to human preferences. In this work, we employ this DPO-diffusion approach to improve the semantic alignment between input prompt and output audio of a text-to-audio model. Particularly, we fine-tune the publicly available text-to-audio latent diffusion model Tango (Ghosal et al., 2023a) on our synthesized preference dataset with DPO-diffusion loss. This preference dataset contains diverse audio descriptions (prompts) with their respective preferred (winner) and undesirable (loser) audios. The preferred audios are supposed to perfectly reflect their respective textual descriptions, whereas the undesirable audios have some flaws, such as some missing concepts from the prompt or in an incorrect temporal order or high noise level. To this end, we perturbed the descriptions to remove or change the order of certain concepts and passed them to Tango to generate undesirable audios. Another strategy that we adopted for undesirable audio generation was adversarial filtering: generate multiple audios from the original prompt and choose the audio samples with CLAP-score below a certain threshold. We call this preference dataset Audio-alpaca. To mitigate the effect of noisy preference pairs stemming from automatic generation, we further choose a subset of samples for DPO fine-tuning based on certain thresholds defined on the CLAP-score differential between preferred and undesirable audios and the CLAP-score of the undesirable audios. This likely ensures a minimal proximity to the input prompt, while guaranteeing a minimum distance between the preference pairs.

We experimentally show that fine-tuning Tango on the pruned Audio-alpaca yields Tango 2 that significantly surpasses Tango and AudioLDM2 in both objective and human evaluations. Moreover, exposure to the contrast between good and bad audio outputs during DPO fine-tuning likely allows Tango 2 to better map the semantics of the input prompt into the audio space, despite relying on the same dataset as Tango for synthetic preference data-creation.

The broad contributions of this paper are the following:

  1. (1)

    We develop a cheap and effective heuristics for semi automatically creating a preference dataset for text-to-audio generation;

  2. (2)

    On the same note, we also share the preference dataset Audio-alpaca for text-to-audio generation that may aid in the future development of such models;

  3. (3)

    Despite not sourcing additional out-of-distribution text-audio pairs over Tango, our model Tango 2 outperforms both Tango and AudioLDM2 on both objective and subjective metrics;

  4. (4)

    Tango 2 demonstrates the applicability of diffusion-DPO in audio generation.

2. Related Work

Text-to-audio generation has garnered serious attention lately thanks to models like AudioLDM (Liu et al., 2023a), Make-an-Audio (Huang et al., 2023), Tango (Ghosal et al., 2023a), and Audiogen (Kreuk et al., 2022). These models rely on diffusion architectures for audio generation from textual prompts. Recently, AudioLM (Borsos et al., 2023) was proposed which utilizes the state-of-the-art semantic model w2v-Bert (Chung et al., 2021) to generate semantic tokens from audio prompts. These tokens condition the generation of acoustic tokens, which are decoded using the acoustic model SoundStream (Zeghidour et al., 2022) to produce audio. The semantic tokens generated by w2v-Bert are crucial for conditioning the generation of acoustic tokens, subsequently decoded by SoundStream.

AudioLDM (Liu et al., 2023a) is a text-to-audio framework that employs CLAP (Wu et al., 2023), a joint audio-text representation model, and a latent diffusion model (LDM). Specifically, an LDM is trained to generate latent representations of melspectrograms obtained using a Variational Autoencoder (VAE). During diffusion, CLAP embeddings guide the generation process. Tango (Ghosal et al., 2023b) utilizes the pre-trained VAE from AudioLDM and replaces the CLAP model with a fine-tuned large language model: FLAN-T5. This substitution aims to achieve comparable or superior results while training with a significantly smaller dataset.

In the realm of aligning generated audio with human perception, Liao et al. (2024) recently introduced BATON, a framework that initially gathers pairs of audio and textual prompts, followed by annotating them based on human preference. This dataset is subsequently employed to train a reward model. The reward generated by this model is then integrated into the standard diffusion loss to guide the network, leveraging feedback from the reward model. However, our approach significantly diverges from this work in two key aspects: 1) we automatically construct a pairwise preference dataset, referred to as Audio-alpaca, utilizing various techniques such as LLM-guided prompt perturbation and re-ranking of generated audio from Tango using CLAP scores, and 2) we then train Tango on Audio-alpaca using diffusion-DPO to generate audio samples preferred by human perception.

3. Background

3.1. Overview of Tango

Tango, proposed by Ghosal et al. (2023a), primarily relies on a latent diffusion model (LDM) and an instruction-tuned LLM for text-to-audio generation. It has three major components:

  1. (1)

    Textual-prompt encoder

  2. (2)

    Latent diffusion model (LDM)

  3. (3)

    Audio VAE and Vocoder

The textual-prompt encoder encodes the input description of the audio. Subsequently, the textual representation is used to construct a latent representation of the audio or audio prior from standard Gaussian noise, using reverse diffusion. Thereafter, the decoder of the mel-spectrogram VAE constructs a mel-spectrogram from the latent audio representation. This mel-spectrogram is fed to a vocoder to generate the final audio.

3.1.1. Textual Prompt Encoder

Tango utilizes the pre-trained LLM Flan-T5-Large (780M) (Chung et al., 2022) as the text encoder (Etextsubscript𝐸𝑡𝑒𝑥𝑡E_{text}italic_E start_POSTSUBSCRIPT italic_t italic_e italic_x italic_t end_POSTSUBSCRIPT) to acquire text encoding τL×dtext𝜏superscript𝐿subscript𝑑𝑡𝑒𝑥𝑡\tau\in\mathbb{R}^{L\times d_{text}}italic_τ ∈ blackboard_R start_POSTSUPERSCRIPT italic_L × italic_d start_POSTSUBSCRIPT italic_t italic_e italic_x italic_t end_POSTSUBSCRIPT end_POSTSUPERSCRIPT, where L𝐿Litalic_L and dtextsubscript𝑑𝑡𝑒𝑥𝑡d_{text}italic_d start_POSTSUBSCRIPT italic_t italic_e italic_x italic_t end_POSTSUBSCRIPT represent the token count and token-embedding size, respectively.

3.1.2. Latent Diffusion Model

For ease of understanding, we briefly introduce the LDM of Tango in this section. The latent diffusion model (LDM) (Rombach et al., 2022) in Tango is derived from the work of Liu et al. (2023b), aiming to construct the audio prior x0subscript𝑥0x_{0}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT guided by text encoding τ𝜏\tauitalic_τ. This task essentially involves approximating the true prior q(x0|τ)𝑞conditionalsubscript𝑥0𝜏q(x_{0}|\tau)italic_q ( italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT | italic_τ ) using parameterized pθ(x0|τ)subscript𝑝𝜃conditionalsubscript𝑥0𝜏p_{\theta}(x_{0}|\tau)italic_p start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT | italic_τ ).

LDM achieves this objective through forward and reverse diffusion processes. The forward diffusion represents a Markov chain of Gaussian distributions with scheduled noise parameters 0<β1<β2<<βN<10subscript𝛽1subscript𝛽2subscript𝛽𝑁10<\beta_{1}<\beta_{2}<\cdots<\beta_{N}<10 < italic_β start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT < italic_β start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT < ⋯ < italic_β start_POSTSUBSCRIPT italic_N end_POSTSUBSCRIPT < 1, facilitating the sampling of noisier versions of x0subscript𝑥0x_{0}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT:

(1) q(xn|xn1)𝑞conditionalsubscript𝑥𝑛subscript𝑥𝑛1\displaystyle q(x_{n}|x_{n-1})italic_q ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT | italic_x start_POSTSUBSCRIPT italic_n - 1 end_POSTSUBSCRIPT ) =𝒩(1βnxn1,βn𝐈),absent𝒩1subscript𝛽𝑛subscript𝑥𝑛1subscript𝛽𝑛𝐈\displaystyle=\mathcal{N}(\sqrt{1-\beta_{n}}x_{n-1},\beta_{n}\mathbf{I}),= caligraphic_N ( square-root start_ARG 1 - italic_β start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG italic_x start_POSTSUBSCRIPT italic_n - 1 end_POSTSUBSCRIPT , italic_β start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT bold_I ) ,
(2) q(xn|x0)𝑞conditionalsubscript𝑥𝑛subscript𝑥0\displaystyle q(x_{n}|x_{0})italic_q ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT | italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT ) =𝒩(α¯nx0,(1α¯n)𝐈),absent𝒩subscript¯𝛼𝑛subscript𝑥01subscript¯𝛼𝑛𝐈\displaystyle=\mathcal{N}(\sqrt{\overline{\alpha}_{n}}x_{0},(1-\overline{% \alpha}_{n})\mathbf{I}),= caligraphic_N ( square-root start_ARG over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT , ( 1 - over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT ) bold_I ) ,

where N𝑁Nitalic_N is the number of forward diffusion steps, αn=1βnsubscript𝛼𝑛1subscript𝛽𝑛\alpha_{n}=1-\beta_{n}italic_α start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT = 1 - italic_β start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT, and α¯n=i=1nαnsubscript¯𝛼𝑛superscriptsubscriptproduct𝑖1𝑛subscript𝛼𝑛\overline{\alpha}_{n}=\prod_{i=1}^{n}\alpha_{n}over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT = ∏ start_POSTSUBSCRIPT italic_i = 1 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_n end_POSTSUPERSCRIPT italic_α start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT. Song et al. (2020) show that Eq. 2 conveniently follows from Eq. 1 through reparametrization trick that allows direct sampling of any xnsubscript𝑥𝑛x_{n}italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT from x0subscript𝑥0x_{0}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT via a non-Markovian process:

(3) xn=α¯nx0+(1α¯n)ϵ,subscript𝑥𝑛subscript¯𝛼𝑛subscript𝑥01subscript¯𝛼𝑛italic-ϵx_{n}=\sqrt{\overline{\alpha}_{n}}x_{0}+(1-\overline{\alpha}_{n})\epsilon,italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT = square-root start_ARG over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT + ( 1 - over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT ) italic_ϵ ,

where the noise term ϵ𝒩(𝟎,𝐈)similar-toitalic-ϵ𝒩0𝐈\epsilon\sim\mathcal{N}(\mathbf{0},\mathbf{I})italic_ϵ ∼ caligraphic_N ( bold_0 , bold_I ). The final step of the forward process yields xN𝒩(𝟎,𝐈)similar-tosubscript𝑥𝑁𝒩0𝐈x_{N}\sim\mathcal{N}(\mathbf{0},\mathbf{I})italic_x start_POSTSUBSCRIPT italic_N end_POSTSUBSCRIPT ∼ caligraphic_N ( bold_0 , bold_I ).

The reverse process denoises and reconstructs x0subscript𝑥0x_{0}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT through text-guided noise estimation (ϵ^θsubscript^italic-ϵ𝜃\hat{\epsilon}_{\theta}over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT) using loss

(4) LDM=n=1Nγn𝔼ϵn𝒩(𝟎,𝐈),x0ϵnϵ^θ(n)(xn,τ)22,subscript𝐿𝐷𝑀superscriptsubscript𝑛1𝑁subscript𝛾𝑛subscript𝔼similar-tosubscriptitalic-ϵ𝑛𝒩0𝐈subscript𝑥0superscriptsubscriptnormsubscriptitalic-ϵ𝑛superscriptsubscript^italic-ϵ𝜃𝑛subscript𝑥𝑛𝜏22\displaystyle\mathcal{L}_{LDM}=\sum_{n=1}^{N}\gamma_{n}\mathbb{E}_{\epsilon_{n% }\sim\mathcal{N}(\mathbf{0},\mathbf{I}),x_{0}}||\epsilon_{n}-\hat{\epsilon}_{% \theta}^{(n)}(x_{n},\tau)||_{2}^{2},caligraphic_L start_POSTSUBSCRIPT italic_L italic_D italic_M end_POSTSUBSCRIPT = ∑ start_POSTSUBSCRIPT italic_n = 1 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_N end_POSTSUPERSCRIPT italic_γ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT blackboard_E start_POSTSUBSCRIPT italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT ∼ caligraphic_N ( bold_0 , bold_I ) , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT end_POSTSUBSCRIPT | | italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT - over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) | | start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT ,

where xnsubscript𝑥𝑛x_{n}italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT is sampled according to Eq. 3 using standard normal noise ϵnsubscriptitalic-ϵ𝑛\epsilon_{n}italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT, τ𝜏\tauitalic_τ represents the text encoding for guidance, and γnsubscript𝛾𝑛\gamma_{n}italic_γ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT denotes the weight of reverse step n𝑛nitalic_n (Hang et al., 2023), interpreted as a measure of signal-to-noise ratio (SNR) relative to α1:Nsubscript𝛼:1𝑁\alpha_{1:N}italic_α start_POSTSUBSCRIPT 1 : italic_N end_POSTSUBSCRIPT. The estimated noise is then employed for the reconstruction of x0subscript𝑥0x_{0}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT:

(5) pθ(x0:N|τ)subscript𝑝𝜃conditionalsubscript𝑥:0𝑁𝜏\displaystyle p_{\theta}(x_{0:N}|\tau)italic_p start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) =p(xN)n=1Npθ(xn1|xn,τ),absent𝑝subscript𝑥𝑁superscriptsubscriptproduct𝑛1𝑁subscript𝑝𝜃conditionalsubscript𝑥𝑛1subscript𝑥𝑛𝜏\displaystyle=p(x_{N})\prod_{n=1}^{N}p_{\theta}(x_{n-1}|x_{n},\tau),= italic_p ( italic_x start_POSTSUBSCRIPT italic_N end_POSTSUBSCRIPT ) ∏ start_POSTSUBSCRIPT italic_n = 1 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_N end_POSTSUPERSCRIPT italic_p start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n - 1 end_POSTSUBSCRIPT | italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) ,
(6) pθ(xn1|xn,τ)subscript𝑝𝜃conditionalsubscript𝑥𝑛1subscript𝑥𝑛𝜏\displaystyle p_{\theta}(x_{n-1}|x_{n},\tau)italic_p start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n - 1 end_POSTSUBSCRIPT | italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) =𝒩(μθ(n)(xn,τ),β~(n)),absent𝒩subscriptsuperscript𝜇𝑛𝜃subscript𝑥𝑛𝜏superscript~𝛽𝑛\displaystyle=\mathcal{N}(\mu^{(n)}_{\theta}(x_{n},\tau),\tilde{\beta}^{(n)}),= caligraphic_N ( italic_μ start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) , over~ start_ARG italic_β end_ARG start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ) ,
(7) μθ(n)(xn,τ)superscriptsubscript𝜇𝜃𝑛subscript𝑥𝑛𝜏\displaystyle\mu_{\theta}^{(n)}(x_{n},\tau)italic_μ start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) =1αn[xn1αn1α¯nϵ^θ(n)(xn,τ)],absent1subscript𝛼𝑛delimited-[]subscript𝑥𝑛1subscript𝛼𝑛1subscript¯𝛼𝑛superscriptsubscript^italic-ϵ𝜃𝑛subscript𝑥𝑛𝜏\displaystyle=\frac{1}{\sqrt{\alpha_{n}}}[x_{n}-\frac{1-\alpha_{n}}{\sqrt{1-% \overline{\alpha}_{n}}}\hat{\epsilon}_{\theta}^{(n)}(x_{n},\tau)],= divide start_ARG 1 end_ARG start_ARG square-root start_ARG italic_α start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG end_ARG [ italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT - divide start_ARG 1 - italic_α start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG start_ARG square-root start_ARG 1 - over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG end_ARG over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT , italic_τ ) ] ,
(8) β~(n)superscript~𝛽𝑛\displaystyle\tilde{\beta}^{(n)}over~ start_ARG italic_β end_ARG start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT =1α¯n11α¯nβn.absent1subscript¯𝛼𝑛11subscript¯𝛼𝑛subscript𝛽𝑛\displaystyle=\frac{1-\bar{\alpha}_{n-1}}{1-\bar{\alpha}_{n}}\beta_{n}.= divide start_ARG 1 - over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n - 1 end_POSTSUBSCRIPT end_ARG start_ARG 1 - over¯ start_ARG italic_α end_ARG start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT end_ARG italic_β start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT .

The parameterization of noise estimation ϵ^θsubscript^italic-ϵ𝜃\hat{\epsilon}_{\theta}over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT involves utilizing U-Net (Ronneberger et al., 2015), incorporating a cross-attention component to integrate the textual guidance τ𝜏\tauitalic_τ.

3.1.3. Audio VAE and Vocoder

The audio variational auto-encoder (VAE) (Kingma and Welling, 2013) compresses the mel-spectrogram of an audio sample, mT×F𝑚superscript𝑇𝐹m\in\mathbb{R}^{T\times F}italic_m ∈ blackboard_R start_POSTSUPERSCRIPT italic_T × italic_F end_POSTSUPERSCRIPT, into an audio prior x0C×T/r×F/rsubscript𝑥0superscript𝐶𝑇𝑟𝐹𝑟x_{0}\in\mathbb{R}^{C\times T/r\times F/r}italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT ∈ blackboard_R start_POSTSUPERSCRIPT italic_C × italic_T / italic_r × italic_F / italic_r end_POSTSUPERSCRIPT, where C𝐶Citalic_C, T𝑇Titalic_T, F𝐹Fitalic_F, and r𝑟ritalic_r denote the number of channels, time-slots, frequency-slots, and compression level, respectively. The latent diffusion model (LDM) reconstructs the audio prior x^0subscript^𝑥0\hat{x}_{0}over^ start_ARG italic_x end_ARG start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT using input-text guidance τ𝜏\tauitalic_τ. Both the encoder and decoder consist of ResUNet blocks (Kong et al., 2021) and are trained by maximizing the evidence lower-bound (ELBO) (Kingma and Welling, 2013) and minimizing adversarial loss (Isola et al., 2016). Tango utilizes the checkpoint of the audio VAE provided by Liu et al. (2023b).

As a vocoder to convert the audio-VAE decoder-generated mel-spectrogram into audio, Tango employs HiFi-GAN (Kong et al., 2020) which is also utilized by Liu et al. (2023b).

Finally, Tango utilizes a data augmentation method that merges two audio signals while considering human auditory perception. This involves computing the pressure level of each audio signal and adjusting the weights of the signals to prevent the dominance of the signal with higher pressure level over the one with lower pressure level. Specifically, when fusing two audio signals, the relative pressure level is computed using the following equation:

(9) p=(1+10G1G220)1,𝑝superscript1superscript10subscript𝐺1subscript𝐺2201\displaystyle p=(1+10^{\frac{G_{1}-G_{2}}{20}})^{-1},italic_p = ( 1 + 10 start_POSTSUPERSCRIPT divide start_ARG italic_G start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT - italic_G start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_ARG start_ARG 20 end_ARG end_POSTSUPERSCRIPT ) start_POSTSUPERSCRIPT - 1 end_POSTSUPERSCRIPT ,

Here G1subscript𝐺1G_{1}italic_G start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and G2subscript𝐺2G_{2}italic_G start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT are the pressure levels of signal x1subscript𝑥1x_{1}italic_x start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and x2subscript𝑥2x_{2}italic_x start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT. Then the audio signals are mixed using the equation below:

(10) mix(x1,x2)=px1+(1p)x2p2+(1p)2.mixsubscript𝑥1subscript𝑥2𝑝subscript𝑥11𝑝subscript𝑥2superscript𝑝2superscript1𝑝2\displaystyle\text{mix}(x_{1},x_{2})=\frac{px_{1}+(1-p)x_{2}}{\sqrt{p^{2}+(1-p% )^{2}}}.mix ( italic_x start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT ) = divide start_ARG italic_p italic_x start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT + ( 1 - italic_p ) italic_x start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_ARG start_ARG square-root start_ARG italic_p start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT + ( 1 - italic_p ) start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT end_ARG end_ARG .

The denominator is to account for the fact that the energy of a sound wave is proportional to the square of its amplitude as shown in Tokozume et al. (2017). Note that in this augmentation, textual prompts are also concatenated.

3.2. Preference Optimization for Language Models

Tuning Large Language Models (LLMs) to generate responses according to human preference has been a great interest to the ML community. The most popular approach for aligning language models to human preference is reinforcement learning with human feedback (RLHF). It comprises the following steps (Rafailov et al., 2023):

Supervised Fine Tuning (SFT)

First, the pre-trained LLM undergoes supervised fine-tuning on high-quality downstream tasks to obtain the fine-tuned model πSFTsuperscript𝜋𝑆𝐹𝑇\pi^{SFT}italic_π start_POSTSUPERSCRIPT italic_S italic_F italic_T end_POSTSUPERSCRIPT.

Refer to caption
Figure 1. An illustration of our pipeline for text-to-audio alignment. The top part depicts the preference dataset creation where three strategies are deployed to generate the undesirable audio outputs to the input prompts. These samples are further filtered to form Audio-alpaca. This preference dataset is finally used to align Tango using DPO-diffusion loss (Eq. 17), resulting in Tango 2.
Reward Modeling

Next, πSFTsuperscript𝜋𝑆𝐹𝑇\pi^{SFT}italic_π start_POSTSUPERSCRIPT italic_S italic_F italic_T end_POSTSUPERSCRIPT is prompted with an input τ𝜏\tauitalic_τ to generate multiple responses. These responses are then shown to human labelers to rank. Once such a rank is obtained, xwxlτsucceedssuperscript𝑥𝑤conditionalsuperscript𝑥𝑙𝜏x^{w}\succ x^{l}\mid\tauitalic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT ≻ italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ∣ italic_τ indicating xwsuperscript𝑥𝑤x^{w}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT is preferred over xlsuperscript𝑥𝑙x^{l}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT, the task is to model these preferences. Among several popular choices of preference modeling, Bradley-Terry (BT) is the most popular one which relies on the equation below:

(11) p(xwxlτ)=exp(r(τ,xw))exp(r(τ,xw))+exp(r(τ,xl))superscript𝑝succeedssuperscript𝑥𝑤conditionalsuperscript𝑥𝑙𝜏superscript𝑟𝜏superscript𝑥𝑤superscript𝑟𝜏superscript𝑥𝑤superscript𝑟𝜏superscript𝑥𝑙\displaystyle p^{*}(x^{w}\succ x^{l}\mid\tau)=\frac{\exp(r^{*}(\tau,x^{w}))}{% \exp(r^{*}(\tau,x^{w}))+\exp(r^{*}(\tau,x^{l}))}italic_p start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ( italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT ≻ italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ∣ italic_τ ) = divide start_ARG roman_exp ( italic_r start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT ) ) end_ARG start_ARG roman_exp ( italic_r start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT ) ) + roman_exp ( italic_r start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ) ) end_ARG

The overall idea is to learn the human preference distribution psuperscript𝑝p^{*}italic_p start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT. r(τ,x)superscript𝑟𝜏𝑥r^{*}(\tau,x)italic_r start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ( italic_τ , italic_x ) is a latent reward function that generates the preferences. With a static dataset created by human annotators, 𝒟={(τ(i),x(i)w,x(i)l)}i=1N𝒟superscriptsubscriptsubscript𝜏𝑖subscriptsuperscript𝑥𝑤𝑖subscriptsuperscript𝑥𝑙𝑖𝑖1𝑁\mathcal{D}=\left\{\left(\tau_{(i)},x^{w}_{(i)},x^{l}_{(i)}\right)\right\}_{i=% 1}^{N}caligraphic_D = { ( italic_τ start_POSTSUBSCRIPT ( italic_i ) end_POSTSUBSCRIPT , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT start_POSTSUBSCRIPT ( italic_i ) end_POSTSUBSCRIPT , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT start_POSTSUBSCRIPT ( italic_i ) end_POSTSUBSCRIPT ) } start_POSTSUBSCRIPT italic_i = 1 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_N end_POSTSUPERSCRIPT, one can train a reward model rϕ(τ,x)subscript𝑟italic-ϕ𝜏𝑥r_{\phi}(\tau,x)italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x ) using maximum likelihood estimation. The negative log-likelihood loss of this training can be written as follows:

(12) R(rϕ,𝒟)=𝔼(τ,xw,xl)𝒟[logσ(rϕ(τ,xw)rϕ(τ,xl))]subscript𝑅subscript𝑟italic-ϕ𝒟subscript𝔼similar-to𝜏superscript𝑥𝑤superscript𝑥𝑙𝒟delimited-[]𝜎subscript𝑟italic-ϕ𝜏superscript𝑥𝑤subscript𝑟italic-ϕ𝜏superscript𝑥𝑙\displaystyle\mathcal{L}_{R}(r_{\phi},\mathcal{D})=-\mathbb{E}_{(\tau,x^{w},x^% {l})\sim\mathcal{D}}\left[\log\sigma(r_{\phi}(\tau,x^{w})-r_{\phi}(\tau,x^{l})% )\right]caligraphic_L start_POSTSUBSCRIPT italic_R end_POSTSUBSCRIPT ( italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT , caligraphic_D ) = - blackboard_E start_POSTSUBSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ) ∼ caligraphic_D end_POSTSUBSCRIPT [ roman_log italic_σ ( italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT ) - italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ) ) ]

This formulation considers framing the problem as a binary classification problem.

RL Optimization

The final step is to leverage rϕ(τ,x)subscript𝑟italic-ϕ𝜏𝑥r_{\phi}(\tau,x)italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x ) to feedback the language model. As explained by Rafailov et al. (2023), this can be embedded into the following learning objective:

(13) maxπθ𝔼τ𝒟,xπθ(x|τ)[rϕ(τ,x)]βDKL[πθ(x|τ)πref(x|τ)]\displaystyle\max_{\pi_{\theta}}\mathbb{E}_{\tau\sim\mathcal{D},x\sim\pi_{% \theta}(x|\tau)}\left[r_{\phi}(\tau,x)\right]-\beta D_{KL}\left[\pi_{\theta}(x% |\tau)\parallel\pi_{\text{ref}}(x|\tau)\right]roman_max start_POSTSUBSCRIPT italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT end_POSTSUBSCRIPT blackboard_E start_POSTSUBSCRIPT italic_τ ∼ caligraphic_D , italic_x ∼ italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x | italic_τ ) end_POSTSUBSCRIPT [ italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x ) ] - italic_β italic_D start_POSTSUBSCRIPT italic_K italic_L end_POSTSUBSCRIPT [ italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x | italic_τ ) ∥ italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT ( italic_x | italic_τ ) ]

Here, πrefsubscript𝜋ref\pi_{\text{ref}}italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT represents the reference model, which in this context is the supervised fine-tuned model denoted as πSFTsuperscript𝜋SFT\pi^{\text{SFT}}italic_π start_POSTSUPERSCRIPT SFT end_POSTSUPERSCRIPT. πθsubscript𝜋𝜃\pi_{\theta}italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT stands for the policy language model, intended for enhancement based on feedback from rϕ(τ,x)subscript𝑟italic-ϕ𝜏𝑥r_{\phi}(\tau,x)italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x ). β𝛽\betaitalic_β governs πθsubscript𝜋𝜃\pi_{\theta}italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT to prevent significant divergence from πrefsubscript𝜋ref\pi_{\text{ref}}italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT. This control is crucial as it ensures that the model stays close to the distributions upon which rϕ(τ,x)subscript𝑟italic-ϕ𝜏𝑥r_{\phi}(\tau,x)italic_r start_POSTSUBSCRIPT italic_ϕ end_POSTSUBSCRIPT ( italic_τ , italic_x ) was trained. Since the outputs from LLM are discrete, Eq. 13 becomes non-differentiable, necessitating reinforcement learning methods like PPO to address this objective.

4. Methodology

The two major parts of our approach (i) creation of preference dataset Audio-alpaca and (ii) DPO for alignment are outlined in Fig. 1.

4.1. Creation of Audio-alpaca

4.1.1. Audio Generation from Text Prompts

Our first step is to create audio samples from various text prompts with the pre-trained Tango model. We follow three different strategies as follows:

Strategy 1: Multiple Inferences from the same Prompt

In the first setting, we start by selecting a subset of diverse captions from the training split of the AudioCaps dataset. We use the sentence embedding model gte-large111hf.co/thenlper/gte-large (Li et al., 2023) to compute dense embedding vectors of all the captions in the training set. We then perform K-Means clustering on the embedded vectors with 200 clusters. Finally, we select 70 samples from each cluster to obtain a total of 14,000 captions. We denote the selected caption set as 𝒯1subscript𝒯1\mathcal{T}_{1}caligraphic_T start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT.

The captions selected through the above process constitute the seed caption set. Now, we follow two settings to generate audio samples from these captions:

  1. (1)

    Strategy 1.1: Prompt Tango-full-FT with the caption to generate four different audio samples with 5, 25, 50, and 100 denoising steps. All samples are created with a guidance scale of 3.

  2. (2)

    Strategy 1.2: Prompt Tango-full-FT with the caption to generate four different audio samples each with 50 denoising steps. All samples are created with a guidance scale of 3.

In summary, we obtain (τ,x1,x2,x3,x4)𝜏subscript𝑥1subscript𝑥2subscript𝑥3subscript𝑥4(\tau,x_{1},x_{2},x_{3},x_{4})( italic_τ , italic_x start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT 3 end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT 4 end_POSTSUBSCRIPT ) from Strategy 1, where τ𝜏\tauitalic_τ denotes the caption from 𝒯1subscript𝒯1\mathcal{T}_{1}caligraphic_T start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and xisubscript𝑥𝑖x_{i}italic_x start_POSTSUBSCRIPT italic_i end_POSTSUBSCRIPT denotes the audios generated from τ𝜏\tauitalic_τ.

Table 1. Prompts used in GPT-4 for creating the perturbations and the corresponding output.
Strategy and Original Caption Prompt to GPT-4 Output from GPT-4
Strategy 2: Perturbed Prompts I have an audio clip for which the original caption is as follows: People cheering and race cars racing by. Can you generate five candidate captions that would satisfy the following requirements: 1. Crowd applauding and bicycles speeding past.
People cheering and race cars racing by. i) Would be closely related to the audio clip and the original caption. 2. Spectators clapping and trains rushing by.
ii) Would contain inaccuracies in terms of describing the audio clip i.e. they would be somewhat wrong captions for the audio clip. 3. Audience cheering and horses galloping past.
iii) However, the new captions should not be completely unrelated. Always keep some concepts from the original caption in the new one. 4. Fans shouting and airplanes flying by.
iv) Would be of similar length to the original caption. 5. Group celebrating and motorcycles revving past.
Generate only the captions in separate lines so that I can programmatically extract them later.
Strategy 3: Temporally-Perturbed Prompts I have an audio clip for which the original caption is as follows: A man is speaking then a sewing machine briefly turns on and off. Can you generate five candidate captions that would satisfy the following requirements: 1. A sewing machine briefly starts, then a man begins speaking.
A man is speaking then a sewing machine briefly turns on and off. i) Would be closely related to the audio clip and the original caption. 2. The sound of a sewing machine is heard after a man’s voice.
ii) Would have a change of order of the events described in the original caption. 3. A man’s voice is followed by the noise of a blender.
iii) Would contain inaccuracies in terms of describing the audio clip i.e. they would be somewhat wrong captions for the audio clip. 4. A woman speaks and then a sewing machine is turned on.
iv) However, the new captions should not be completely unrelated. Always keep some concepts from the original caption in the new one. 5. The noise of a sewing machine is interrupted by a man talking.
v) Would be of similar length to the original caption.
Generate only the captions in separate lines so that I can programmatically extract them later.
Strategy 2: Inferences from Perturbed Prompts

We start from the selected set 𝒯1subscript𝒯1\mathcal{T}_{1}caligraphic_T start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and make perturbations of the captions using the GPT-4 language model (OpenAI, 2023b). For a caption τ𝜏\tauitalic_τ from 𝒯1subscript𝒯1\mathcal{T}_{1}caligraphic_T start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT, we denote τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT as the perturbed caption generated from GPT-4. We add specific instructions in our input prompt to make sure that τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT is semantically or conceptually close to τ𝜏\tauitalic_τ. We show an illustration of the process in Table 1. In practice, we create five different perturbed τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT for each τ𝜏\tauitalic_τ from GPT-4, as shown in Table 1.

We then prompt Tango-full-FT with τ𝜏\tauitalic_τ and τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT to generate audio samples xτsubscript𝑥𝜏x_{\tau}italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT and xτ1subscript𝑥subscript𝜏1x_{\tau_{1}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT end_POSTSUBSCRIPT. We use 50 denoising steps with a guidance scale of 3 to generate these audio samples.

To summarize, we obtain (τ,xτ,xτ1)𝜏subscript𝑥𝜏subscript𝑥subscript𝜏1(\tau,x_{\tau},x_{\tau_{1}})( italic_τ , italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT end_POSTSUBSCRIPT ) from Strategy 2. Note that, we considered τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT only to generate the audio sample xτ1subscript𝑥subscript𝜏1x_{\tau_{1}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT end_POSTSUBSCRIPT. We do not further consider τ1subscript𝜏1\tau_{1}italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT while creating the preference dataset.

Strategy 3: Inferences from Temporally Perturbed Prompts

This strategy is aimed at prompts that describe some composition of sequence and simultaneity of events. To identify such prompts in AudioCaps’ training dataset, as a heuristics, we look for the following keywords in a prompt: while, before, after, then, or followed. We denote the set of such prompts as 𝒯2subscript𝒯2\mathcal{T}_{2}caligraphic_T start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT.

For each caption τ2subscript𝜏2\tau_{2}italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT in 𝒯2subscript𝒯2\mathcal{T}_{2}caligraphic_T start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT, we then prompt GPT-4 to create a set of temporal perturbations. The temporal perturbations include changing the order of the events in the original caption, or introducing a new event or removing an existing event, etc. We aim to create these temporal perturbations by providing specific instructions to GPT-4, which we also illustrate in Table 1.

We denote the temporally perturbed caption as τ2subscript𝜏2\tau_{2}italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT. We then follow the same process as mentioned earlier in Strategy 2 to create the audio samples xτsubscript𝑥𝜏x_{\tau}italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT and xτ2subscript𝑥subscript𝜏2x_{\tau_{2}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_POSTSUBSCRIPT. Finally, we pair the (τ,xτ,xτ2)𝜏subscript𝑥𝜏subscript𝑥subscript𝜏2(\tau,x_{\tau},x_{\tau_{2}})( italic_τ , italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT , italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_POSTSUBSCRIPT ) samples from this strategy. Analogous to the previous strategy, the τ2subscript𝜏2\tau_{2}italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT is only used to create the xτ2subscript𝑥subscript𝜏2x_{\tau_{2}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_POSTSUBSCRIPT, and is not used anywhere else for preference data creation.

We collect the paired text prompt and audio samples from the three strategies and denote it overall as (τ,x)𝜏delimited-⟨⟩𝑥(\tau,\langle x\rangle)( italic_τ , ⟨ italic_x ⟩ ), where xdelimited-⟨⟩𝑥\langle x\rangle⟨ italic_x ⟩ indicates the set of 4 or 2 generated audio samples depending upon the corresponding strategy.

4.1.2. Ranking and Preference-Data Selection

We first create a pool of candidate preference data for the three strategies as follows:

For Strategy 1

Let’s assume we have an instance (τ,x)𝜏delimited-⟨⟩𝑥(\tau,\langle x\rangle)( italic_τ , ⟨ italic_x ⟩ ) from Strategy 1. We first compute the CLAP matching score following Wu et al. (2023) between τ𝜏\tauitalic_τ and all the four audio samples in xdelimited-⟨⟩𝑥\langle x\rangle⟨ italic_x ⟩. We surmise that the sample in xdelimited-⟨⟩𝑥\langle x\rangle⟨ italic_x ⟩ that has the highest matching score with τ𝜏\tauitalic_τ is most aligned with τ𝜏\tauitalic_τ, compared to the other three audio samples that have a relatively lower matching score. We consider this audio with the highest matching score as the winning sample xwsuperscript𝑥𝑤x^{w}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT and the other three audio samples as the losing sample xlsuperscript𝑥𝑙x^{l}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT. In this setting, we can thus create a pool of three preference data points: (τ,xw,xl)𝜏superscript𝑥𝑤superscript𝑥𝑙(\tau,x^{w},x^{l})( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ), for the three losing audio samples xlsuperscript𝑥𝑙x^{l}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT.

For Strategy 2 and 3

Let’s assume we have an instance (τ,x)𝜏delimited-⟨⟩𝑥(\tau,\langle x\rangle)( italic_τ , ⟨ italic_x ⟩ ) from Strategy 2 or 3. We compute the CLAP matching score between i) τ𝜏\tauitalic_τ with xτsubscript𝑥𝜏x_{\tau}italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT, and ii) τ𝜏\tauitalic_τ with the xτ1subscript𝑥subscript𝜏1x_{\tau_{1}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT end_POSTSUBSCRIPT or xτ2subscript𝑥subscript𝜏2x_{\tau_{2}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_POSTSUBSCRIPT, corresponding to the strategy. We consider only those instances where the CLAP score of i) is higher than the CLAP score of ii). For these instances, we use xτsubscript𝑥𝜏x_{\tau}italic_x start_POSTSUBSCRIPT italic_τ end_POSTSUBSCRIPT as the winning audio xwsuperscript𝑥𝑤x^{w}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT and xτ1subscript𝑥subscript𝜏1x_{\tau_{1}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT end_POSTSUBSCRIPT or xτ2subscript𝑥subscript𝜏2x_{\tau_{2}}italic_x start_POSTSUBSCRIPT italic_τ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT end_POSTSUBSCRIPT as the losing audio xlsuperscript𝑥𝑙x^{l}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT to create the preference data point: (τ,xw,xl)𝜏superscript𝑥𝑤superscript𝑥𝑙(\tau,x^{w},x^{l})( italic_τ , italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ).

Final Selection

We want to ensure that the winning audio sample xwsuperscript𝑥𝑤x^{w}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT is strongly aligned with the text prompt τ𝜏\tauitalic_τ. At the same time, the winning audio sample should have a considerably higher alignment with the text prompt than the losing audio sample. We use the CLAP score as a measurement to fulfill these conditions. The CLAP score is measured using cosine similarity between the text and audio embeddings, where higher scores indicate higher alignment between the text and the audio. We thus use the following conditions to select a subset of instances from the pool of preference data:

  1. (1)

    The winning audio must have a minimum CLAP score of α𝛼\alphaitalic_α with the text prompt to ensure that the winning audio is strongly aligned with the text prompt.

  2. (2)

    The losing audio must have a minimum CLAP score of β𝛽\betaitalic_β with the text prompt to ensure that we have semantically close negatives that are useful for preference modeling.

  3. (3)

    The winning audio must have a higher CLAP score than the losing audio w.r.t to the text prompt.

  4. (4)

    We denote ΔΔ\Deltaroman_Δ to be the difference in CLAP score between the text prompt with the winning audio222In our paper, we employ the terms ”winner” and ”preferred” interchangeably. Likewise, we use ”loser” and ”undesirable” interchangeably throughout the text. and the text prompt with the losing audio. The ΔΔ\Deltaroman_Δ should lie between certain thresholds, where the lower bound will ensure that the losing audio is not too close to the winning audio, and the upper bound will ensure that the losing audio is not too undesirable.

We use an ensemble filtering strategy based on two different CLAP models: 630k-audioset-best and 630k-best (Wu et al., 2023). This can reduce the effect of noise from individual CLAP checkpoints and increase the robustness of the selection process. In this strategy, samples are included in our preference dataset if and only if they satisfy all the above conditions based on CLAP scores from both of the models. We denote the conditional scores mentioned above as α1,β1,Δ1subscript𝛼1subscript𝛽1subscriptΔ1\alpha_{1},\beta_{1},\Delta_{1}italic_α start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT , italic_β start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT , roman_Δ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT, and α2,β2,Δ2subscript𝛼2subscript𝛽2subscriptΔ2\alpha_{2},\beta_{2},\Delta_{2}italic_α start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT , italic_β start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT , roman_Δ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT for the two CLAP models, respectively. Based on our analysis of the distribution of the CLAP scores as shown in Figure 2, we choose their values as follows: α1=0.45,α2=0.60formulae-sequencesubscript𝛼10.45subscript𝛼20.60\alpha_{1}=0.45,\alpha_{2}=0.60italic_α start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT = 0.45 , italic_α start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT = 0.60, β1=0.40,β2=0.0formulae-sequencesubscript𝛽10.40subscript𝛽20.0\beta_{1}=0.40,\beta_{2}=0.0italic_β start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT = 0.40 , italic_β start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT = 0.0, 0.05Δ10.350.05subscriptΔ10.350.05\leq\Delta_{1}\leq 0.350.05 ≤ roman_Δ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT ≤ 0.35, and 0.08Δ20.700.08subscriptΔ20.700.08\leq\Delta_{2}\leq 0.700.08 ≤ roman_Δ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT ≤ 0.70.

Finally, our preference dataset Audio-alpaca has a total of \approx 15k samples after this selection process. We report the distribution of Audio-alpaca in Table 2.

Refer to caption
Refer to caption
Figure 2. The distribution of α1subscript𝛼1\alpha_{1}italic_α start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and Δ1subscriptΔ1\Delta_{1}roman_Δ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT scores in the unfiltered dataset. We see that for an unfiltered dataset: i) the winner audio sample is not always strongly aligned to the text prompt in the α1subscript𝛼1\alpha_{1}italic_α start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT plot; ii) winner and loser audio samples can be too close in the Δ1subscriptΔ1\Delta_{1}roman_Δ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT plot. We thus choose the values of our α1subscript𝛼1\alpha_{1}italic_α start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT, Δ1subscriptΔ1\Delta_{1}roman_Δ start_POSTSUBSCRIPT 1 end_POSTSUBSCRIPT and other selection parameters to ensure the filtered dataset is less noisy with more separation between the winner and loser audios.
Table 2. Statistics of Audio-alpaca.
Strategy # Samples Avg. Winner Score Avg. Loser Score Avg. Delta
Inference w/ Different Denoising Steps (Strategy 1.1) 3004 0.645 0.447 0.198
Inference w/ Same Denoising Steps (Strategy 1.2) 2725 0.647 0.494 0.153
GPT-4 Perturbed Prompts (Strategy 2) 4544 0.641 0.425 0.216
GPT-4 Temporally Perturbed Prompts (Strategy 3) 4752 0.649 0.458 0.191
Overall 15025 0.645 0.452 0.193

4.2. DPO for Preference Modeling

As opposed to RLHF, recently DPO has emerged as a more robust and often more practical and straightforward alternative for LLM alignment that is based on the very same BT preference model (Eq. 11). In contrast with supervised fine-tuning (SFT) that only optimizes for the desirable (winner) outputs, the DPO objective also allows the model to learn from undesirable (loser) outputs, which is key in the absence of a high-quality reward model, as required for RLHF. To this end, the DPO objective is derived by substituting the globally optimal reward—obtained by solving Eq. 13—in the negative log-likelihood (NLL) loss (Eq. 12).

This success spurred on Wallace et al. (2023) to bring the same benefits of DPO to diffusion networks. However, unlike DPO, the goal for diffusion networks is to maximize the following learning objective (Eq. 14) with a reward (Eq. 15) defined on the entire diffusion path x0:Nsubscript𝑥:0𝑁x_{0:N}italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT:

maxπθ𝔼τ𝒟,x0:Nπθ(x0:N|τ)subscriptsubscript𝜋𝜃subscript𝔼formulae-sequencesimilar-to𝜏𝒟similar-tosubscript𝑥:0𝑁subscript𝜋𝜃conditionalsubscript𝑥:0𝑁𝜏\displaystyle\max_{\pi_{\theta}}\mathbb{E}_{\tau\sim\mathcal{D},x_{0:N}\sim\pi% _{\theta}(x_{0:N}|\tau)}roman_max start_POSTSUBSCRIPT italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT end_POSTSUBSCRIPT blackboard_E start_POSTSUBSCRIPT italic_τ ∼ caligraphic_D , italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT ∼ italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) end_POSTSUBSCRIPT [r(τ,x0)]delimited-[]𝑟𝜏subscript𝑥0\displaystyle[r(\tau,x_{0})][ italic_r ( italic_τ , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT ) ]
(14) βDKL𝛽subscript𝐷KL\displaystyle-\beta D_{\text{KL}}- italic_β italic_D start_POSTSUBSCRIPT KL end_POSTSUBSCRIPT [πθ(x0:N|τ)||πref(x0:N|τ)].\displaystyle[\pi_{\theta}(x_{0:N}|\tau)||\pi_{\text{ref}}(x_{0:N}|\tau)].[ italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) | | italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) ] .
(15) r(τ,x0):=𝔼πθ(x1:N|x0,τ)assign𝑟𝜏subscript𝑥0subscript𝔼subscript𝜋𝜃conditionalsubscript𝑥:1𝑁subscript𝑥0𝜏\displaystyle r(\tau,x_{0}):=\mathbb{E}_{\pi_{\theta}(x_{1:N}|x_{0},\tau)}italic_r ( italic_τ , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT ) := blackboard_E start_POSTSUBSCRIPT italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUBSCRIPT 1 : italic_N end_POSTSUBSCRIPT | italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT , italic_τ ) end_POSTSUBSCRIPT [R(τ,x0:N)],delimited-[]𝑅𝜏subscript𝑥:0𝑁\displaystyle[R(\tau,x_{0:N})],[ italic_R ( italic_τ , italic_x start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT ) ] ,

Solving this objective and substituting the optimal reward in the NLL loss (Eq. 12) yields the following DPO objective for diffusion:

DPO-Diff=𝔼(τ,x0w,x0l)𝒟preflogσ(\displaystyle\mathcal{L}_{\text{DPO-Diff}}=-\mathbb{E}_{(\tau,x_{0}^{w},x_{0}^% {l})\sim\mathcal{D_{\text{pref}}}}\log\sigma(caligraphic_L start_POSTSUBSCRIPT DPO-Diff end_POSTSUBSCRIPT = - blackboard_E start_POSTSUBSCRIPT ( italic_τ , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ) ∼ caligraphic_D start_POSTSUBSCRIPT pref end_POSTSUBSCRIPT end_POSTSUBSCRIPT roman_log italic_σ (
(16) β𝔼x1:Nπθ(x1:N|x0,τ)[logπθ(x0:Nw|τ)πref(x0:Nw|τ)logπθ(x0:Nl|τ)πref(x0:Nl|τ)]).\displaystyle\beta\mathbb{E}_{x^{*}_{1:N}\sim\pi_{\theta}(x^{*}_{1:N}|x^{*}_{0% },\tau)}[\log\frac{\pi_{\theta}(x^{w}_{0:N}|\tau)}{\pi_{\text{ref}}(x^{w}_{0:N% }|\tau)}-\log\frac{\pi_{\theta}(x^{l}_{0:N}|\tau)}{\pi_{\text{ref}}(x^{l}_{0:N% }|\tau)}]).italic_β blackboard_E start_POSTSUBSCRIPT italic_x start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 1 : italic_N end_POSTSUBSCRIPT ∼ italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 1 : italic_N end_POSTSUBSCRIPT | italic_x start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT , italic_τ ) end_POSTSUBSCRIPT [ roman_log divide start_ARG italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) end_ARG start_ARG italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT ( italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) end_ARG - roman_log divide start_ARG italic_π start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT ( italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) end_ARG start_ARG italic_π start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT ( italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 : italic_N end_POSTSUBSCRIPT | italic_τ ) end_ARG ] ) .

Now, applying Jensen’s inequality by taking advantage of the convexity of logσ𝜎-\log\sigma- roman_log italic_σ allows the inner expectation to be pushed outside. Subsequently, approximating the denoising process with the forward process yields the following final form in terms of the L2 noise-estimation losses from LDM (Eq. 4):

DPO-Diff:=assignsubscriptDPO-Diffabsent\displaystyle\mathcal{L}_{\text{DPO-Diff}}:=caligraphic_L start_POSTSUBSCRIPT DPO-Diff end_POSTSUBSCRIPT := 𝔼n,ϵw,ϵllogσ(βNω(λn)(||ϵnwϵ^θ(n)(xnw,τ)||22\displaystyle-\mathbb{E}_{n,\epsilon^{w},\epsilon^{l}}\log\sigma(-\beta N% \omega(\lambda_{n})(||\epsilon_{n}^{w}-\hat{\epsilon}_{\theta}^{(n)}(x_{n}^{w}% ,\tau)||_{2}^{2}- blackboard_E start_POSTSUBSCRIPT italic_n , italic_ϵ start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_ϵ start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT end_POSTSUBSCRIPT roman_log italic_σ ( - italic_β italic_N italic_ω ( italic_λ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT ) ( | | italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT - over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_τ ) | | start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT
ϵnwϵ^ref(n)(xnw,τ)22superscriptsubscriptnormsuperscriptsubscriptitalic-ϵ𝑛𝑤superscriptsubscript^italic-ϵref𝑛superscriptsubscript𝑥𝑛𝑤𝜏22\displaystyle-||\epsilon_{n}^{w}-\hat{\epsilon}_{\text{ref}}^{(n)}(x_{n}^{w},% \tau)||_{2}^{2}- | | italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT - over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_τ ) | | start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT
(17) (||ϵnlϵ^θ(n)(xnl,τ)||22||ϵnlϵ^ref(n)(xnl,τ)||22)),\displaystyle-(||\epsilon_{n}^{l}-\hat{\epsilon}_{\theta}^{(n)}(x_{n}^{l},\tau% )||_{2}^{2}-||\epsilon_{n}^{l}-\hat{\epsilon}_{\text{ref}}^{(n)}(x_{n}^{l},% \tau)||_{2}^{2})),- ( | | italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT - over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT italic_θ end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT , italic_τ ) | | start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT - | | italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT - over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ( italic_n ) end_POSTSUPERSCRIPT ( italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT , italic_τ ) | | start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT 2 end_POSTSUPERSCRIPT ) ) ,

where 𝒟pref:={(τ,x0w,x0l)}assignsubscript𝒟pref𝜏superscriptsubscript𝑥0𝑤superscriptsubscript𝑥0𝑙\mathcal{D}_{\text{pref}}:=\{(\tau,x_{0}^{w},x_{0}^{l})\}caligraphic_D start_POSTSUBSCRIPT pref end_POSTSUBSCRIPT := { ( italic_τ , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT , italic_x start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT ) } is our preference dataset Audio-alpaca, τ𝜏\tauitalic_τ, x0wsubscriptsuperscript𝑥𝑤0x^{w}_{0}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT, and x0lsubscriptsuperscript𝑥𝑙0x^{l}_{0}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT start_POSTSUBSCRIPT 0 end_POSTSUBSCRIPT being the input prompt, preferred, and undesirable output, respectively. Furthermore, n𝒰(0,N)similar-to𝑛𝒰0𝑁n\sim\mathcal{U}(0,N)italic_n ∼ caligraphic_U ( 0 , italic_N ) is the diffusion step, ϵn𝒩(0,𝕀)similar-tosuperscriptsubscriptitalic-ϵ𝑛𝒩0𝕀\epsilon_{n}^{*}\sim\mathcal{N}(0,\mathbb{I})italic_ϵ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT ∼ caligraphic_N ( 0 , blackboard_I ) and xnsuperscriptsubscript𝑥𝑛x_{n}^{*}italic_x start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT start_POSTSUPERSCRIPT ∗ end_POSTSUPERSCRIPT are noise and noisy posteriors, respectively, at some step n𝑛nitalic_n. λnsubscript𝜆𝑛\lambda_{n}italic_λ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT is the signal-to-noise ratio (SNR) and ω(λn)𝜔subscript𝜆𝑛\omega(\lambda_{n})italic_ω ( italic_λ start_POSTSUBSCRIPT italic_n end_POSTSUBSCRIPT ) is a weighting function defined on SNR. We use Tango-full-FT as our reference model through its noise estimation ϵ^refsubscript^italic-ϵref\hat{\epsilon}_{\text{ref}}over^ start_ARG italic_ϵ end_ARG start_POSTSUBSCRIPT ref end_POSTSUBSCRIPT.

5. Experiments

5.1. Datasets and Training Details

We fine-tuned our model starting from the Tango-full-FT checkpoint on our preference dataset Audio-alpaca.

As mentioned earlier in Section 4.1.2, we have a total of 15,025 preference pairs in Audio-alpaca, which we use for fine-tuning. We use AdamW (Loshchilov and Hutter, 2017) with a learning rate of 9.6e-7 and a linear learning-rate scheduler for fine-tuning. Following Wallace et al. (2023), we set the β𝛽\betaitalic_β in DPO loss (Eq. 17) to 2000. We performed 1 epoch of supervised fine-tuning on the prompt and the preferred audio as training samples, followed by 4 epochs of DPO. The entirety of the fine-tuning was executed on two A100 GPUs which takes about 3.5 hours in total. We use a per GPU batch size of 4 and a gradient accumulation step of 4, resulting in an effective batch size of 32.

5.2. Baselines

We primarily compare Tango 2 to three strong baselines:

  1. (1)

    AudioLDM (Liu et al., 2023a): A text-to-audio model that uses CLAP (Wu et al., 2023), a joint audio-text representation model, and a latent diffusion model (LDM). Specifically, the LDM is trained to generate the latent representations of melspectrograms obtained from a pre-trained Variational Autoencoder (VAE). During diffusion, CLAP text-embeddings guide the generation process.

  2. (2)

    AudioLDM2 (Liu et al., 2023c): An any-to-audio framework which uses language of audio (LOA) as a joint encoding of audio, text, image, video, and other modalities. Audio modality is encoded into LOA using a self-supervised masked auto-encoder. The remaining modalities, including audio again, are mapped to LOA through a composition of GPT-2 (Radford et al., 2019) and ImageBind (Girdhar et al., 2023). This joint encoding is used as a conditioning in the diffusion network for audio generation.

  3. (3)

    Tango (Ghosal et al., 2023a): Utilizes the pre-trained VAE from AudioLDM but replaces the CLAP text-encoder with an instruction-tuned large language model: FLAN-T5. As compared to AudioLDM, its data-augmentation strategy is also cognizant of the audio pressure levels of the source audios. These innovations attain comparable or superior results while training on a significantly smaller dataset.

Baton (Liao et al., 2024) represents another recent approach in human preference based text-to-audio modeling. It trains a reward model to maximize rewards through supervised fine-tuning, aiming to maximize the probability of generating audio from a textual prompt. As discussed in Section 2, Baton’s reward model is not trained using the pairwise preference objective presented in Equation 12. In this approach, each text (τ𝜏\tauitalic_τ) and audio (x𝑥xitalic_x) pair is classified as 1 or 0, indicating whether human annotators favored the text-audio pair or not. Subsequently, this reward is incorporated into the generative objective function of the diffusion. This methodology stands in contrast to the prevailing approach in LLM alignment research. As of now, neither the dataset nor the code has been made available for comparison.

Table 3. Text-to-audio generation results on AudioCaps evaluation set. Due to time and budget constraints, we could only subjectively evaluate AudioLDM 2-Full-Large and Tango-full-FT. Notably these two models are considered open-sourced SOTA models for text-to-audio generation as reported in (Vyas et al., 2023).
Model #Parameters Objective Subjective
FAD \downarrow KL \downarrow IS \uparrow CLAP \uparrow OVL \uparrow REL \uparrow
AudioLDM-M-Full-FT 416416416416M 2.572.572.572.57 1.261.261.261.26 8.348.348.348.34 0.430.430.430.43 -- --
AudioLDM-L-Full 739739739739M 4.184.184.184.18 1.761.761.761.76 7.767.767.767.76 0.430.430.430.43 -- --
AudioLDM 2-Full 346346346346M 2.182.182.182.18 1.621.621.621.62 6.926.926.926.92 0.430.430.430.43 -- --
AudioLDM 2-Full-Large 712712712712M 2.11 1.541.541.541.54 8.298.298.298.29 0.440.440.440.44 3.563.563.563.56 3.193.193.193.19
Tango-full-FT 866866866866M 2.512.512.512.51 1.151.151.151.15 7.877.877.877.87 0.540.540.540.54 3.813.813.813.81 3.773.773.773.77
Tango 2 866866866866M 2.692.692.692.69 1.12 9.09 0.57 3.99 4.07
   w/o Strategy 2 & 3 866866866866M 2.642.642.642.64 1.131.131.131.13 8.068.068.068.06 0.540.540.540.54 -- --
   w/o Strategy 1 866M866𝑀866M866 italic_M 2.472.472.472.47 1.131.131.131.13 8.588.588.588.58 0.560.560.560.56 -- --
   w/o Strategy 2 866866866866M 2.282.282.282.28 1.12 8.388.388.388.38 0.550.550.550.55 -- --
   w/o Strategy 3 866866866866M 2.462.462.462.46 1.131.131.131.13 8.638.638.638.63 0.560.560.560.56 -- --

5.3. Evaluation Metrics

Objective Metrics

We evaluated the text-to-audio generation using the standard Frechet Audio Distance (FAD), KL divergence, Inception score (IS), and CLAP score (Liu et al., 2023a). FAD is adapted from Frechet Inception Distance (FID) and measures the distribution-level gap between generated and reference audio samples. KL divergence is an instance-level reference-dependent metric that measures the divergence between the acoustic event posteriors of the ground truth and the generated audio sample. FAD and KL are computed using PANN, an audio-event tagger. IS evaluates the specificity and coverage of a set of samples, not needing reference audios. IS is inversely proportional to the entropy of the instance posteriors and directly proportional to the entropy of the marginal posteriors. CLAP score is defined as the cosine similarity between the CLAP encodings of an audio and its textual description. We borrowed the AudioLDM evaluation toolkit (Liu et al., 2023a) for the computation of FAD, IS, and KL scores.

Subjective Metrics

Our subjective assessment examines two key aspects of the generated audio: overall audio quality (OVL) and relevance to the text input (REL), mirroring the approach outlined in the previous works, such as, (Ghosal et al., 2023a; Vyas et al., 2023). The OVL metric primarily gauges the general sound quality, clarity, and naturalness irrespective of its alignment with the input prompt. Conversely, the REL metric assesses how well the generated audio corresponds to the given text input. Annotators were tasked with rating each audio sample on a scale from 1 (least) to 5 (highest) for both OVL and REL. This evaluation was conducted on a subset of 50 randomly-selected prompts from the AudioCaps test set, with each sample being independently reviewed by at least four annotators. Please refer to the supplementary material for more details on the evaluation instructions and evaluators.

Table 4. Objective evaluation results for audio generation in the presence of multiple concepts or a single concept in the text prompt in the AudioCaps test set.
Model Multiple Events/Concepts Single Event/Concept
Objective Subjective Objective Subjective
FAD \downarrow KL \downarrow IS \uparrow CLAP \uparrow OVL \uparrow REL \uparrow FAD \downarrow KL \downarrow IS \uparrow CLAP \uparrow OVL \uparrow REL \uparrow
AudioLDM 2-Full 2.03 1.64 7.88 0.43 -- -- 7.93 1.24 4.50 0.47 -- --
AudioLDM 2-Full-Large 2.33 1.58 8.14 0.44 3.54 3.16 5.82 1.09 4.60 0.49 3.65 3.41
Tango-full-FT 2.69 1.16 7.85 0.54 3.83 3.80 7.52 1.01 4.87 0.51 3.67 3.49
Tango 2 2.60 1.11 8.98 0.57 3.99 4.07 5.48 1.00 4.95 0.52 3.95 4.10
Table 5. Objective evaluation results for audio generation in the presence of temporal events or non-temporal events in the text prompt in the AudioCaps test set.
Model Temporal Events Non Temporal Events
Objective Subjective Objective Subjective
FAD \downarrow KL \downarrow IS \uparrow CLAP \uparrow OVL \uparrow REL \uparrow FAD \downarrow KL \downarrow IS \uparrow CLAP \uparrow OVL \uparrow REL \uparrow
AudioLDM 2-Full 1.95 1.71 6.37 0.41 -- -- 2.38 1.56 7.38 0.44 -- --
AudioLDM 2-Full-Large 2.39 1.65 6.10 0.42 3.35 2.82 2.68 1.46 8.12 0.46 3.79 3.62
Tango-full-FT 2.55 1.16 5.82 0.55 3.83 3.67 3.04 1.15 7.70 0.53 3.78 3.88
Tango 2 3.29 1.07 6.88 0.58 3.92 3.99 2.84 1.16 8.62 0.55 4.05 4.16

5.4. Main Results

We report the comparative evaluations of Tango 2 against the baselines Tango (Ghosal et al., 2023a) and AudioLDM2 (Liu et al., 2023c) in Table 3. For a fair comparison, we used exactly 200 inference steps in all our experiments. Tango and Tango 2 were evaluated with a classifier-free guidance scale of 3 while AudioLDM2 uses a default guidance scale of 3.5. We generate only one sample per text prompt.

Objective Evaluations

Tango 2 achieves notable improvements in objective metrics, with scores of 2.69 for FAD, 1.12 for KL, 9.09 for IS, and 0.57 for CLAP. While FAD, KL, and IS assess general naturalness, diversity, and audio quality, CLAP measures the semantic alignment between the input prompt and the generated audio. As documented in Melechovsky et al. (2024), enhancing audio quality typically relies on improving the pre-training process of the backbone, either through architectural modifications or by leveraging larger or refined datasets. However, in our experiments, we observe enhanced audio quality in two out of three metrics, specifically KL and IS. Notably, Tango 2 also significantly outperforms various versions of AudioLDM and AudioLDM2 on these two metrics.

On the other hand, we note a substantial enhancement in the CLAP score. CLAP score is particularly crucial in our experimental setup as it directly measures the semantic alignment between the textual prompt and the generated audio. This outcome suggests that DPO-based fine-tuning on the preference data from Audio-alpaca yields superior audio generation to Tango and AudioLDM2.

Subjective Evaluations

In our subjective evaluation, Tango 2 achieves high ratings of 3.99 in OVL (overall quality) and 4.07 in REL (relevance), surpassing both Tango and AudioLDM2. This suggests that Tango 2 significantly benefits from preference modeling on Audio-alpaca. Interestingly, our subjective findings diverge from those reported by Melechovsky et al. (2024). In their study, the authors observed lower audio quality when Tango was fine-tuned on music data. However, in our experiments, the objective of preference modeling enhances both overall sound quality and the relevance of generated audio to the input prompts. Notably, in our experiments, AudioLDM2 performed the worst, with the scores of only 3.56 in OVL and 3.19 in REL, significantly lower than both Tango and Tango 2.

Table 6. GPT-4 prompt used to extract events or concepts from audio prompts.
You are to extract all the indivisible events in the given text, labeled as input. Imagine experiencing the events in the input as you are reading it and write down the indivisible events one by one. After writing your experience, list all the events in the sequence you observed them as a python list. Think step-by-step. Do not directly give the answer. Please refer to these following examples as refernce for input and output:
Example 1 -
Input: An aircraft engine runs and vibrates, metal spinning and grinding occur, and the engine accelerates and fades into the distance
Output: Firstly, an aircraft engine runs and vibrates. Then, I hear metal spinning and grinding. Then, the aircraft engine accelerates. Finally, the aircraft fades into the distance.
So, here is the list of events that I observed:
["aircraft engine runs", "aircraft engine vibrates", "metal spinning", "metal grinding", "aircraft engine acclerates", "aircraft fades into the distance"]
Example 2 -
Input: Bubbles gurgling and water spraying as a man speaks softly while crowd of people talk in the background
Output: Firstly, I hear bubble gurgling. Also, I hear water spraying. Simultaneously, a man is speaking softly. Also, a crowd of people are talking in the background.
So, here is the list of events that I observed:
["bubble gurgling", "water spraying", "a man is speaking softly", "crowd talking"]
Example 3 -
Input: A man talking then meowing and hissing
Output: Firstly, I hear a man talking. Subsequently, I hear meowing. I also hear hissing.
So, here is the list of events that I observed:
["a man talking", "meowing", "hissing"]
*** Examples end here
Now, given the input text below extract all the indivisible events one by one as explained above with examples. Also, remember to follow the exact format of the examples.
Input: {PROMPT}
Output:

Additionally, we categorize prompts based on the presence of multiple concepts or events, exemplified by phrases like A woman speaks while cooking. As underlined, this prompt contains two distinct events i.e., “sound of a woman speaking” and “sound of cooking”. Through manual scrutiny, we discovered that pinpointing prompts with such multi-concepts is challenging using basic parts-of-speech or named entity-based rules. Consequently, we task GPT-4 with extracting the various concepts or events from the prompts using in-context exemplars. The specific prompt is displayed in Table 6. To evaluate GPT-4’s performance on this task, we randomly selected 30 unique prompts and manually verified their annotations from GPT-4’s. No errors attributable to GPT-4 were found. In general, Tango 2 outperforms AudioLDM2 and Tango across most objective and subjective metrics, following Table 4. We proceed to visualize the CLAP scores of the models in Figure 3. This visualization illustrates that Tango 2 consistently outperforms the baselines as the number of events or concepts per prompt increases. In particular, specifically, Tango closely matches the performance of Tango 2 only when the textual prompt contains a single concept. However, the disparity between these two models widens as the complexity of the prompt increases with multiple concepts.

Refer to caption
Figure 3. CLAP score of the models vs the number of events or concepts in the textual prompt.

The supremacy of Tango 2 over the baselines in both of these cases can be strongly ascribed to DPO training the diffusion model the differences between generating a preferred and an undesirable audio output. Particularly, the undesirable audio outputs with missing concepts and wrong temporal orderings of events are penalized. Conversely, the preferred audio samples with the correct event orderings and presence are promoted by the noise-estimation terms in the DPO-diffusion loss (Eq. 17).

Ablations on Audio-alpaca.

We conducted an ablation study on Audio-alpaca to gauge the impact of different negative data construction strategies. As shown in Table 3, excluding the data samples from by strategies 2 and 3 notably diminishes the performance of Tango 2. This underscores the significance of event and temporal prompt perturbations.

The Effect of Filtering.

In our experiments, we noticed that filtering to create different Audio-alpaca can impact the performance (refer to Section 4.1). Figure 4 depicts the impact of this filtering process. We found setting Δ20.08subscriptΔ20.08\Delta_{2}\geq 0.08roman_Δ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT ≥ 0.08, and α20.6subscript𝛼20.6\alpha_{2}\geq 0.6italic_α start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT ≥ 0.6 gives the best results.

Refer to caption
Figure 4. The impact of filtering Audio-alpaca on performance observed through Δ2subscriptΔ2\Delta_{2}roman_Δ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT, and α2subscript𝛼2\alpha_{2}italic_α start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT. The CLAP score of the winning audio must be at least α2subscript𝛼2\alpha_{2}italic_α start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT and Δ2subscriptΔ2\Delta_{2}roman_Δ start_POSTSUBSCRIPT 2 end_POSTSUBSCRIPT represents the difference in CLAP scores between the winning audio xwsuperscript𝑥𝑤x^{w}italic_x start_POSTSUPERSCRIPT italic_w end_POSTSUPERSCRIPT and the losing audio xlsuperscript𝑥𝑙x^{l}italic_x start_POSTSUPERSCRIPT italic_l end_POSTSUPERSCRIPT given a prompt τ𝜏\tauitalic_τ.

6. Conclusion

In this work, we propose aligning text-to-audio generative models through direct preference optimization. To the best of our knowledge, this represents the first attempt to advance text-to-audio generation through preference optimization. We achieve this by automatically generating a preference dataset using a combination of Large Language Models (LLMs) and adversarial filtering. Our preference dataset, Audio-alpaca, comprises diverse audio descriptions (prompts) paired with their respective preferred (winner) and undesirable (loser) audios. The preferred audios are expected to accurately reflect their corresponding textual descriptions, while the undesirable audios exhibit flaws such as missing concepts, incorrect temporal order, or high noise levels. To generate undesirable audios, we perturb the descriptions by removing or rearranging certain concepts and feeding them to Tango. Additionally, we employ adversarial filtering, generating multiple audios from the original prompt and selecting those with CLAP scores below a specified threshold. Subsequently, we align a diffusion-based text-to-audio model, Tango, on Audio-alpaca using DPO-diffusion loss. Our results demonstrate significant performance leap over the previous models, both in terms of objective and subjective metrics. We anticipate that our dataset, Audio-alpaca, and the proposed model, Tango 2, will pave the way for further advancements in alignment techniques for text-to-audio generation.

References

  • (1)
  • Betker et al. ([n. d.]) James Betker, Gabriel Goh, Li Jing, † TimBrooks, Jianfeng Wang, Linjie Li, † LongOuyang, † JuntangZhuang, † JoyceLee, † YufeiGuo, † WesamManassra, † PrafullaDhariwal, † CaseyChu, † YunxinJiao, and Aditya Ramesh. [n. d.]. Improving Image Generation with Better Captions. https://api.semanticscholar.org/CorpusID:264403242
  • Borsos et al. (2023) Zalán Borsos, Raphaël Marinier, Damien Vincent, Eugene Kharitonov, Olivier Pietquin, Matt Sharifi, Dominik Roblek, Olivier Teboul, David Grangier, Marco Tagliasacchi, et al. 2023. Audiolm: a language modeling approach to audio generation. IEEE/ACM Transactions on Audio, Speech, and Language Processing (2023).
  • Chung et al. (2022) Hyung Won Chung, Le Hou, Shayne Longpre, Barret Zoph, Yi Tay, William Fedus, Eric Li, Xuezhi Wang, Mostafa Dehghani, Siddhartha Brahma, Albert Webson, Shixiang Shane Gu, Zhuyun Dai, Mirac Suzgun, Xinyun Chen, Aakanksha Chowdhery, Sharan Narang, Gaurav Mishra, Adams Yu, Vincent Zhao, Yanping Huang, Andrew Dai, Hongkun Yu, Slav Petrov, Ed H. Chi, Jeff Dean, Jacob Devlin, Adam Roberts, Denny Zhou, Quoc V. Le, and Jason Wei. 2022. Scaling Instruction-Finetuned Language Models. https://doi.org/10.48550/ARXIV.2210.11416
  • Chung et al. (2021) Yu-An Chung, Yu Zhang, Wei Han, Chung-Cheng Chiu, James Qin, Ruoming Pang, and Yonghui Wu. 2021. w2v-BERT: Combining Contrastive Learning and Masked Language Modeling for Self-Supervised Speech Pre-Training. In IEEE Automatic Speech Recognition and Understanding Workshop, ASRU 2021, Cartagena, Colombia, December 13-17, 2021. IEEE, 244–250. https://doi.org/10.1109/ASRU51503.2021.9688253
  • Ghosal et al. (2023a) Deepanway Ghosal, Navonil Majumder, Ambuj Mehrish, and Soujanya Poria. 2023a. Text-to-audio generation using instruction-tuned llm and latent diffusion model. arXiv preprint arXiv:2304.13731 (2023).
  • Ghosal et al. (2023b) Deepanway Ghosal, Navonil Majumder, Ambuj Mehrish, and Soujanya Poria. 2023b. Text-to-Audio Generation using Instruction Tuned LLM and Latent Diffusion Model. arXiv preprint arXiv:2304.13731 (2023).
  • Girdhar et al. (2023) Rohit Girdhar, Alaaeldin El-Nouby, Zhuang Liu, Mannat Singh, Kalyan Vasudev Alwala, Armand Joulin, and Ishan Misra. 2023. ImageBind: One Embedding Space To Bind Them All. In CVPR.
  • Hang et al. (2023) Tiankai Hang, Shuyang Gu, Chen Li, Jianmin Bao, Dong Chen, Han Hu, Xin Geng, and Baining Guo. 2023. Efficient Diffusion Training via Min-SNR Weighting Strategy. arXiv:2303.09556 [cs.CV]
  • Huang et al. (2023) Rongjie Huang, Jiawei Huang, Dongchao Yang, Yi Ren, Luping Liu, Mingze Li, Zhenhui Ye, Jinglin Liu, Xiang Yin, and Zhou Zhao. 2023. Make-an-audio: Text-to-audio generation with prompt-enhanced diffusion models. arXiv preprint arXiv:2301.12661 (2023).
  • Isola et al. (2016) Phillip Isola, Jun-Yan Zhu, Tinghui Zhou, and Alexei A. Efros. 2016. Image-to-Image Translation with Conditional Adversarial Networks. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (2016), 5967–5976.
  • Kingma and Welling (2013) Diederik P. Kingma and Max Welling. 2013. Auto-Encoding Variational Bayes. CoRR abs/1312.6114 (2013).
  • Kong et al. (2020) Jungil Kong, Jaehyeon Kim, and Jaekyoung Bae. 2020. Hifi-gan: Generative adversarial networks for efficient and high fidelity speech synthesis. Advances in Neural Information Processing Systems 33 (2020), 17022–17033.
  • Kong et al. (2021) Qiuqiang Kong, Yin Cao, Haohe Liu, Keunwoo Choi, and Yuxuan Wang. 2021. Decoupling Magnitude and Phase Estimation with Deep ResUNet for Music Source Separation. In International Society for Music Information Retrieval Conference.
  • Kreuk et al. (2022) Felix Kreuk, Gabriel Synnaeve, Adam Polyak, Uriel Singer, Alexandre Défossez, Jade Copet, Devi Parikh, Yaniv Taigman, and Yossi Adi. 2022. Audiogen: Textually guided audio generation. arXiv preprint arXiv:2209.15352 (2022).
  • Li et al. (2023) Zehan Li, Xin Zhang, Yanzhao Zhang, Dingkun Long, Pengjun Xie, and Meishan Zhang. 2023. Towards general text embeddings with multi-stage contrastive learning. arXiv preprint arXiv:2308.03281 (2023).
  • Liao et al. (2024) Huan Liao, Haonan Han, Kai Yang, Tianjiao Du, Rui Yang, Zunnan Xu, Qinmei Xu, Jingquan Liu, Jiasheng Lu, and Xiu Li. 2024. BATON: Aligning Text-to-Audio Model with Human Preference Feedback. arXiv:2402.00744 [cs.SD]
  • Liu et al. (2023a) Haohe Liu, Zehua Chen, Yi Yuan, Xinhao Mei, Xubo Liu, Danilo Mandic, Wenwu Wang, and Mark D Plumbley. 2023a. Audioldm: Text-to-audio generation with latent diffusion models. arXiv preprint arXiv:2301.12503 (2023).
  • Liu et al. (2023b) Haohe Liu, Zehua Chen, Yi Yuan, Xinhao Mei, Xubo Liu, Danilo P. Mandic, Wenwu Wang, and Mark D . Plumbley. 2023b. AudioLDM: Text-to-Audio Generation with Latent Diffusion Models. ArXiv abs/2301.12503 (2023).
  • Liu et al. (2023c) Haohe Liu, Qiao Tian, Yi Yuan, Xubo Liu, Xinhao Mei, Qiuqiang Kong, Yuping Wang, Wenwu Wang, Yuxuan Wang, and Mark D Plumbley. 2023c. AudioLDM 2: Learning holistic audio generation with self-supervised pretraining. arXiv preprint arXiv:2308.05734 (2023).
  • Loshchilov and Hutter (2017) Ilya Loshchilov and Frank Hutter. 2017. Decoupled weight decay regularization. arXiv preprint arXiv:1711.05101 (2017).
  • Melechovsky et al. (2024) Jan Melechovsky, Zixun Guo, Deepanway Ghosal, Navonil Majumder, Dorien Herremans, and Soujanya Poria. 2024. Mustango: Toward Controllable Text-to-Music Generation. arXiv:2311.08355 [eess.AS]
  • OpenAI (2023a) OpenAI. 2023a. DALL·E 2. https://openai.com/dall-e-2
  • OpenAI (2023b) OpenAI. 2023b. GPT-4. https://openai.com/gpt-4
  • OpenAI (2023c) OpenAI. 2023c. Introducing ChatGPT. https://openai.com/blog/chatgpt
  • Radford et al. (2019) Alec Radford, Jeff Wu, Rewon Child, David Luan, Dario Amodei, and Ilya Sutskever. 2019. Language Models are Unsupervised Multitask Learners. (2019).
  • Rafailov et al. (2023) Rafael Rafailov, Archit Sharma, Eric Mitchell, Stefano Ermon, Christopher D. Manning, and Chelsea Finn. 2023. Direct Preference Optimization: Your Language Model is Secretly a Reward Model. arXiv:2305.18290 [cs.LG]
  • Rombach et al. (2022) Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, and Björn Ommer. 2022. High-resolution image synthesis with latent diffusion models. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 10684–10695.
  • Ronneberger et al. (2015) Olaf Ronneberger, Philipp Fischer, and Thomas Brox. 2015. U-Net: Convolutional Networks for Biomedical Image Segmentation. In Medical Image Computing and Computer-Assisted Intervention – MICCAI 2015, Nassir Navab, Joachim Hornegger, William M. Wells, and Alejandro F. Frangi (Eds.). Springer International Publishing, Cham, 234–241.
  • Song et al. (2020) Jiaming Song, Chenlin Meng, and Stefano Ermon. 2020. Denoising Diffusion Implicit Models. ArXiv abs/2010.02502 (2020).
  • Tokozume et al. (2017) Yuji Tokozume, Yoshitaka Ushiku, and Tatsuya Harada. 2017. Learning from Between-class Examples for Deep Sound Recognition. CoRR abs/1711.10282 (2017). arXiv:1711.10282 http://arxiv.org/abs/1711.10282
  • Vyas et al. (2023) Apoorv Vyas, Bowen Shi, Matthew Le, Andros Tjandra, Yi-Chiao Wu, Baishan Guo, Jiemin Zhang, Xinyue Zhang, Robert Adkins, William Ngan, Jeff Wang, Ivan Cruz, Bapi Akula, Akinniyi Akinyemi, Brian Ellis, Rashel Moritz, Yael Yungster, Alice Rakotoarison, Liang Tan, Chris Summers, Carleigh Wood, Joshua Lane, Mary Williamson, and Wei-Ning Hsu. 2023. Audiobox: Unified Audio Generation with Natural Language Prompts. arXiv:2312.15821 [cs.SD]
  • Wallace et al. (2023) Bram Wallace, Meihua Dang, Rafael Rafailov, Linqi Zhou, Aaron Lou, Senthil Purushwalkam, Stefano Ermon, Caiming Xiong, Shafiq Joty, and Nikhil Naik. 2023. Diffusion Model Alignment Using Direct Preference Optimization. arXiv:2311.12908 [cs.CV]
  • Wu et al. (2023) Yusong Wu, Ke Chen, Tianyu Zhang, Yuchen Hui, Taylor Berg-Kirkpatrick, and Shlomo Dubnov. 2023. Large-scale contrastive language-audio pretraining with feature fusion and keyword-to-caption augmentation. In ICASSP 2023-2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 1–5.
  • Zeghidour et al. (2022) Neil Zeghidour, Alejandro Luebs, Ahmed Omran, Jan Skoglund, and Marco Tagliasacchi. 2022. SoundStream: An End-to-End Neural Audio Codec. IEEE ACM Trans. Audio Speech Lang. Process. 30 (2022), 495–507. https://doi.org/10.1109/TASLP.2021.3129994