CORTEXA
← Browse
arxivcs.LG2026-06-29

Quality-Aware Modulation for Diffusion Transformers

Luke Budny, Yuhong Guo, Kevin Cheung

Modern text-to-image diffusion models, such as diffusion transformers (DiT), rely on timestep or prompt embeddings to modulate the strength of the denoising process in each timestep. While this modulation communicates the current noise level, it does not provide any quality-aware information, which can lead to generated images that are unaligned, visually inconsistent, and lacking in fidelity. In this paper, we propose the Quality Representation Module (QRM), a lightweight transformer module that learns a quality-aware representation based on existing model inputs, and produces a set of vectors $M_{qrm}$. These vectors adjust the adaptive LayerNorm modulation within the DiT transformer blocks, thereby injecting a quality-sensitive signal into the denoising parameters. The QRM introduces no significant changes to the sampling schedule or diffusion backbone. Experiments include ablations on QRM training losses and architectures, as well as empirical results demonstrating consistent image quality improvements over baseline DiT-based models.

View free PDFSource page

Related papers

arxivcs.LGcs.CV2026-07-23

KroQuant: Kronecker-Structured Block Transforms for Efficient Post-Training Quantization of Diffusion Transformers

Yann Bouquet, Alireza Khodamoradi, Kristof Denolf, Mathieu Salzmann

Post-training quantization (PTQ) of diffusion transformers (DiTs) to W4A4 severely degrades output quality, because activations entering each linear layer contain outliers that 4-bit formats cannot represent. The standard fix applies an invertible linear transform to the activati…

View free PDFSource page
arxivcs.LGcs.AI2026-07-31

MBDiff: Multi-view Behavior-aware Diffusion Model for Probabilistic Utility Data Imputation

Rongchao Xu, Lin Jiang, Dahai Yu, Ximiao Li, Guang Wang

Utility data (e.g., electricity, water, and gas consumption), collected by ubiquitous sensors and embedded devices, often contains substantial missing values due to various factors such as device failures and data transmission issues. The data missingness can severely impact util…

View free PDFSource page
arxiveess.SPcs.LG2026-07-23

RadioTrace: Transmitter-Aware Diffusion for Radio Map Estimation without Deployment-Time Fine-Tuning

Liu Yang, Qiang Li, Zhuo Cao, Weijie Xiong, Guomin Sun, Jingran Lin

Radio map (RM) estimation aims to reconstruct the spatial distribution of wireless signal characteristics, such as received signal strength (RSS), from sparse measurements, a task that is critical for spectrum management, interference mitigation, and localization in modern wirele…

View free PDFSource page
arxivstat.MLcs.LG2026-07-23

Transformer-based Diffusion models for Hydrological Time Series Probabilistic Imputation and Forecasting

Ferdinand Bhavsar, Lionel Benoit, Maxime Savatier, Edith Gabriel

The modeling of hydrometeorological time series with limited observations is a key challenge in the monitoring of hydro-systems and water resources, as well as for flood or drought risk assessment. Due to the high variability of the underlying processes and the sparsity of availa…

View free PDFSource page
arxivcs.CVcs.LG2026-07-23

DART: A Degradation-Aware Recurrent Transformer for Archival Film Restoration

Mikołaj Jastrzębski, Wojciech Kozłowski, Kamil Adamczewski

Archival film restoration is a challenging problem because historical footage contains compound degradations such as scratches, dust, blur, noise, flicker, and photometric aging, while clean reference videos are unavailable. Existing video restoration methods largely treat these…

View free PDFSource page