Lin-Xi Zhao

Self-Augmenting Retrieval for Diffusion Language Models

This work shows that intermediate states surface salient entities early in the denoising trajectory, making them a useful lookahead signal for retrieval-augmented generation (RAG), and demonstrates that grounding generation in retrieved evidence sharply reduces inter-token dependence, making RAG particularly well suite...