aiqbench analysis
DiffPDE introduces a discrete diffusion language model framework that optimizes code repair for Partial Differential Equation solvers by focusing regeneration on localized errors rather than using inefficient global autoregressive decoding. The approach further employs an iterative reinforcement learning scheme to handle coupled bugs, demonstrating superior accuracy and speed over standard models in experimental benchmarks. This work represents a significant methodological shift in applying generative AI techniques to specialized scientific computing tasks.
DiffPDE: Masked Diffusion Language Models as PDE Solver
arXiv:2608.30532v1 Announce Type: new Abstract: Existing approaches for synthesizing Partial Differential Equation (PDE) solvers predominantly rely on autoregressive models, yet their global left-to-right decoding incurs substantial redundancy when addressing inherently localized bugs. In this work, we challenge this inefficient paradigm and propose DiffPDE, a framework leveraging discrete diffusion language models for targeted code repair. By introducing a localized re-masking and infilling strategy, DiffPDE regenerates only erroneous regions while preserving correct context, naturally aligning generation with the sparse nature of PDE errors. Furthermore, to handle coupled bugs requiring sequential interventions, we present Iterative Debugging GRPO (ID-GRPO), a reinforcement learning scheme that enables multi-round debugging within single trajectories via intermediate rewards. Experiments on PDEBench show that DiffPDE achieves competitive accuracy, outperforms same-scale AR models, and significantly accelerates repair.
Abstract:Existing approaches for synthesizing Partial Differential Equation (PDE) solvers predominantly rely on autoregressive models, yet their global left-to-right decoding incurs substantial redundancy when addressing inherently localized bugs. In this work, we challenge this inefficient paradigm and propose DiffPDE, a framework leveraging discrete diffusion language models for targeted code repair. By introducing a localized re-masking and infilling strategy, DiffPDE regenerates only erroneous regions while preserving correct context, naturally aligning generation with the sparse nature of PDE errors. Furthermore, to handle coupled bugs requiring sequential interventions, we present Iterative Debugging GRPO (ID-GRPO), a reinforcement learning scheme that enables multi-round debugging within single trajectories via intermediate rewards. Experiments on PDEBench show that DiffPDE achieves competitive accuracy, outperforms same-scale AR models, and significantly accelerates repair.
Artificial Intelligence (cs.AI)
arXiv:2608.30532 [cs.AI]
(or arXiv:2608.30532v1 [cs.AI] for this version)
https://doi.org/10.48550/arXiv.2608.30532 Focus to learn more arXiv-issued DOI via DataCite (pending registration)
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.
Full text under the publisher's open licence · 215 words
Related coverage
Test-Time Scaling with Diffusion Language Models via Reward-Guided Stitching
arXiv · Oct 10, 2026
DCGC: Draft-Conditioned Global Correction for Complex Reasoning with Masked Diffusion Models
arXiv · Aug 27, 2026
What Does a Discrete Diffusion Model Learn? One Reverse Process in Three Coordinates
arXiv · Oct 15, 2026
SUAVE: Unified Video-Action Models via Masked Diffusion
arXiv · Oct 6, 2026