UPLiFT:基于局部注意力算子的高效像素稠密特征上采样
UPLiFT: Efficient Pixel-Dense Feature Upsampling with Local Attenders
UPLiFT 提供了一种任务无关的像素稠密特征上采样模块,可直接替换 FPN 式插值或交叉注意力上采样,在分割、深度等稠密预测任务中以更低推理成本获得更稳定的高分辨率特征;其局部注意力池化算子也可迁移到其他需要局部注意力的轻量模块设计中。
原文依据 they can achieve state-of-the-art performance with lower inference costs
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论任务无关的特征上采样空间已成为一个有前景的研究方向,其目标是从预训练视觉骨干网络高效地生成更稠密的特征。这类方法通过将低分辨率特征映射到高分辨率版本,以很小的代价获得稠密特征,从而起到捷径作用。该领域早期工作采用迭代式上采样方法,而近期工作则转向基于交叉注意力的方法,但后者可能陷入与其所上采样的骨干网络相同的效率扩展问题。在本工作中,我们证明迭代式上采样方法仍可与基于交叉注意力的方法相竞争;此外,它们能够在更低的推理成本下达到最优性能。我们提出 UPLiFT,一种通用像素稠密轻量特征变换架构。我们还提出一种高效的局部注意力算子,以克服先前迭代式特征上采样方法的局限。该算子采用一种完全在局部定义的替代性注意力池化形式。我们表明,局部注意力算子使 UPLiFT 在整个上采样过程中保持稳定的特征,从而以低于现有像素稠密特征上采样器的推理成本达到最优性能。此外,我们将 UPLiFT 应用于生成式下游任务,并表明其在变分自编码器特征上采样方面与最优的耦合流匹配模型相比具有竞争力。总之,UPLiFT 为生成更稠密特征提供了一种通用且高效的方法。
英文原文摘要 1552 字符
The space of task-agnostic feature upsampling has emerged as a promising area of research to efficiently create denser features from pre-trained visual backbones. These methods act as a shortcut to achieve dense features for a fraction of the cost by learning to map low-resolution features to high-resolution versions. While early works in this space used iterative upsampling approaches, more recent works have switched to cross-attention-based methods, which risk falling into the same efficiency scaling problems of the backbones they are upsampling. In this work, we demonstrate that iterative upsampling methods can still compete with cross-attention-based methods; moreover, they can achieve state-of-the-art performance with lower inference costs. We propose UPLiFT, an architecture for Universal Pixel-dense Lightweight Feature Transforms. We also propose an efficient Local Attender operator to overcome the limitations of prior iterative feature upsampling methods. This operator uses an alternative attentional pooling formulation defined fully locally. We show that our Local Attender allows UPLiFT to maintain stable features throughout upsampling, enabling state-of-the-art performance with lower inference costs than existing pixel-dense feature upsamplers. In addition, we apply UPLiFT to generative downstream tasks and show that it achieves competitive performance with state-of-the-art Coupled Flow Matching models for VAE feature upsampling. Altogether, UPLiFT offers a versatile and efficient approach to creating denser features.