arXiv reaDer
ポイントレベルの注釈を使用した都市シーンのセマンティックセグメンテーションのための一貫性-正則化された領域成長ネットワーク
Consistency-Regularized Region-Growing Network for Semantic Segmentation of Urban Scenes with Point-Level Annotations
深層学習アルゴリズムは、非常に高解像度(VHR)の画像のセマンティックセグメンテーションで大きな成功を収めています。それにもかかわらず、これらのモデルのトレーニングには、通常、大量の正確なピクセル単位の注釈が必要であり、収集するのに非常に手間と時間がかかります。注釈の負担を軽減するために、この論文では、ポイントレベルの注釈を使用してVHR画像のセマンティックセグメンテーションを実現するための整合性正則化領域成長ネットワーク(CRGNet)を提案します。 CRGNetの重要なアイデアは、ラベルのないピクセルを高い信頼度で繰り返し選択して、元のまばらなポイントから注釈付きの領域を拡張することです。ただし、拡張された注釈にはエラーやノイズが存在する可能性があるため、それらから直接学習すると、ネットワークのトレーニングを誤解させる可能性があります。この目的のために、基本分類器と拡張分類器を使用する整合性正則化戦略をさらに提案します。具体的には、基本分類器は元のスパース注釈によって監視されますが、拡張分類器は、領域拡大メカニズムを使用して基本分類器によって生成された拡張注釈から学習することを目的としています。これにより、一貫性の正則化は、基本分類子と拡張分類子の両方からの予測間の不一致を最小限に抑えることによって実現されます。このような単純な正則化戦略は、地域拡大メカニズムの品質を制御するのに非常に役立つことがわかります。 2つのベンチマークデータセットでの広範な実験は、提案されたCRGNetが既存の最先端の方法を大幅に上回っていることを示しています。コードと事前トレーニング済みモデルはオンラインで入手できます(https://github.com/YonghaoXu/CRGNet)。
Deep learning algorithms have obtained great success in semantic segmentation of very high-resolution (VHR) images. Nevertheless, training these models generally requires a large amount of accurate pixel-wise annotations, which is very laborious and time-consuming to collect. To reduce the annotation burden, this paper proposes a consistency-regularized region-growing network (CRGNet) to achieve semantic segmentation of VHR images with point-level annotations. The key idea of CRGNet is to iteratively select unlabeled pixels with high confidence to expand the annotated area from the original sparse points. However, since there may exist some errors and noises in the expanded annotations, directly learning from them may mislead the training of the network. To this end, we further propose the consistency regularization strategy, where a base classifier and an expanded classifier are employed. Specifically, the base classifier is supervised by the original sparse annotations, while the expanded classifier aims to learn from the expanded annotations generated by the base classifier with the region-growing mechanism. The consistency regularization is thereby achieved by minimizing the discrepancy between the predictions from both the base and the expanded classifiers. We find such a simple regularization strategy is yet very useful to control the quality of the region-growing mechanism. Extensive experiments on two benchmark datasets demonstrate that the proposed CRGNet significantly outperforms the existing state-of-the-art methods. Codes and pre-trained models are available online (https://github.com/YonghaoXu/CRGNet).
updated: Sat Jun 18 2022 13:33:00 GMT+0000 (UTC)
published: Tue Feb 08 2022 09:27:01 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アソシエイト