DLIG Provides Token Attribution for Diffusion Language Models

Diffusion Layer Integrated Gradients (DLIG) extends Integrated Gradients to track token attribution across arbitrary layers and denoising steps in diffusion language models. It maps how models commit to specific completions during the denoising trajectory for tasks like reasoning and infilling.