マッチングコストの学習は、3D畳み込みを4Dフィーチャボリュームに適用して3Dコストボリュームを学習する、最先端のディープステレオマッチング手法の成功に不可欠であることが示されています。ただし、このメカニズムはオプティカルフロータスクに採用されたことはありません。これは主に、オプティカルフロー計算の場合に検索次元が大幅に増加するためです。つまり、単純な拡張では、5Dフィーチャボリュームを処理するために高密度の4D畳み込みが必要になりますが、これは計算上非常に困難です。このホワイトペーパーでは、ネットワークがデータから適切なマッチングコストを学習できるようにしながら、5Dフィーチャボリュームを構築する要件を回避できる新しいソリューションを提案します。私たちの重要な革新は、2D変位間の接続を切り離し、各2D変位仮説でのマッチングコストを個別に学習することです。つまり、変位不変のコスト学習です。具体的には、同じ2D畳み込みベースのマッチングネットを各2D変位仮説に個別に適用して、4Dコストボリュームを学習します。さらに、異なる変位候補間の相関を再検討し、学習されたコストボリュームのマルチモーダル問題を軽減する、学習されたコストボリュームをスケーリングするための変位認識投影層を提案します。次に、コストボリュームは、2Dソフトアルグミン層を介したオプティカルフロー推定に投影されます。広範な実験により、私たちのアプローチはさまざまなデータセットで最先端の精度を達成し、Sintelベンチマークで公開されているすべてのオプティカルフロー手法よりも優れていることが示されています。
Learning matching costs has been shown to be critical to the success of the state-of-the-art deep stereo matching methods, in which 3D convolutions are applied on a 4D feature volume to learn a 3D cost volume. However, this mechanism has never been employed for the optical flow task. This is mainly due to the significantly increased search dimension in the case of optical flow computation, ie, a straightforward extension would require dense 4D convolutions in order to process a 5D feature volume, which is computationally prohibitive. This paper proposes a novel solution that is able to bypass the requirement of building a 5D feature volume while still allowing the network to learn suitable matching costs from data. Our key innovation is to decouple the connection between 2D displacements and learn the matching costs at each 2D displacement hypothesis independently, ie, displacement-invariant cost learning. Specifically, we apply the same 2D convolution-based matching net independently on each 2D displacement hypothesis to learn a 4D cost volume. Moreover, we propose a displacement-aware projection layer to scale the learned cost volume, which reconsiders the correlation between different displacement candidates and mitigates the multi-modal problem in the learned cost volume. The cost volume is then projected to optical flow estimation through a 2D soft-argmin layer. Extensive experiments show that our approach achieves state-of-the-art accuracy on various datasets, and outperforms all published optical flow methods on the Sintel benchmark.