AIBA: Attention-based Instrument Band Alignment for Text-to-Audio Diffusion
On the map: Diffusion models → Audio, speech and music
Abstract (arXiv)
We present AIBA (Attention-In-Band Alignment), a lightweight, training-free pipeline to quantify where text-to-audio diffusion models attend on the time-frequency (T-F) plane. AIBA (i) hooks cross-attention at inference to record attention probabilities without modifying weights; (ii) projects them to fixed-size mel grids that are directly comparable to audio energy; and (iii) scores agreement with instrument-band ground truth via interpretable metrics (T-F IoU/AP, frequency-profile correlation, and a pointing game). On Slakh2100 with an AudioLDM2 backbone, AIBA reveals consistent instrument-dependent trends (e.g., bass favoring low bands) and achieves high precision with moderate recall.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Audio, speech and music, one of 14 directions in Diffusion models. Builds on 20 of its 23 references that are in the map. First posted on arXiv 25 Sep 2025.
Builds on
20 of 23 references.
Closest ancestors · 3
Same direction, newest first
- 2023AudioLDM 2: Learning Holistic Audio Generation with Self-supervised Pretraining10 Aug
- 2023Make-An-Audio: Text-To-Audio Generation with Prompt-Enhanced Diffusion Models30 Jan
- 2023AudioLDM: Text-to-Audio Generation with Latent Diffusion Models29 Jan
Other references · 17
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
- 2019Are Sixteen Heads Really Better than One?cited by 816
- 2020Quantifying Attention Flow in Transformerscited by 775
- 2019Attention is not not Explanationcited by 636
- 2023Simple and Controllable Music Generationcited by 488
5 more
- 2022AudioGen: Textually Guided Audio Generationcited by 295
- 2022What the DAAM: Interpreting Stable Diffusion Using Cross Attentioncited by 236
- 2021Efficient Training of Audio Transformers with Patchoutcited by 226
- 2019Cutting Music Source Separation Some Slakh: A Dataset to Study the Impact of Training Data Quality and Quantitycited by 86
- 2021Towards Interpretable and Transferable Speech Emotion Recognition: Latent Representation Based Analysis of Features, Methods and Corporacited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026TADA! Tuning Audio Diffusion Models through Activation Steering
- 2026SoundWeaver: Semantic Warm-Starting for Text-to-Audio Diffusion Serving
- 2026Improving Text-to-Music Generation with Human Preference Rewards
- 2026Inside the Latent Flow: Causal Deciphering of Attention Dynamics in Audio Separation Foundation Models
- 2024Auffusion: Leveraging the Power of Diffusion and Large Language Models for Text-to-Audio Generation
- 2025Melodia: Training-Free Music Editing Guided by Attention Probing in Diffusion Models
- 2025SAMUeL: Efficient Vocal-Conditioned Music Generation via Soft Alignment Attention and Latent Diffusion
- 2024Improving Musical Accompaniment Co-creation via Diffusion Transformers