arXiv:2412.06439cs.CV2024-12

改进光流上采样,用局部注意力提升细节精度

Local Attention Transformers for High-Detail Optical Flow Upsampling

  • 将凸上采样重构为局部注意力机制,扩大感受野
  • 在飞鸟数据集上将光流误差降低0.14至0.16,效果显著
  • 可无缝集成到现有主流光流模型中,适合追求高精度的开发者

当前大多数光流方法在最后阶段采用凸上采样以获得高分辨率结果。本文指出该方法存在若干问题与局限性,并提出一系列改进方案。首先,解耦凸上采样器权重,使其更易找到正确的凸组合;同时引入额外上下文特征增强表达能力。其次,提出基于注意力的凸上采样器——使用局部注意力掩码替代传统固定大小的凸掩码,以增大感受野。我们证明更大的掩码尺寸能提高凸组合存在的可能性。最后,设计一种新训练策略,消除双线性插值带来的伪影。所提方法可适配几乎所有现有先进光流架构。在FlyingChairs + FlyingThings3D训练设置下,仅替换凸上采样器,便使RAFT的Sintel Clean端点误差从1.42降至1.26,GMA从1.31降至1.18,FlowFormer从0.94降至0.90。

原文摘要 · Abstract (English)

Most recent works on optical flow use convex upsampling as the last step to obtain high-resolution flow. In this work, we show and discuss several issues and limitations of this currently widely adopted convex upsampling approach. We propose a series of changes, in an attempt to resolve current issues. First, we propose to decouple the weights for the final convex upsampler, making it easier to find the correct convex combination. For the same reason, we also provide extra contextual features to the convex upsampler. Then, we increase the convex mask size by using an attention-based alternative convex upsampler; Transformers for Convex Upsampling. This upsampler is based on the observation that convex upsampling can be reformulated as attention, and we propose to use local attention masks as a drop-in replacement for convex masks to increase the mask size. We provide empirical evidence that a larger mask size increases the likelihood of the existence of the convex combination. Lastly, we propose an alternative training scheme to remove bilinear interpolation artifacts from the model output. Our proposed ideas could theoretically be applied to almost every current state-of-the-art optical flow architecture. On the FlyingChairs + FlyingThings3D training setting we reduce the Sintel Clean training end-point-error of RAFT from 1.42 to 1.26, GMA from 1.31 to 1.18, and that of FlowFormer from 0.94 to 0.90, by solely adapting the convex upsampler.

光流注意力机制上采样图像对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。