arXiv reaDer
画像検索の再ランキングを理解する:グラフニューラルネットワークの視点
Understanding Image Retrieval Re-Ranking: A Graph Neural Network Perspective
再ランク付けアプローチでは、信頼性の高い検索サンプルを活用して検索結果を絞り込みます。これは、画像検索タスクの後処理ツールとして広く採用されています。ただし、再ランク付けの主な欠陥の1つ、つまり計算の複雑さが高いことに気づきました。これは、実際のアプリケーションにとって手ごろな時間のコストにつながります。この論文では、再ランク付けを再検討し、再ランク付けが高並列性グラフニューラルネットワーク(GNN)関数として再定式化できることを示します。特に、従来の再ランク付けプロセスを2つのフェーズに分割します。つまり、高品質のギャラリーサンプルの取得と機能の更新です。最初のフェーズはkに最も近い隣接グラフを作成することと同じであり、2番目のフェーズはグラフ内にメッセージを広めることと見なすことができると主張します。実際には、GNNは、接続されたエッジを持つ頂点のみを考慮する必要があります。グラフがまばらであるため、頂点の特徴を効率的に更新できます。 Market-1501データセットでは、1つのK40m GPUで再ランク付け処理を89.2秒から9.4ミリ秒に高速化し、リアルタイムの後処理を容易にします。同様に、私たちの方法では、他の4つの画像検索ベンチマーク、つまりVeRi-776、Oxford-5k、Paris-6k、University-1652で、限られた時間コストで同等またはそれ以上の検索結果が得られることがわかります。私たちのコードは公開されています。
The re-ranking approach leverages high-confidence retrieved samples to refine retrieval results, which have been widely adopted as a post-processing tool for image retrieval tasks. However, we notice one main flaw of re-ranking, i.e., high computational complexity, which leads to an unaffordable time cost for real-world applications. In this paper, we revisit re-ranking and demonstrate that re-ranking can be reformulated as a high-parallelism Graph Neural Network (GNN) function. In particular, we divide the conventional re-ranking process into two phases, i.e., retrieving high-quality gallery samples and updating features. We argue that the first phase equals building the k-nearest neighbor graph, while the second phase can be viewed as spreading the message within the graph. In practice, GNN only needs to concern vertices with the connected edges. Since the graph is sparse, we can efficiently update the vertex features. On the Market-1501 dataset, we accelerate the re-ranking processing from 89.2s to 9.4ms with one K40m GPU, facilitating the real-time post-processing. Similarly, we observe that our method achieves comparable or even better retrieval results on the other four image retrieval benchmarks, i.e., VeRi-776, Oxford-5k, Paris-6k and University-1652, with limited time cost. Our code is publicly available.
updated: Mon Dec 14 2020 15:12:36 GMT+0000 (UTC)
published: Mon Dec 14 2020 15:12:36 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アソシエイト