mlsanityData to 7 Oct 2026

TADA! Tuning Audio Diffusion Models through Activation Steering

arXiv 2602.11910 ↗First posted 12 Feb 2026cs.SD cs.LGcited by 2

On the map: Diffusion models → Audio, speech and music

Abstract (arXiv)

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.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Audio, speech and music, one of 14 directions in Diffusion models. Builds on 63 of its 74 references that are in the map. First posted on arXiv 12 Feb 2026.

Builds on

63 of 74 references.

Closest ancestors · 9

Same direction, newest first

  1. 2025SAM Audio: Segment Anything in Audio19 Dec
  2. 2025Melodia: Training-Free Music Editing Guided by Attention Probing in Diffusion Models11 Nov
  3. 2025ACE-Step: A Step Towards Music Generation Foundation Model28 May
  4. 2025AudioMorphix: Training-free audio editing with diffusion probabilistic models21 May
  5. 2024AudioEditor: A Training-Free Diffusion-Based Audio Editing Framework19 Sep
  6. 2024Prompt-guided Precise Audio Editing with Diffusion Models11 May
  7. 2024Zero-Shot Unsupervised and Text-Based Audio Editing Using DDPM Inversion15 Feb
  8. 2024MusicMagus: Zero-Shot Text-to-Music Editing via Diffusion Models9 Feb
  9. 2023AudioLDM 2: Learning Holistic Audio Generation with Self-supervised Pretraining10 Aug

Other references · 54

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  4. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  5. 2023GPT-4 Technical Reportcited by 18,132
  6. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  7. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  8. 2022Scalable Diffusion Models with Transformerscited by 5,931
  9. 2022Flow Matching for Generative Modelingcited by 5,426
  10. 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
  11. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  12. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
38 more
  1. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
  2. 2022Large-scale Contrastive Language-Audio Pretraining with Feature Fusion and Keyword-to-Caption Augmentationcited by 707
  3. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
  4. 2021XLS-R: Self-supervised Cross-lingual Speech Representation Learning at Scalecited by 589
  5. 2017The Perception-Distortion Tradeoffcited by 582
  6. 2024Scaling and evaluating sparse autoencoderscited by 530
  7. 2023MusicLM: Generating Music From Textcited by 442
  8. 2025Persona Vectors: Monitoring and Controlling Character Traits in Language Modelscited by 303
  9. 2013Direct and Indirect Effectscited by 276
  10. 2014Understanding image representations by measuring their equivariance and equivalencecited by 240
  11. 2024Stable Audio Opencited by 192
  12. 2023Concept Sliders: LoRA Adaptors for Precise Control in Diffusion Modelscited by 148
  13. 2023UniMax: Fairer and more Effective Language Sampling for Large-Scale Multilingual Pretrainingcited by 147
  14. 2021Taming Visually Guided Sound Generationcited by 136
  15. 2025Meta Audiobox Aesthetics: Unified Automatic Quality Assessment for Speech, Music, and Soundcited by 129
  16. 2025Are Sparse Autoencoders Useful? A Case Study in Sparse Probingcited by 109
  17. 2025SAeUron: Interpretable Concept Unlearning in Diffusion Models with Sparse Autoencoderscited by 75
  18. 2025MuQ: Self-Supervised Music Representation Learning with Mel Residual Vector Quantizationcited by 66
  19. 2024Controlling Language and Diffusion Models by Transporting Activationscited by 59
  20. 2023Localizing and Editing Knowledge in Text-to-Image Generative Modelscited by 52
  21. 2024NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internalscited by 43
  22. 2024Continuous, Subject-Specific Attribute Control in T2I Models by Identifying Semantic Directionscited by 39
  23. 2024One-Step is Enough: Sparse Autoencoders for Text-to-Image Diffusion Modelscited by 30
  24. 2025SongGen: A Single Stage Auto-regressive Transformer for Text-to-Song Generationcited by 29
  25. 2024On Mechanistic Knowledge Localization in Text-to-Image Generative Modelscited by 27
  26. 2026ACE-Step 1.5: Pushing the Boundaries of Open-Source Music Generationcited by 22
  27. 2025CASteer: Cross-Attention Steering for Controllable Concept Erasurecited by 14
  28. 2024Do Music Generation Models Encode Music Theory?cited by 12
  29. 2025LinEAS: End-to-end Learning of Activation Steering with a Distributional Losscited by 12
  30. 2025Discovering and Steering Interpretable Concepts in Large Generative Music Modelscited by 11
  31. 2024SMITIN: Self-Monitored Inference-Time INtervention for Generative Music Transformerscited by 8
  32. 2026AudioSAE: Towards Understanding of Audio-Processing Models with Sparse AutoEncoderscited by 8
  33. 2025Activation Patching for Interpretable Steering in Music Generationcited by 7
  34. 2025Temporal Concept Dynamics in Diffusion Models via Prompt-Conditioned Interventionscited by 7
  35. 2025SteerMusic: Enhanced Musical Consistency for Zero-shot Text-guided and Personalized Music Editingcited by 6
  36. 2025Localizing Knowledge in Diffusion Transformerscited by 6
  37. 2026Fine-Grained Activation Steering: Steering Less, Achieving Morecited by 6
  38. 2025Precise Parameter Localization for Textual Generation in Diffusion Modelscited by 5

50 most cited of 54 All 54 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Pitch-class Steering for Diffusion-based Music Generation via Latent-space Probes
  2. 2024Improving Musical Accompaniment Co-creation via Diffusion Transformers
  3. 2026S2Accompanist: A Semantic-Aware and Structure-Guided Diffusion Model for Music Accompaniment Generation
  4. 2024Diff-A-Riff: Musical Accompaniment Co-creation via Latent Diffusion Models
  5. 2024Combining audio control and style transfer using latent diffusion
  6. 2024Multi-Track MusicLDM: Towards Versatile Music Generation with Latent Diffusion Model
  7. 2026SoundWeaver: Semantic Warm-Starting for Text-to-Audio Diffusion Serving
  8. 2024Audio Prompt Adapter: Unleashing Music Editing Abilities for Text-to-Music with Lightweight Finetuning