arXiv reaDer
GLAN:グラフベースの線形割り当てネットワーク
GLAN: A Graph-based Linear Assignment Network
線形割り当て問題(LAP)の微分可能ソルバーは、近年多くの研究の注目を集めており、通常、コンポーネントとして学習フレームワークに組み込まれています。ただし、学習戦略の有無にかかわらず、以前のアルゴリズムは通常、問題のサイズが大きくなるにつれて最適性が低下するという問題があります。本論文では、深いグラフネットワークに基づく学習可能な線形割り当てソルバーを提案した。具体的には、最初にコストマトリックスを2部グラフに変換し、割り当てタスクを、構築されたグラフから信頼できるエッジを選択する問題に変換します。続いて、ノードとエッジの機能を集約および更新するために、ディープグラフネットワークが開発されます。最後に、ネットワークは、割り当て関係を示す各エッジのラベルを予測します。合成データセットでの実験結果は、私たちの方法が最先端のベースラインを上回り、問題のサイズの増加とともに一貫して高い精度を達成することを明らかにしています。さらに、最先端のベースラインソルバーと比較して、提案されたソルバーを一般的なマルチオブジェクトトラッキング(MOT)フレームワークに組み込み、トラッカーをエンドツーエンドでトレーニングします。 MOTベンチマークの実験結果は、提案されたLAPソルバーがトラッカーを最大のマージンで改善することを示しています。
Differentiable solvers for the linear assignment problem (LAP) have attracted much research attention in recent years, which are usually embedded into learning frameworks as components. However, previous algorithms, with or without learning strategies, usually suffer from the degradation of the optimality with the increment of the problem size. In this paper, we propose a learnable linear assignment solver based on deep graph networks. Specifically, we first transform the cost matrix to a bipartite graph and convert the assignment task to the problem of selecting reliable edges from the constructed graph. Subsequently, a deep graph network is developed to aggregate and update the features of nodes and edges. Finally, the network predicts a label for each edge that indicates the assignment relationship. The experimental results on a synthetic dataset reveal that our method outperforms state-of-the-art baselines and achieves consistently high accuracy with the increment of the problem size. Furthermore, we also embed the proposed solver, in comparison with state-of-the-art baseline solvers, into a popular multi-object tracking (MOT) framework to train the tracker in an end-to-end manner. The experimental results on MOT benchmarks illustrate that the proposed LAP solver improves the tracker by the largest margin.
updated: Wed Jan 05 2022 13:18:02 GMT+0000 (UTC)
published: Wed Jan 05 2022 13:18:02 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト