arXiv reaDer
新規クラスセグメンテーションに対するマスクヘッドアーキテクチャの驚くべき影響
The surprising impact of mask-head architecture on novel class segmentation
今日のインスタンスセグメンテーションモデルは、大きな注釈付きデータセットでトレーニングすると非常に正確ですが、マスク注釈を大規模に収集することは非常にコストがかかります。すべてのカテゴリの(大幅に安価な)バウンディングボックスでトレーニングできるが、カテゴリのサブセットにのみマスクを使用する、部分的に監視されたインスタンスセグメンテーションの問題に対処します。この作業では、微分可能なトリミングをフィーチャマップに適用し、結果のトリミングに基づいてマスクを予測する、人気のあるモデルファミリに焦点を当てます。このファミリの下で、Mask R-CNNを研究し、提案とグラウンドトゥルースボックスの組み合わせでマスクヘッドをトレーニングするというデフォルトの戦略の代わりに、グラウンドトゥルースボックスのみでマスクヘッドをトレーニングすると、新しいクラスでのパフォーマンスが劇的に向上することを発見しました。このトレーニング戦略により、2〜4層の一般的なマスクヘッドを非常に深い既製のアーキテクチャ(ResNet、砂時計モデルなど)に置き換えることで、代替のマスクヘッドアーキテクチャを活用することもできます。これらのアーキテクチャの多くは、完全な教師ありモードでトレーニングすると同様に機能しますが、主な発見は、劇的に異なる方法で新しいクラスに一般化できることです。マスクヘッドが目に見えないクラスに一般化するこの機能を強力なマスク一般化効果と呼び、特殊なモジュールや損失がなくても、部分的に監視されたCOCOインスタンスセグメンテーションベンチマークで最先端の結果を達成できることを示します。最後に、基礎となる検出方法(アンカーベース、アンカーフリー、または検出器なしを含む)全体、およびさまざまなバックボーンネットワーク全体で、効果が一般的であることを示します。コードと事前トレーニング済みモデルは、https://git.io/deepmacで入手できます。
Instance segmentation models today are very accurate when trained on large annotated datasets, but collecting mask annotations at scale is prohibitively expensive. We address the partially supervised instance segmentation problem in which one can train on (significantly cheaper) bounding boxes for all categories but use masks only for a subset of categories. In this work, we focus on a popular family of models which apply differentiable cropping to a feature map and predict a mask based on the resulting crop. Under this family, we study Mask R-CNN and discover that instead of its default strategy of training the mask-head with a combination of proposals and groundtruth boxes, training the mask-head with only groundtruth boxes dramatically improves its performance on novel classes. This training strategy also allows us to take advantage of alternative mask-head architectures, which we exploit by replacing the typical mask-head of 2-4 layers with significantly deeper off-the-shelf architectures (e.g. ResNet, Hourglass models). While many of these architectures perform similarly when trained in fully supervised mode, our main finding is that they can generalize to novel classes in dramatically different ways. We call this ability of mask-heads to generalize to unseen classes the strong mask generalization effect and show that without any specialty modules or losses, we can achieve state-of-the-art results in the partially supervised COCO instance segmentation benchmark. Finally, we demonstrate that our effect is general, holding across underlying detection methodologies (including anchor-based, anchor-free or no detector at all) and across different backbone networks. Code and pre-trained models are available at https://git.io/deepmac.
updated: Wed Aug 18 2021 03:06:02 GMT+0000 (UTC)
published: Thu Apr 01 2021 16:46:37 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アソシエイト