
摘要
基于组合优化与深度学习交叉领域近期取得的进展,我们提出了一种端到端可训练的深度图匹配架构,该架构包含未经修改的组合优化求解器。通过结合高度优化的组合求解器以及对网络结构设计的若干改进,我们在关键点对应任务的深度图匹配基准测试中实现了当前最优性能。此外,我们强调了将求解器嵌入深度学习架构所具有的概念性优势,例如可利用强大的多图匹配求解器进行后处理,或对训练设置的变化具有较强的鲁棒性。最后,我们提出了两种新的具有挑战性的实验设置。相关代码已开源,地址为:https://github.com/martius-lab/blackbox-deep-graph-matching。
代码仓库
q3erf/topo_bbgm
pytorch
GitHub 中提及
LPMP/LPMP
pytorch
GitHub 中提及
martius-lab/blackbox-deep-graph-matching
官方
pytorch
GitHub 中提及
Thinklab-SJTU/ThinkMatch
pytorch
martius-lab/blackbox-backprop
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| graph-matching-on-pascal-voc | BBGM-Multi | F1 score: 0.628 |
| graph-matching-on-pascal-voc | BBGM | F1 score: 0.614 matching accuracy: 0.801 |
| graph-matching-on-spair-71k | BBGM | matching accuracy: 0.8215 |
| graph-matching-on-willow-object-class | BBGM | matching accuracy: 0.9718 |