AudioGenX: Explainability on Text-to-Audio Generative Models
Hyunju Kang, Geonhee Han, Yoonjae Jeong, Hogun Park
TL;DR
AudioGenX tackles the lack of transparency in text-to-audio generation by introducing a faithful, perturbation-based explainability framework that uses factual and counterfactual reasoning to quantify token-level importance in cross-attention. It defines soft masks over input tokens, optimizes them via a dual-objective loss $ extbf{L} = extbf{L}_{F} + extbf{L}_{CF} + oldsymbol{ ext{alpha}} extbf{L}_{1}( extbf{M}) + oldsymbol{ ext{beta}} extbf{L}_{2}( extbf{M})$, and provides both holistic and interval-focused explanations. Extensive experiments on AudioCaps with PaSST demonstrate that AudioGenX yields faithful explanations (low $Fid_{F}$ and $KL_{F}$) and strong counterfactual sensitivity (high $Fid_{CF}$ and $KL_{CF}$), while delivering compact masks and enabling practical audio editing insights. This work enhances transparency, trust, and debugging capabilities for TAG models, with implications for more controllable and auditable audio generation, though it notes hyperparameter sensitivity and dataset biases as limitations.
Abstract
Text-to-audio generation models (TAG) have achieved significant advances in generating audio conditioned on text descriptions. However, a critical challenge lies in the lack of transparency regarding how each textual input impacts the generated audio. To address this issue, we introduce AudioGenX, an Explainable AI (XAI) method that provides explanations for text-to-audio generation models by highlighting the importance of input tokens. AudioGenX optimizes an Explainer by leveraging factual and counterfactual objective functions to provide faithful explanations at the audio token level. This method offers a detailed and comprehensive understanding of the relationship between text inputs and audio outputs, enhancing both the explainability and trustworthiness of TAG models. Extensive experiments demonstrate the effectiveness of AudioGenX in producing faithful explanations, benchmarked against existing methods using novel evaluation metrics specifically designed for audio generation tasks.
