Search papers, labs, and topics across Lattice.
FlowIt, a new optical flow estimation architecture, uses a hierarchical transformer to capture global context and model long-range correspondences. It formulates flow initialization as an optimal transport problem, generating robust initial flow fields and explicit occlusion/confidence maps. These maps guide a refinement stage that propagates motion estimates from high-confidence regions, achieving SOTA results on Sintel and KITTI, and strong zero-shot generalization.
Global context and confidence-guided refinement can unlock state-of-the-art optical flow estimation, even in challenging scenarios with large displacements and occlusions.
We present FlowIt, a novel architecture for optical flow estimation designed to robustly handle large pixel displacements. At its core, FlowIt leverages a hierarchical transformer architecture that captures extensive global context, enabling the model to effectively model long-range correspondences. To overcome the limitations of localized matching, we formulate the flow initialization as an optimal transport problem. This formulation yields a highly robust initial flow field, alongside explicitly derived occlusion and confidence maps. These cues are then seamlessly integrated into a guided refinement stage, where the network actively propagates reliable motion estimates from high-confidence regions into ambiguous, low-confidence areas. Extensive experiments across the Sintel, KITTI, Spring, and LayeredFlow datasets validate the efficacy of our approach. FlowIt achieves state-of-the-art results on the competitive Sintel and KITTI benchmarks, while simultaneously establishing new state-of-the-art cross-dataset zero-shot generalization performance on Sintel, Spring, and LayeredFlow.
