arXiv reaDer
タスク間の一貫性正則化による単一段階のオープンワールド インスタンス セグメンテーション
Single-Stage Open-world Instance Segmentation with Cross-task Consistency Regularization
オープンワールド インスタンス セグメンテーション (OWIS) は、クラスに依存しないインスタンスを画像からセグメント化することを目的としています。これには、自動運転などの幅広い実世界のアプリケーションがあります。ほとんどの既存のアプローチは、2 段階のパイプラインに従います。最初にクラスに依存しない検出を実行し、次にクラス固有のマスク セグメンテーションを実行します。対照的に、この論文では、各インスタンスのマスクを直接生成するための単一段階のフレームワークを提案しています。また、インスタンス マスクの注釈は、既存のデータセットではノイズが多い可能性があります。この問題を克服するために、新しい正則化損失を導入します。具体的には、最初に前景領域 (つまり、任意のオブジェクト インスタンスに属する領域) を予測する補助タスクを実行する追加ブランチをトレーニングし、次に補助ブランチからの予測がインスタンス マスクの予測と一致するようにします。重要な洞察は、このようなクロスタスクの一貫性の喪失が、注釈のエラーに対処するためのエラー修正メカニズムとして機能する可能性があるということです。さらに、提案されたクロスタスクの一貫性の喪失は、注釈なしで画像に適用できることを発見し、半教師あり学習法に役立ちます。広範な実験を通じて、提案された方法が完全教師あり設定と半教師あり設定の両方で印象的な結果を達成できることを示します。 SOTA メソッドと比較して、提案されたメソッドは AP_100 スコアを UVO→UVO 設定で 4.75%、COCO→UVO 設定で 4.05% 大幅に改善します。半教師あり学習の場合、私たちのモデルは 30% のラベル付きデータのみで学習し、50% のラベル付きデータで完全に教師ありのモデルよりも優れています。コードは近日公開予定です。
Open-world instance segmentation (OWIS) aims to segment class-agnostic instances from images, which has a wide range of real-world applications such as autonomous driving. Most existing approaches follow a two-stage pipeline: performing class-agnostic detection first and then class-specific mask segmentation. In contrast, this paper proposes a single-stage framework to produce a mask for each instance directly. Also, instance mask annotations could be noisy in the existing datasets; to overcome this issue, we introduce a new regularization loss. Specifically, we first train an extra branch to perform an auxiliary task of predicting foreground regions (i.e. regions belonging to any object instance), and then encourage the prediction from the auxiliary branch to be consistent with the predictions of the instance masks. The key insight is that such a cross-task consistency loss could act as an error-correcting mechanism to combat the errors in annotations. Further, we discover that the proposed cross-task consistency loss can be applied to images without any annotation, lending itself to a semi-supervised learning method. Through extensive experiments, we demonstrate that the proposed method can achieve impressive results in both fully-supervised and semi-supervised settings. Compared to SOTA methods, the proposed method significantly improves the AP_100 score by 4.75% in UVO→UVO setting and 4.05% in COCO→UVO setting. In the case of semi-supervised learning, our model learned with only 30% labeled data, even outperforms its fully-supervised counterpart with 50% labeled data. The code will be released soon.
updated: Thu Aug 18 2022 18:55:09 GMT+0000 (UTC)
published: Thu Aug 18 2022 18:55:09 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アソシエイト