Michal Rolínek, Paul Swoboda, Dominik Zietlow, Anselm Paulus, Vít Musil, Georg Martius
Building on recent progress at the intersection of combinatorial optimization and deep learning, we propose an end-to-end trainable architecture for deep graph matching that contains unmodified combinatorial solvers. Using the presence of heavily optimized combinatorial solvers together with some improvements in architecture design, we advance state-of-the-art on deep graph matching benchmarks for keypoint correspondence. In addition, we highlight the conceptual advantages of incorporating solvers into deep learning architectures, such as the possibility of post-processing with a strong multi-graph matching solver or the indifference to changes in the training setting. Finally, we propose two new challenging experimental setups. The code is available at https://github.com/martius-lab/blackbox-deep-graph-matching
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Graph Matching | SPair-71k | matching accuracy | 0.8215 | BBGM |
| Graph Matching | Willow Object Class | matching accuracy | 0.9718 | BBGM |
| Graph Matching | PASCAL VOC | F1 score | 0.628 | BBGM-Multi |
| Graph Matching | PASCAL VOC | F1 score | 0.614 | BBGM |
| Graph Matching | PASCAL VOC | matching accuracy | 0.801 | BBGM |