---
title: 'DiffRhythm+: Controllable Song Generation'
url: https://www.emergentmind.com/topics/diffrhythm-5ced9ef4-5e12-4c9b-87b4-745cbefc5d42
type: topic
---

# DiffRhythm+: Controllable Song Generation

DiffRhythm+ is an enhanced diffusion-based framework for controllable and flexible full-length song generation. It was introduced to address major challenges in full-length song synthesis, including data imbalance, insufficient controllability, and inconsistent musical quality. Relative to DiffRhythm, a pioneering diffusion-based model for generating full-length songs with expressive vocals and accompaniment, DiffRhythm+ combines a substantially expanded and balanced training dataset, a multi-modal style conditioning strategy based on descriptive text and reference audio, and direct performance optimization aligned with user preferences [2507.12890].

## 1. Research context and design goals

Full-length song generation differs from short audio continuation or instrumental music generation because it must jointly maintain lyric intelligibility, vocal-accompaniment coordination, stylistic consistency, and long-range structural coherence. The motivating claim behind DiffRhythm+ is that prior systems still exhibited noticeable quality disparities and restricted creative flexibility, especially when training data were imbalanced across languages and when style control was limited to narrower conditioning channels [2507.12890].

The immediate precursor was DiffRhythm, which trained on approximately \(50\mathrm{k}\) hours of music in a \(3{:}6{:}1\) Chinese:English:instrumental ratio. That imbalance led to frequent lyric repetition and omission, especially on Chinese tracks. DiffRhythm+ was therefore framed not merely as a larger model, but as a reworked full-stack system in which dataset composition, conditioning design, and post-pretraining optimization were all modified together. The resulting emphasis was on naturalness, arrangement complexity, listener satisfaction, and flexible user control over musical style [2507.12890].

A later overview characterizes DiffRhythm+ as, to its knowledge, the first system to integrate direct preference optimization into a continuous diffusion-based music generator. That secondary framing is important because it places DiffRhythm+ at the intersection of long-form music generation and preference alignment, rather than treating it as only a scaling exercise [2511.15038].

## 2. Latent architecture and flow-matching formulation

DiffRhythm+ consists of two modules: a VAE encoder/decoder that maps raw waveforms \(x_0\) to and from a lower-dimensional latent \(z_0\), and a conditional flow-matching backbone based on a diffusion transformer (DiT) that models the distribution \(p_1(z_0)\) of clean musical latents by non-autoregressively transforming simple “noise” latents \(p_0\) into music latents \(p_1\) [2507.12890].

In practice, the system borrows the same VAE architecture and pre-trained weights from DiffRhythm, with approximately \(1.1\)B total parameters. The forward “diffusion” or flow-matching process is defined by coupling
\[
y^- \sim p_0,\qquad y^+ \sim p_1
\]
and mixing them along a straight path,
\[
y_t = (1-t)\cdot y^- + t\cdot y^+,\qquad u_t(y|z)=\frac{d}{dt}y_t = y^+-y^-.
\]
The model then learns a vector field \(v_\theta(t,y,c)\) so that integrating
\[
\frac{dx}{dt}=v_\theta
\]
drives \(p_0 \rightarrow p_1\). Given multi-modal condition \(c\), the training objective is the conditional flow-matching loss
\[
L_{\mathrm{CFM}}(\theta)=\mathbb{E}_{t\sim \mathrm{Uniform}[0,1],\,z\sim q(z),\,y\sim p_t(y|z)}
\left\|v_\theta(t,y,c)-u_t(y|z)\right\|^2.
\]
At inference time, the model solves
\[
\frac{dy}{dt}=v_\theta(t,y,c),\qquad y(0)=y^- \sim p_0
\]
from \(t=0\) to \(1\) with an Euler solver in \(32\) steps, and the resulting \(y(1)\) is passed through the VAE decoder to produce a full-length waveform [2507.12890].

The DiT backbone uses \(16\) LLaMA-style transformer decoder layers with hidden size \(2048\) and \(32\) heads. Lyrics, style, timestep, and noisy latent all enter as conditioning tokens concatenated along the channel dimension. Classifier-free guidance with scale \(4\) is applied by stochastically dropping out lyrics or style embeddings, with \(20\%\) dropout each during training. This architecture makes controllability an explicit part of the denoising dynamics rather than a post hoc reranking mechanism [2507.12890].

## 3. Dataset expansion, balancing, and supervised refinement

