PixRestore: 基于像素扩散 Transformer 的统一图像恢复

摘要

PixRestore 是一种无 VAE 的像素空间扩散 Transformer，用于统一图像恢复（UIR）。它直接在补丁化像素上进行流匹配，保留细节。通过 LQ-HQ DINO 特征相似度预测层特征可靠性，实现自适应条件融合。在大规模数据上训练，并通过 DINO 对抗目标微调为单步生成器。约 50M 参数和单步推理下，在保真度、感知质量和鲁棒性上优于现有 UIR 模型，且效率更高。更大变体可进一步提升性能，展现了可扩展性。