TADA! Tuning Audio Diffusion Models through Activation Steering
Organizations: Warsaw University of Technology · IDEAS Research Institute
Abstract
Audio diffusion models can synthesize high-fidelity music from text, yet achieving fine-grained control over specific musical attributes remains challenging, as their internal mechanisms for representing high-level concepts are poorly understood. In this work, we use activation patching to demonstrate that recent audio diffusion architectures exhibit a semantic bottleneck, where a small, shared subset of consecutive attention layers controls distinct musical concepts, such as the presence of specific instruments, vocals, or genres. Building on this, we systematically evaluate a broad spectrum of steering paradigms, comparing activation steering against prompt-level, score-space, and weight-space interventions, analyzing the interaction between the steering mechanism and the intervention site. Our new benchmark, supported by an extensive user study, demonstrates that localized activation steering establishes a new state-of-the-art in audio concept modulation.
Figures & tables
| Method | AUC ( ) | Smoothness ( ) | Audio Quality ( ) | ||
|---|---|---|---|---|---|
| MuQ | CLAP | MuQ | CLAP | ||
| Prompt-level interventions | |||||
| PCI ( Görgün et al. [26] ) | |||||
| Text Embeddings ( Ezra et al. [17] ) | |||||
| Token Embeddings ( Baumann et al. [7] ) | |||||
| Score-space steering | |||||
| Method | AUC ( ) | Smoothness ( ) | Audio Quality ( ) | ||
|---|---|---|---|---|---|
| MuQ | CLAP | MuQ | CLAP | ||
| PCI ( Görgün et al. [26] ) | |||||
| (localization) | |||||
| Text Embeddings ( Ezra et al. [17] ) | |||||
| (localization) | |||||
| Token Embeddings ( Baumann et al. [7] ) | |||||
| Method | Steering Two Concepts | Steering Three Concepts | |||||||
|---|---|---|---|---|---|---|---|---|---|
| P + V | P + FV | A + FV | P + | J + | P + V + J | A + FV + M | P + V + | A + + | |
| AUSteer | +0.057 | +0.001 | +0.022 | +0.013 | +0.090 | +0.062 | +0.005 | +0.048 | +0.045 |
| CAA | +0.048 | +0.011 | +0.028 | +0.008 | +0.068 | +0.047 | +0.026 | +0.029 | +0.032 |
| AUSteer (loc.) | +0.099 | +0.018 | +0.029 | +0.039 | +0.086 | +0.088 | +0.021 | +0.069 | +0.051 |
| CAA (loc.) | +0.094 | +0.034 | +0.052 | +0.027 | +0.103 | +0.081 | +0.035 | +0.061 | +0.039 |
| SAE (loc.) | +0.114 | +0.027 | +0.041 | +0.034 | +0.110 | +0.100 | +0.018 | +0.081 | +0.063 |
Appendix figures & tables51 assets
Supplementary material from the paper’s appendix.
Appendix
| Concept | Original Prompt | Counterfactual Prompt |
|---|---|---|
| Vocal Gender | "The low quality recording features a ballad song that contains sustained strings, mellow piano melody, and soft female vocal singing over it. It sounds sad and soulful." | "The low quality recording features a ballad song that contains sustained strings, mellow piano melody, and soft male vocal singing over it. It sounds sad and soulful." |
| Tempo | "This is a country music piece. There is a fiddle playing the main melody. The acoustic guitar and electric guitar are playing gently. The song has a slow tempo. The atmosphere is sentimental." | "This is a country music piece. There is a fiddle playing the main melody. The acoustic guitar and electric guitar are playing gently. The song has a fast tempo. The atmosphere is sentimental." |
| Mood | "This is a Hindustani classical music piece. There is a harmonium playing the main tune. A bansuri joins in to play, supporting a melody. The rhythmic background consists of tabla percussion and electronic drums. The atmosphere is joyful ." | "This is a Hindustani classical music piece. There is a harmonium playing the main tune. A bansuri joins in to play, supporting a melody. The rhythmic background consists of tabla percussion and electronic drums. The atmosphere is sorrowful ." |
| Instrument | "This folk song features a male voice singing the main melody in an emotional mood. This is accompanied by an accordion playing fills in the background. A violin plays a droning melody." | "This folk song features a male voice singing the main melody in an emotional mood. This is accompanied by an accordion playing fills in the background. A trumpet plays a droning melody." |
| Genre | "The low quality recording features a reggae /dub song that consists of a flat male vocal singing over punchy 808 bass, punchy snare, shimmering hi hats and groovy piano chords. It sounds energetic, groovy and the recording is noisy and in mono." | "The low quality recording features a metal /dub song that consists of a flat male vocal singing over punchy 808 bass, punchy snare, shimmering hi hats and groovy piano chords. It sounds energetic, groovy and the recording is noisy and in mono." |
| Concept | Example Filter Keywords | Example Replacements |
|---|---|---|
| female | female, woman, girlish | female male woman man girlish boyish |
| male | male, man | man woman his her himself herself |
| slow | slow, slowed, slower, slowly, slow-paced, leisurely | slow fast slower faster slow-paced fast-paced |
| fast | fast, rapid, dynamic, quick, energetic, groovy, lively | fast slow quickly slowly lively slowly |
| happy | happy, joyful, upbeat, uplifting, cheerful | joyful sorrowful cheerful gloomy warm cold |
| sad | sad, depressing, dejected | sad happy lonely connected emotional lighthearted |
| Prompts / concept | Budget % | (top-1) | (top-2) | (top-3) | (top-5) |
|---|---|---|---|---|---|
| 4 | 2% | 0.54 | 0.82 | 0.66 | 0.45 |
| 12 | 5% | 0.65 | 0.98 | 0.81 | 0.56 |
| 17 | 8% | 0.67 | 0.99 | 0.90 | 0.63 |
| 23 | 10% | 0.60 | 0.99 | 0.92 | 0.67 |
| 34 | 15% | 0.65 | 1.00 | 0.97 | 0.74 |
| 46 | 20% | 0.69 | 1.00 | 0.99 | 0.80 |
| AudioLDM2 | Stable Audio Open | Ace-Step | |
| 0.05 | 45, 46, 49, 50, 51 | 5, 6, 12, 13, 14, 19 | 6, 7, 8 |
| 0.075 | 45, 46, 49, 50, 51 | 5, 12, 13, 14 | 6, 7, 8 |
| 0.10 | 45, 46, 51 | 12, 13, 14 | 7, 8 |
| 0.15 | 45, 51 | 12, 13 | 7, 8 |
| 0.20 | 45, 51 | 12, 13 | 7, 8 |
| 0.30 | 45, 51 | 13 | 8 |
| Concept | Positive Prompt | Negative Prompt |
|---|---|---|
| Piano | “{base}, with piano” | “{base}” |
| Violin | “{base}, with violin” | “{base}” |
| Guitar Type | “{base}, with acoustic guitar” | “{base}, with electric guitar” |
| Mood | “happy song, {base}” | “sad song, {base}” |
| Tempo | “fast song, {base}” | “slow song, {base}” |
| Vocal Gender | “{base}, with female vocal” | “{base}, with male vocal” |
| Concept | Alignment query |
|---|---|
| Piano | “a piano song” |
| Violin | “a song with violin” |
| Guitar Type | “a song with acoustic guitar” |
| Mood | “a cheerful track” |
| Tempo | “a fast track” |
| Vocal Gender | “female vocal singing” |
| Method | LPAPS ( ) | Harmony ( ) | Rhythm ( ) | Melody ( ) | Structure ( ) | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| MuQ | CLAP | MuQ | CLAP | MuQ | CLAP | MuQ | CLAP | MuQ | CLAP | |
| PCI | ||||||||||
| PCI (loc.) | ||||||||||
| Text Embeddings | ||||||||||
| Text Embeddings (loc.) | ||||||||||
| Token Embeddings | ||||||||||
| CAA | CAA (loc.) | AUSteer | AUSteer (loc.) | SAE (loc.) | |
|---|---|---|---|---|---|
| 0.0 | |||||
| 0.2 | |||||
| 0.4 | |||||
| 0.6 | |||||
| 0.8 | |||||
| 1.0 |
| Learning rate | Train guidance | Training iters | |||||||
| Value | 1e-4 | 5e-4 | 5e-3 | 3 | 5 | 7 | 500 | 1000 | 1500 |
| Avg AUC | 0.102 | 0.101 | 0.083 | 0.104 | 0.103 | 0.102 | 0.102 | 0.101 | 0.101 |
| Rank | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| 4 | ||||||||||
| 8 | ||||||||||
| 16 |
| Top- | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| All-layer steering | ||||||||||
| 256 | 0.049 | 0.034 | 0.035 | 0.034 | 0.038 | 0.024 | 0.052 | 0.087 | 0.097 | 0.050 |
| 1024 | 0.048 | 0.046 | 0.042 | 0.037 | 0.030 | 0.069 | 0.068 | 0.108 | 0.135 | 0.065 |
| 2048 | 0.054 | 0.039 | 0.048 | 0.039 | 0.022 | 0.082 | 0.075 | 0.118 | 0.143 | 0.069 |
| 4096 | 0.040 | 0.030 | 0.055 | 0.063 | 0.034 | 0.106 | 0.069 | 0.109 | 0.151 | 0.073 |
| 8192 | 0.024 | 0.028 | 0.046 | 0.065 | 0.029 | 0.124 | 0.051 | 0.105 | 0.115 | 0.065 |
| Method | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| add. (eq. 9 ) | ||||||||||
| mult. (eq. 8 ) |
| Layer | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. |
|---|---|---|---|---|---|---|---|---|---|
| 5 | 9.6 | 0.3 | 1.3 | 0.3 | 17.5 | 11.1 | 1.8 | 0.2 | 17.7 |
| 6 | 5.3 | 0.0 | 0.4 | 0.0 | 8.4 | 2.7 | 2.2 | 0.0 | 0.2 |
| 7 | 0.6 | 0.0 | 17.6 | 17.6 | 1.4 | 3.0 | 1.6 | 8.8 | 1.6 |
| 8 | 1.8 | 12.0 | 30.1 | 37.6 | 14.9 | 17.3 | 62.1 | 53.5 | 71.8 |
| 9 | 0.0 | 0.7 | 1.1 | 0.0 | 0.2 | 15.2 | 0.2 | 2.5 | 0.4 |
| 12 | 0.4 | 0.0 | 0.0 | 5.8 | 0.6 | 0.1 | 0.1 | 0.0 | 0.1 |
| Steered CFG passes | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| cond | ||||||||||
| cond + null |
| ACE-Step Layer | FVU ( ) | ( ) | ( ) |
|---|---|---|---|
| 0.216 | 0.37% | 19.4% | |
| 0.243 | 0.25% | 20.1% |
| Weighting | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| unit | ||||||||||
| TF | ||||||||||
| TF–IDF | ||||||||||
| TF, | ||||||||||
| TF–IDF, |
| Negative SV Construction | Piano | Mood | Tempo | Voc. gen. | Violin | Voc. style | Guitar | Rock | Electr. | Avg |
|---|---|---|---|---|---|---|---|---|---|---|
| negated positive dir. vector | ||||||||||
| TF–IDF on negative prompts |
| Method (loc.) | Casual listeners ( ) | Trained musicians ( ) | |
|---|---|---|---|
| PCI | |||
| Text Embeddings | |||
| Token Embeddins | |||
| FreeSliders | |||
| CAA | |||
| AUSteer |
| Concept | Layers | AUC (MuQ) | AUC (CLAP) | Smoothness (MuQ) | Smoothness (CLAP) |
|---|---|---|---|---|---|
| Piano | all | 0.331 | 0.175 | 0.081 | 0.098 |
| localized | 0.930 | 0.390 | 0.111 | 0.064 | |
| ablated | 0.099 | 0.099 | 0.176 | 0.171 | |
| Tempo | all | 0.236 | 0.008 | 0.125 | 0.222 |
| localized | 0.300 | 0.067 | 0.065 | 0.073 | |
| ablated | 0.201 | -0.007 | 0.135 | 0.605 |
| Concept | Layers | AUC (MuQ) | AUC (CLAP) | Smoothness (MuQ) | Smoothness (CLAP) |
|---|---|---|---|---|---|
| Piano | all | 0.792 | 0.228 | 0.093 | 0.105 |
| localized | 0.742 | 0.220 | 0.128 | 0.154 | |
| ablated | 0.383 | 0.094 | 0.086 | 0.129 | |
| Tempo | all | 0.268 | 0.108 | 0.110 | 0.100 |
| localized | 0.323 | 0.216 | 0.113 | 0.085 | |
| ablated | 0.183 | 0.031 | 0.096 | 0.202 |
| Concept | Layers | AUC (MuQ) | AUC (CLAP) | Smoothness (MuQ) | Smoothness (CLAP) |
|---|---|---|---|---|---|
| Piano | all | 0.060 | 0.026 | 0.038 | 0.063 |
| localized | 0.141 | 0.061 | 0.033 | 0.032 | |
| ablated | -0.008 | -0.010 | 0.488 | – | |
| Tempo | all | 0.095 | 0.022 | 0.032 | 0.075 |
| localized | 0.111 | 0.023 | 0.029 | 0.042 | |
| ablated | 0.059 | 0.018 | 0.034 | 0.076 |
| Method | AUC ( ) | Smoothness ( ) | Audio Quality ( ) | ||
|---|---|---|---|---|---|
| MuQ | CLAP | MuQ | CLAP | ||
| PCI ( Görgün et al. [26] ) | |||||
| + localized | |||||
| Text Embeddings ( Ezra et al. [17] ) | |||||
| + localized | |||||
| Method ( piano ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( tempo ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( mood ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( vocal gender ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( vocal style ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( violin ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( guitar type ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( jazz genre ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method ( classical genre ) | AUC MuQ ( ) | AUC CLAP ( ) | Smoothness MuQ ( ) | Smoothness CLAP ( ) | Avg Quality ( ) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| + | - | avg | + | - | avg | + | - | avg | + | - | avg | ||
| PCI | |||||||||||||
| PCI (loc.) | |||||||||||||
| Text Embeddings | |||||||||||||
| Text Embeddings (loc.) | |||||||||||||
| Token Embeddings | |||||||||||||
| Method | Steering Two Concepts | Steering Three Concepts | |||||||
|---|---|---|---|---|---|---|---|---|---|
| P + V | P + FV | A + FV | P + | J + | P + V + J | A + FV + M | P + V + | A + + | |
| AUC MuQ | |||||||||
| AUSteer | +0.057 | +0.001 | +0.022 | +0.013 | +0.090 | +0.062 | +0.005 | +0.048 | +0.045 |
| CAA | +0.048 | +0.011 | +0.028 | +0.008 | +0.068 | +0.047 | +0.026 | +0.029 | +0.032 |
| AUSteer | +0.099 | +0.018 | +0.029 | +0.039 | +0.086 | +0.088 | +0.021 | +0.069 | +0.051 |
| CAA | +0.094 | +0.034 | +0.052 | +0.027 | +0.103 | +0.081 | +0.035 | +0.061 | +0.039 |