SpectralDiT: Timestep-Conditioned Spectral Residual Correction for Flow-Matching DiTs

Jiayu Tian

We propose SpectralDiT, a lightweight modification to flow-matching Diffusion Transformers that adds timestep-conditioned spectral correction to the MLP residual branch. The module…