A central intervention in DiffRhythm+ is dataset reconstruction. The system collected \(300\mathrm{k}\) hours of raw recordings and filtered them via a rule-based pipeline to approximately \(120\mathrm{k}\) hours balanced \(2{:}2{:}1\) in Chinese:English:instrumental. A further audio-quality filtering stage using Audiobox and SongEval yielded a \(25\mathrm{k}\)-hour high-quality subset for supervised fine-tuning (SFT) [2507.12890].

The paper attributes two specific effects to this restructuring. First, balanced Chinese and English reduces language-specific artifacts such as repeated or missing phonemes. Second, larger scale fosters richer arrangements, more complex instrumentation, and fewer collapsed sections in long sequences. These claims are not presented as generic scaling laws; they are tied to the observed failure modes of the original DiffRhythm dataset composition [2507.12890].

The ablation results reinforce that the balance decision mattered independently of raw scale. In Table 4, \(120\mathrm{k}\) hours at \(1{:}1\) Chinese:English outperforms \(60\mathrm{k}\) hours and \(120\mathrm{k}\) hours at \(1{:}2\). This suggests that multilingual balance was treated as a structural variable for lyric-conditioned song generation, not simply as a corpus hygiene measure [2507.12890].

## 4. Multi-modal style conditioning and controllability

DiffRhythm+ leverages MuLan, and specifically its MuQ-MuLan variant, as a cross-modal style extractor. MuLan maps either a reference audio clip to a style embedding \(s_{\mathrm{audio}}\in\mathbb{R}^d\) or a text description prompt to a style embedding \(s_{\mathrm{text}}\in\mathbb{R}^d\), with both placed in the same shared latent space. This shared embedding space is the basis for the system’s claim of precise style specification through both descriptive text and reference audio [2507.12890].

Lyrics are first converted to phonemes via a G2P tokenizer and then embedded into \(l\in\mathbb{R}^d\). The timestep \(t\) is embedded through a standard sinusoidal or learned embedding \(\tau(t)\). At each diffusion step, the conditioning vector is formed as
\[
c=\mathrm{concat}[\,z_n,\,l,\,s,\,\tau(t)\,],
\]
which is linearly projected and summed with the DiT’s key/value and feed-forward inputs. During training, the style embedding is randomly replaced with zero with \(20\%\) chance and the lyric embedding is randomly replaced with zero with \(20\%\) chance. At inference, the model blends the predicted vector fields from the full condition and from conditions with style or lyric dropped in order to sharpen stylization [2507.12890].

The style-conditioning ablation is also specific. Mixed MuLan, defined as training on both audio and text prompts, yields the best generalization to text prompts. A plausible implication is that DiffRhythm+ treats text-prompt and reference-audio control not as disjoint modes, but as two access paths into a common style geometry [2507.12890].

## 5. Preference-driven performance optimization

Beyond reconstruction loss, DiffRhythm+ introduces Direct Preference Optimization for diffusion, described in the paper as aligning outputs to human aesthetic metrics. The reward model \(r_\phi(\tau,x_0)\) is defined by two off-the-shelf evaluators: SongEval, on a \(1\)-to-\(5\) scale for full songs, and Audiobox, mapped to \(1\)-to-\(5\) for instrumentals. From each batch, the system selects a “winner” \(x_0^w\) and “loser” \(x_0^l\) whose score gap is at least \(0.4\) and whose winner’s score is at least \(3\), forming \((\tau,x_0^w,x_0^l)\) preference tuples [2507.12890].

The diffusion-DPO objective is derived from a KL-constrained reward maximization over the entire diffusion path and is implemented as an \(L_2\)-based preference loss between noise predictions:
\[
L_{\mathrm{DPO\text{-}Diff}}
= -\mathbb{E}_{n,\,\epsilon^w,\,\epsilon^l}
\log \sigma\!\left[
-\beta\cdot N\cdot \omega(\lambda_n)\cdot
\left(
\|\epsilon_n^w-\hat{\epsilon}_\theta^{(n)}(x_n^w,\tau)\|^2
-\|\epsilon_n^w-\hat{\epsilon}_{\mathrm{ref}}^{(n)}(x_n^w,\tau)\|^2
-\left(
\|\epsilon_n^l-\hat{\epsilon}_\theta^{(n)}(x_n^l,\tau)\|^2
-\|\epsilon_n^l-\hat{\epsilon}_{\mathrm{ref}}^{(n)}(x_n^l,\tau)\|^2
\right)
\right)
\right].
\]
Here, \(\hat{\epsilon}_\theta\) is the model’s noise estimate at step \(n\), \(\hat{\epsilon}_{\mathrm{ref}}\) is a frozen reference model’s estimate, \(\omega(\lambda_n)\) is a scheduling weight, \(\beta=2000\), \(N\) is the number of diffusion steps, and \(\sigma\) is the logistic sigmoid. DPO training runs for \(8\) epochs on the generated preference dataset, with per-GPU batch size \(8\) and EMA of weights with decay \(0.99\) [2507.12890].

