---
title: Autoregressive Diffusion DPO for Speech Models
url: https://www.emergentmind.com/topics/autoregressive-diffusion-direct-preference-optimization-ardm-dpo
type: topic
---

# Autoregressive Diffusion DPO for Speech Models

Searching arXiv for the specified papers and closely related work to ground the article.
Autoregressive Diffusion-Direct Preference Optimization (ARDM-DPO) is a post-training preference-alignment method for speech autoregressive diffusion models (ARDMs) that fine-tunes a pretrained ARDM toward preferred speech outputs using a DPO-style objective rather than explicit reward modeling or reinforcement learning. In the formulation introduced for zero-shot text-to-speech (TTS), the method is applied to DiTAR and is designed to improve two weak points of zero-shot speech generation—expressiveness and robustness on hard or long texts—while preserving speaker similarity and naturalness reasonably well [2509.18928].

## 1. Definition and problem setting

ARDM-DPO is defined for the setting in which a pretrained speech ARDM already generates intelligible speech, but its outputs remain suboptimal with respect to preference criteria such as prosodic expressiveness or robustness on difficult text. The intended adjustment is not full retraining but post-training alignment: the generation distribution is biased toward preferred outputs while remaining close to the base model distribution [2509.18928].

The model class of interest is an autoregressive diffusion model for speech. Unlike discrete-token TTS systems, ARDMs generate continuous latent speech tokens. Generation is autoregressive over token positions, but each token is denoised through a diffusion process rather than produced by direct next-token prediction. This is presented as attractive for speech because it preserves fine acoustic detail and avoids the complexity of discrete speech tokenization [2509.18928].

The preference-alignment problem is motivated by concrete failure modes of zero-shot speech generation. Emotion prompts may still produce monotone speech; hard long texts may trigger omissions or insertions; and naive attempts to increase expressiveness may degrade speaker similarity and naturalness. ARDM-DPO addresses this by adapting the DPO paradigm to the ARDM setting, using preference pairs rather than an explicit reward model [2509.18928].

## 2. ARDM generative structure and mathematical formulation

The sampling process is modeled as a Markov chain over states indexed by token index \(n\) and diffusion time \(t\). Each state contains the already denoised history \(x_{<n}^0\) and the current noisy token \(x_n^t\). For a fixed number of tokens \(N\) and diffusion steps \(T\), the ARDM denoises token \(n\) from \(t=T\) down to \(0\), then advances to token \(n+1\) by reinitializing noise [2509.18928].

The forward perturbation is specified as
\[
q(x_t \mid x_0) = \mathcal N(x_t; \alpha_t x_0, \sigma_t^2 I_d),
\]
with decreasing \(\alpha_t\) and increasing \(\sigma_t\). A DDPM-like sampling step is written as
\[
x_{t-1} = a_t x_t + b_t \nabla_{x_t} \log q(x_t) + c_t \epsilon, \qquad \epsilon \sim \mathcal N(0, I).
\]
For ARDMs, the conditional score is
\[
\nabla \log q(x_n^t \mid x_{<n}^0),
\]
so denoising proceeds one token at a time. The trajectory notation \(\mathbf x\) denotes the full ARDM sampling trajectory including all intermediate states [2509.18928].

