arXiv reaDer
最適なトランスポートマッチングとメッセージフローによる数ショットのセグメンテーション
Few-shot Segmentation with Optimal Transport Matching and Message Flow
この作業では、数ショットのセグメンテーションという難しいタスクに対処します。数ショットのセマンティックセグメンテーションでは、サポート情報を十分に活用することが不可欠です。以前の方法は、通常、サポート機能にマスクされた平均プーリングを適応させて、サポートの手がかりをグローバルベクトルとして抽出します。これは通常、顕著な部分によって支配され、いくつかの重要な手がかりを失います。この作業では、すべてのサポートピクセルの情報をすべてのクエリピクセルに転送することが望ましいと主張し、クエリとサポート画像間の対応を掘り起こすための最適なトランスポートマッチングモジュールを備えた通信マッチングネットワーク(CMNet)を提案します。さらに、注釈付きのサポート画像からのローカル情報とグローバル情報の両方を十分に活用することが重要です。この目的のために、同じ画像内の内部フローに沿ってメッセージを伝播し、サポート画像とクエリ画像の間のクロスフローを行うメッセージフローモジュールを提案します。これにより、ローカル機能の表現が大幅に強化されます。さらに、異なるデータセット間のドメインギャップの問題を軽減するためのマルチタスク学習問題として、数ショットのセグメンテーションに対処します。 PASCAL VOC 2012、MS COCO、およびFSS-1000データセットでの実験は、私たちのネットワークが新しい最先端の数ショットのセグメンテーションパフォーマンスを達成することを示しています。
We address the challenging task of few-shot segmentation in this work. It is essential for few-shot semantic segmentation to fully utilize the support information. Previous methods typically adapt masked average pooling over the support feature to extract the support clues as a global vector, usually dominated by the salient part and loses some important clues. In this work, we argue that every support pixel's information is desired to be transferred to all query pixels and propose a Correspondence Matching Network (CMNet) with an Optimal Transport Matching module to mine out the correspondence between the query and support images. Besides, it is important to fully utilize both local and global information from the annotated support images. To this end, we propose a Message Flow module to propagate the message along the inner-flow within the same image and cross-flow between support and query images, which greatly help enhance the local feature representations. We further address the few-shot segmentation as a multi-task learning problem to alleviate the domain gap issue between different datasets. Experiments on PASCAL VOC 2012, MS COCO, and FSS-1000 datasets show that our network achieves new state-of-the-art few-shot segmentation performance.
updated: Thu Aug 19 2021 06:26:11 GMT+0000 (UTC)
published: Thu Aug 19 2021 06:26:11 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アソシエイト