3 个月前

通过组合求解器的黑箱微分实现深度图匹配

通过组合求解器的黑箱微分实现深度图匹配

摘要

基于组合优化与深度学习交叉领域近期取得的进展,我们提出了一种端到端可训练的深度图匹配架构,该架构包含未经修改的组合优化求解器。通过结合高度优化的组合求解器以及对网络结构设计的若干改进,我们在关键点对应任务的深度图匹配基准测试中实现了当前最优性能。此外,我们强调了将求解器嵌入深度学习架构所具有的概念性优势,例如可利用强大的多图匹配求解器进行后处理,或对训练设置的变化具有较强的鲁棒性。最后,我们提出了两种新的具有挑战性的实验设置。相关代码已开源,地址为: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 中提及
martius-lab/blackbox-backprop
pytorch
GitHub 中提及

基准测试

基准方法指标
graph-matching-on-pascal-vocBBGM-Multi
F1 score: 0.628
graph-matching-on-pascal-vocBBGM
F1 score: 0.614
matching accuracy: 0.801
graph-matching-on-spair-71kBBGM
matching accuracy: 0.8215
graph-matching-on-willow-object-classBBGM
matching accuracy: 0.9718

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过组合求解器的黑箱微分实现深度图匹配 | 论文 | HyperAI超神经