The training-stage ablation gives the clearest summary of its contribution: Pre-training \(\rightarrow\) SFT \(\rightarrow\) DPO progressively boosts mean subjective score from \(2.80 \rightarrow 2.86 \rightarrow 3.19\). This suggests that preference optimization is not redundant with corpus filtering or supervised refinement, but instead changes the ranking of generated outputs in ways that are audible to listeners [2507.12890].

## 6. Empirical performance and ablation profile

The principal objective metrics reported for DiffRhythm+ compare the original DiffRhythm baseline with DiffRhythm+ in base and full variants [2507.12890].

| Metric | DiffRhythm | DiffRhythm+ (base / full) |
|---|---:|---:|
| KL\(\downarrow\) | .723 | .488 / .512 |
| FAD\(\downarrow\) | 2.113 | 1.835 / 1.872 |
| CLaMP 3\(\uparrow\) (style alignment) | .103 | .152 / .165 |
| PER\(\downarrow\) (phoneme error) | 17.47% | 14.85% / 14.96% |
| RTF\(\downarrow\) (real-time factor) | 0.034 | 0.036 / 0.039 |

On aesthetic metrics, DiffRhythm+ wins among baselines on AudioBox (CE, CU, PC, PQ), approaching or exceeding YuE in clarity and complexity. On SongEval (Coh, Mem, NVBP, CSS, OM), DiffRhythm+ is approximately equal to YuE and significantly above DiffRhythm. In subjective MOS evaluation, listeners \((n=30)\) rated intelligibility, musicality, and audio quality, and DiffRhythm+ (full) achieved median MOS around \(3.1\)–\(3.2\) versus DiffRhythm around \(2.2\)–\(2.4\). YuE remained slightly higher at around \(3.5\), but DiffRhythm+ ran more than \(250\times\) faster at inference [2507.12890].

The ablations identify three major levers. Data scale and balance matter: \(120\mathrm{k}\) hours at \(1{:}1\) Chinese:English outperforms \(60\mathrm{k}\) hours and \(120\mathrm{k}\) hours at \(1{:}2\). Style conditioning matters: Mixed MuLan yields the best generalization to text prompts. Preference stages matter: Pre-training, SFT, and DPO each contribute measurable gains in subjective quality [2507.12890].

## 7. Subsequent interpretations, related work, and points of ambiguity

DiffRhythm+ has been read by later work as a template for preference-aligned music generation. The perspective paper “Aligning Generative Music AI with Human Preferences: Methods and Challenges” cites DiffRhythm+ as a diffusion-based preference optimization framework and states that it is, to its knowledge, the first system to integrate direct preference optimization into a continuous diffusion-based music generator [2511.15038]. That same source also highlights challenges that remain relevant to the DiffRhythm+ line: scalability to long-form compositions, reliability of automated scorers, multi-objective trade-offs, inference cost, and preference drift [2511.15038].

There is, however, a source-level ambiguity in how the system is described. The primary DiffRhythm+ paper specifies a conditional flow-matching DiT backbone, whereas the later overview summarizes DiffRhythm+ using DDPM-style notation and LDM-style UNet terminology. This indicates a divergence in exposition across sources rather than a settled alternative canonical architecture [2507.12890].

A second source of confusion arises from later nomenclature. “DiffRhythm 2: Efficient and High Fidelity Song Generation via Block Flow Matching” is a distinct successor system, but its details note that it is sometimes referred to as “DiffRhythm+” in those experiments. DiffRhythm 2 introduces a semi-autoregressive architecture based on block flow matching, a \(5\) Hz music VAE, cross-pair preference optimization, and stochastic block representation alignment loss, and it reports improvements over DiffRhythm+ rather than being identical to it [2510.22950].

Finally, interpretability work on audio latent spaces has connected DiffRhythm-derived VAEs to sparse autoencoder analysis. Paek et al. analyze DiffRhythm’s latent space and sketch extensions to DiffRhythm+ involving additional probes and controls for rhythmic features, harmonic features, and instrument identity, as well as the insertion of SAE-derived control vectors as conditioning tokens in a cross-attention block. This suggests one concrete future direction for DiffRhythm+-style systems: richer controllability grounded in interpretable latent coordinates rather than only text or reference-audio style embeddings [2510.23802].

Source: https://www.emergentmind.com/topics/diffrhythm-5ced9ef4-5e12-4c9b-87b4-745cbefc5d42