Preferences are assumed over output pairs \((x_{1..N}^0, y_{1..N}^0)\), where \(x\) is preferred over \(y\), and are modeled with a Bradley–Terry likelihood:
\[
\log P(x_{1..N}^0 \succ y_{1..N}^0) = \log \sigma\!\left(r(x_{1..N}^0)-r(y_{1..N}^0)\right).
\]
The paper then formulates KL-regularized policy optimization over ARDM trajectories:
\[
\max_{\pi}\; \mathbb E_{\pi(\mathbf x)}[r(\mathbf x)] - \beta D_{\mathrm{KL}\!\left(\pi(\mathbf x)\,\|\,\mu(\mathbf x)\right),
\]
where \(\pi\) is the fine-tuned ARDM policy, \(\mu\) is the reference model, and \(\beta>0\) controls closeness to the reference. The reward-optimal policy takes the standard DPO form
\[
\pi_r(\mathbf x) = \frac{1}{Z_r}\mu(\mathbf x)\exp\!\left(\frac{1}{\beta}r(\mathbf x)\right),
\]
with
\[
r(\mathbf x)=\beta \log \frac{\pi_r(\mathbf x)}{\mu(\mathbf x)}+\beta\log Z_r.
\]
Marginalizing over intermediate ARDM states yields a log-ratio expression over output samples:
\[
r(x_{1..N}^0) = \beta \mathbb E_{\pi_r(\mathbf x\mid x_{1..N}^0)} \left[\log \frac{\pi_r(\mathbf x)}{\mu(\mathbf x)}\right] +\beta\log Z_r.
\]
This provides the theoretical basis for adapting DPO to autoregressive diffusion speech generation [2509.18928].

## 3. Preference objective and denoising-loss interpretation

At trajectory level, ARDM-DPO defines
\[
\ell_n^t(\mathbf x) := \log \frac{\pi(x_n^{t-1}\mid x_n^t, x_{<n}^0)} {\mu(x_n^{t-1}\mid x_n^t, x_{<n}^0)}.
\]
For a preference pair \((x,y)\), the DPO preference likelihood becomes
\[
\mathcal J(x_{1..N}^0, y_{1..N}^0) = \log \sigma\Big( T N \beta\; \mathbb E\big[\ell_n^t(\mathbf x)-\ell_n^t(\mathbf y)\big] \Big),
\]
where the expectation is over diffusion time \(t\), token index \(n\), and the corresponding ARDM transition distributions [2509.18928].

Because direct optimization is intractable, the expectations are moved outside the sigmoid using Jensen’s inequality, producing a lower bound \(\mathcal L\). The inner expectation of \(\ell_n^t\) can then be expressed as a difference of KL divergences:
\[
D_{\mathrm{KL}\!\left(q(x_n^{t-1}\mid x_n^t,x_n^0)\,\|\,\mu(x_n^{t-1}\mid x_n^t,x_{<n}^0)\right) - D_{\mathrm{KL}\!\left(q(x_n^{t-1}\mid x_n^t,x_n^0)\,\|\,\pi(x_n^{t-1}\mid x_n^t,x_{<n}^0)\right).
\]
Under the assumption that the ARDM is trained with a denoising objective, this becomes a difference of squared denoising errors:
\[
\omega_t\left( -\|v_\theta(x_n^t,x_{<n}^0)-x_n^0\|_2^2 +\|v_{\mathrm{ref}(x_n^t,x_{<n}^0)-x_n^0\|_2^2 \right).
\]
The resulting ARDM-DPO objective is
\[
\begin{aligned}
\mathcal L(x_{1..N}^0, y_{1..N}^0) :=\mathbb E_{\mathcal U(t)\,q(x_n^t\mid x_n^0)\,q(y_n^t\mid y_n^0)} \Bigg[ \log \sigma\Big( \beta\omega_t\mathbb E_{\mathcal U(n)}[ &-\|v_\theta(x_n^t, x_{<n}^0)-x_n^0\|_2^2 +\|v_{\mathrm{ref}(x_n^t, x_{<n}^0)-x_n^0\|_2^2 \\
&+\|v_\theta(y_n^t, y_{<n}^0)-y_n^0\|_2^2 -\|v_{\mathrm{ref}(y_n^t, y_{<n}^0)-y_n^0\|_2^2 ]\Big) \Bigg].
\end{aligned}
\]
Its stated interpretation is that, relative to the reference model, the preferred sample is trained to receive a better normalized denoising score than the dispreferred sample [2509.18928].

The empirical instantiation uses DiTAR, a diffusion transformer autoregressive model using \(v\)-prediction and continuous diffusion time \(t\in[0,1]\), where \(t=1\) is pure noise. Noisy tokens are generated as
\[
x_n^t = \alpha_t x_n^0 + \sigma_t x_n^1,
\]
with \(x_n^1\sim \mathcal N(0,I)\), and
\[
\dot x_n^t = \dot \alpha_t x_n^0 + \dot \sigma_t x_n^1.
\]
The specialized preference objective becomes
\[
\begin{aligned}
\mathcal L(x_{1..N_x}^0, y_{1..N_y}^0) :=\mathbb E_{t \sim \mathcal U(0,1),\, x_{1..N_x}^1,\, y_{1..N_y}^1} \Big[ \log \sigma \Big( & d^{-1}\beta \mathbb E_n\big[ \|v_{\mathrm{ref}(x_n^t, x_{<n}^0)-\dot x_n^t\|_2^2 -\|v_\theta(x_n^t, x_{<n}^0)-\dot x_n^t\|_2^2 \big] \\
-& d^{-1}\beta \mathbb E_n\big[ \|v_{\mathrm{ref}(y_n^t, y_{<n}^0)-\dot y_n^t\|_2^2 -\|v_\theta(y_n^t, y_{<n}^0)-\dot y_n^t\|_2^2 \big] \Big) \Big],
\end{aligned}
\]
where \(d\) is the token dimension and \(d^{-1}=1/256\) in the reported experiments. The time-dependent weight \(\omega_t\) is omitted following prior Diffusion-DPO practice [2509.18928].

## 4. Specialization to DiTAR and training pipeline

The reported experiments fine-tune a 0.4B-parameter DiTAR model pretrained on around 280,000 hours of Chinese and English speech. The architecture uses 24 Transformer blocks for the language-model component, 4 blocks for the diffusion head, hidden size 1024, and 16 attention heads [2509.18928].

The same sampler is used for data generation and evaluation: a 16-step DDPM sampler with a linear time schedule and LM Guidance with weight \(w=2\), described as similar to classifier-free guidance. DPO training is conducted on 32 A100 GPUs with local batch size 1 pair, gradient accumulation 32, and effective batch size 1024 pairs. The optimizer is AdamW with learning rate \(2\times 10^{-6}\), weight decay 0.01, and \(\beta_1=0.9,\ \beta_2=0.95\) [2509.18928].

A notable implementation feature is that the preference pairs are not derived from human preference labels directly. Instead, the method constructs preference data from automatic reward signals that differ by task. This places ARDM-DPO within a class of preference-optimization methods that preserve the pairwise-comparison interface of DPO while replacing direct human annotation with metric-defined winners and losers [2509.18928].

This suggests that, in the reported formulation, ARDM-DPO functions as a bridge between preference learning and metric-guided post-training for speech diffusion models. The paper does not describe this as explicit reward-model training; rather, the automatic metrics are used only to construct pairwise preference data [2509.18928].

## 5. Preference data construction and evaluation methodology

Two task-specific preference datasets are reported. For expressiveness, the data source is LibriTTS. For each prompt-text pair, the base model generates 32 candidates, scores them by F0 variance (F0V), and selects the highest-F0V sample as winner and the lowest-F0V sample as loser. This yields 256k preference pairs, approximately 1,000 hours. The purpose is to increase prosodic expressiveness and reduce monotonicity [2509.18928].

For long-text robustness, prompts come from DidiSpeech-2, described as 227h and 1500 speakers. The texts are long Chinese sentences with artificially introduced repetitive phrases or clauses. For each prompt-text pair, the base model generates 16 candidates and scores them using a phoneme-based CTC model. The reward proxy is negative log-likelihood per phoneme, with lower NLL indicating better text alignment and intelligibility. The winner is the sample with the lowest CTC NLL and the loser is the sample with the highest CTC NLL. This produces 430k preference pairs, about 3,500 hours. The CTC model itself is a small transformer with 6 transformer blocks, hidden size 1024, and 16 heads [2509.18928].

Evaluation combines objective and subjective protocols. Objective metrics are F0 variance for expressiveness; WER for English from Whisper-large-v3; CER for Chinese from Paraformer-zh; speaker similarity (SIM) using WavLM-TDCNN speaker embeddings; and token average KL divergence as a proxy for divergence from the reference model:
\[
d^{-1}\mathbb E_{\pi(\mathbf x), \mathcal U(n), \mathcal U(t), q(x_n^t\mid x_n^0)}
\left\|v_\theta(x_n^t, x_{<n}^0)-v_{\mathrm{ref}(x_n^t, x_{<n}^0)\right\|_2^2.
\]
This quantity is reported as a measure of how far the fine-tuned model moves from the base model [2509.18928].

Subjective evaluation uses pairwise listening tests with 20 listeners comparing base and DPO outputs on naturalness, speaker similarity, and expressiveness. For the long-text task, subjective comparison is reported on 40 test cases [2509.18928].

## 6. Empirical findings, baselines, and limitations

On the expressiveness task, the base DiTAR model has F0V 14.2, SIM 0.770, and WER 5.17. ARDM-DPO with \(\beta=200\) and 200 steps achieves F0V 29.2, SIM 0.765, WER 3.73, and KL 0.010. The paper describes this as the strongest result in the table and notes that it nearly doubles F0 variance with only a small speaker-similarity drop [2509.18928].

The expressiveness baselines are Best-of-16, Best-of-64, and RAFT. Best-of-16 obtains F0V 22.5; Best-of-64 obtains F0V 26.6; and RAFT iterations improve F0V only to about 20.1 while degrading SIM and WER more. The reported conclusion is that DPO outperforms rejection-sampling fine-tuning and best-of-\(K\) selection on this task. A grid search over \(\beta \in \{200, 400, 800\}\) indicates that larger \(\beta\) strengthens the KL constraint, leading to better SIM preservation but smaller F0V gains. Early stopping is recommended because training can become unstable and quality may degrade over longer runs. The paper also notes an observation that, although the DPO objective should reduce loss on winners and increase it on losers, in practice both losses can increase during training, similar to phenomena reported in LLM DPO [2509.18928].

Subjectively, listeners report that DPO slightly reduces naturalness and speaker similarity but significantly improves expressiveness on the expressiveness task. On the long-text task, the base model has NLL 0.55, SIM 0.711, and CER 8.37. ARDM-DPO with \(\beta=1600\) and 9000 steps achieves NLL 0.32, SIM 0.712, CER 6.32, and KL 0.009. The paper characterizes this as roughly a 25% CER reduction with essentially unchanged SIM, indicating strong prior preservation [2509.18928].

For long-text robustness, Best-of-8 based on CER gives CER 4.99 but worse NLL; Best-of-8 based on NLL gives NLL 0.27 but CER 6.79. The paper presents DPO as providing a more balanced improvement and as being better aligned with the training objective than ad hoc selection. In subjective evaluation on 40 test cases, the DPO model performs almost identically to the base model on naturalness and speaker similarity: naturalness 4.4% lose / 88.7% tie / 6.9% win, and speaker similarity 2.1% lose / 94.3% tie / 3.6% win. The main gain is therefore described as robustness and intelligibility rather than an obvious shift in perceived quality [2509.18928].

The principal contribution is stated as showing that DPO can be adapted cleanly to autoregressive diffusion speech models by expressing preference alignment in terms of diffusion denoising losses and reference-model ratios. The authors further emphasize that this is, to their knowledge, the first preference-alignment method tailored to ARDMs for TTS. The main limitations identified are training instability on the expressiveness task, strong dependence on the quality of metric-driven preference data construction, and open questions about why both preferred and dispreferred diffusion losses can increase during DPO [2509.18928].

## 7. Relation to autoregressive DPO theory

ARDM-DPO is specific to autoregressive diffusion speech models, whereas Autoregressive DPO (ADPO) revisits DPO for autoregressive language models by making the autoregressive assumption explicit before applying the Bradley–Terry model [2602.09533]. The two methods address different model classes and derive different training objectives, but both are motivated by aligning preference optimization more closely with autoregressive generation.

In ADPO, the central mathematical distinction from standard DPO is that the summation over token-level log-ratios is moved outside the log-sigmoid, yielding a sum of prefix-wise preference terms rather than a single response-level term. The paper presents this as a theoretically consistent refinement of standard DPO rather than a departure from it, and introduces the distinction between token length \(\mu\) and feedback length \(\mu'\) to characterize preference granularity [2602.09533].

A plausible implication is that ARDM-DPO and ADPO occupy analogous positions in different autoregressive generative settings. ARDM-DPO adapts preference optimization to a trajectory over token indices and diffusion times in continuous latent speech generation, whereas ADPO adapts it to prefix-wise discrete autoregressive sequence modeling. The commonality is the attempt to reconcile DPO-style preference alignment with the internal factorization structure of the generator, but the concrete losses, state spaces, and theoretical constructions remain distinct [2509.18928; 2602.09533].

Source: https://www.emergentmind.com/topics/autoregressive-diffusion-direct-preference-optimization-ardm-dpo