PixelDense:将密集预测视为像素扩散的表征对齐
摘要
PixelDense 将密集预测任务重构为像素空间扩散的表征对齐目标,通过独立的投影流分别路由语义教师(DINOv2、SAM2)和几何教师(Depth Anything v2、Metric3D v2),并加入正交性惩罚。该方法提升了 PixelGen 和 DeCo 上的 GenEval、DPG-Bench 与 HPSv2.1 得分,在部分噪声重建中 PQ 提升最高达 53.1%,且训练速度比基线快 1.23 倍。
查看缓存全文
缓存时间: 2026/10/02 04:27
论文页面 - PixelDense:将密集预测作为像素扩散的表征对齐
来源:https://huggingface.co/papers/2610.00483 作者:
摘要
表征对齐(Representation Alignment, REPA)能够加速扩散 Transformer 的训练,但其对齐目标几乎完全局限于 DINOv2 和 CLIP 这类语义编码器。近期的分析表明,承载对齐效果的是空间结构而非全局语义;然而,那些被训练来预测这一结构的密集预测基础模型,作为 REPA 的目标却一直被忽视。在像素空间扩散中,SAM2、DepthAnythingv2 和 Metric3Dv2 均优于仅使用 DINOv2 的 GenEval 基线,其中两个几何教师的性能领先于分割教师。然而,将四个教师的输出简单相加后,效果反而低于表现最好的单一几何教师,原因在于语义梯度与几何梯度会竞争同一个去噪器投影空间。我们提出 PixelDense:将 DINOv2 和 SAM2 路由到一个语义投影流,将 DepthAnythingv2 和 Metric3Dv2 路由到一个几何投影流,并引入权重空间的正交性惩罚,使两个流保持在互不相交的子空间中。四个教师在训练期间全部冻结,并在推理阶段被丢弃。在 PixelGen 和 DeCo 上使用同一套配方,PixelDense 在 GenEval、DPG-Bench 和 HPSv2.1 上均取得提升,将 PixelGen-XXL 的 GenEval Overall 从 0.7927 提高到 0.8093,并超越了所有单一教师以及未分解的多教师变体。在部分噪声重建中,使用独立的全景、深度和表面法线探针评估,在 COCO 和 Flickr30K 上、τ=0.5 时,PQ 提升最高达 53.1%,深度 AbsRel 最多降低 36.0%。从随机初始化出发,PixelDense 达到基线峰值 GenEval 的速度快 1.23 倍。在 PIE-Bench 上进行 SDEdit 编辑时,PixelDense 在每种编辑强度下都能保留更多的源图背景与布局,背景 PSNR 最高提升 2.2 dB。
查看 arXiv 页面 (https://arxiv.org/abs/2610.00483) 查看 PDF (https://arxiv.org/pdf/2610.00483) 项目页面 (https://bio.lehanyang.info/PixelDense/) GitHub3 (https://github.com/Hansxsourse/PixelDense) 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2610.00483)
引用本文的模型 0
没有模型链接到本文
在模型的 README.md 中引用 arxiv.org/abs/2610.00483,即可从本页面链接到它。
引用本文的数据集 0
没有数据集链接到本文
在数据集的 README.md 中引用 arxiv.org/abs/2610.00483,即可从本页面链接到它。
引用本文的 Spaces 0
没有 Space 链接到本文
在 Space 的 README.md 中引用 arxiv.org/abs/2610.00483,即可从本页面链接到它。
包含本文的合集 0
没有合集包含本文
将本文添加到合集 (https://huggingface.co/new-collection),即可从本页面链接到它。
相似文章
从RGB生成到密集场读出:基于文本到图像模型的像素空间密集预测
ReChannel通过将令牌直接映射到像素空间块,适配预训练的扩散变换器(如FLUX-Klein)用于密集预测任务,在无三元图的抠图、KITTI深度和指代分割上以极少的额外参数取得了最先进的成果。
PointDiT: 像素空间扩散用于单目几何估计
PointDiT提出了一种极简的像素空间扩散变换器,使用纯ViT架构进行单目几何估计,在超越复杂基于潜空间模型的同时,在模糊区域保持简洁性和鲁棒性。
学习表征以引导像素扩散 Transformer
论文提出表示引导(Representation Guidance,RG),在保持纯 Transformer 架构的前提下,通过低维瓶颈形成并主动利用内部表示来指导像素扩散模型的生成。RG-XL/16 在 ImageNet 256x256 上取得 FID 1.55,并可扩展至文本到图像生成(GenEval 0.89,DPG-Bench 87.1)。
像素扩散的对抗训练
本文首次系统研究了像素扩散模型的对抗后训练,表明它能有效纠正高频内容缺陷,从而提高分布保真度、覆盖范围和感知质量,且不改变模型架构。
像素空间文本到图像扩散模型的训练实证研究
本文针对像素空间文本到图像扩散模型提出了一种从潜在空间到像素空间的训练策略,加速了收敛并提高了推理速度,同时性能与潜在空间模型相当或更优。