arXiv reaDer
SemiCurv:半教師あり曲線構造セグメンテーション
SemiCurv: Semi-Supervised Curvilinear Structure Segmentation
曲線構造のセグメンテーションに関する最近の研究は、主にバックボーンネットワークの設計と損失エンジニアリングに焦点を当てています。ラベル付けされたデータを収集するという課題は、費用と労力を要するプロセスであり、見過ごされてきました。ラベル付きデータの取得には費用がかかりますが、ラベルなしデータはすぐに利用できることがよくあります。この作業では、このようなラベル付けされていないデータを利用してラベル付けの負担を軽減できる、曲線構造セグメンテーションのための半教師あり学習(SSL)フレームワークであるSemiCurvを提案します。私たちのフレームワークは、半教師ありの方法で曲線セグメンテーションを定式化する際の2つの重要な課題に対処します。まず、整合性ベースのSSLの能力を十分に活用するために、強力なデータ拡張として幾何学的変換を導入し、次に、ピクセル単位の整合性の計算を可能にするために、微分可能な逆変換を介してセグメンテーション予測を調整します。第2に、ラベルのないデータの従来の平均二乗誤差(MSE)は予測が崩壊する傾向があり、この問題は深刻なクラスの不均衡(大幅に多くの背景ピクセル)で悪化します。ラベル付けされていないデータの些細な予測を回避するために、Nペアの整合性の損失を提案します。 6つの曲線セグメンテーションデータセットでSemiCurvを評価し、ラベル付けされたデータの5%以下で、完全に監視された対応物と比較して95%近くのパフォーマンスを達成することを発見しました。
Recent work on curvilinear structure segmentation has mostly focused on backbone network design and loss engineering. The challenge of collecting labelled data, an expensive and labor intensive process, has been overlooked. While labelled data is expensive to obtain, unlabelled data is often readily available. In this work, we propose SemiCurv, a semi-supervised learning (SSL) framework for curvilinear structure segmentation that is able to utilize such unlabelled data to reduce the labelling burden. Our framework addresses two key challenges in formulating curvilinear segmentation in a semi-supervised manner. First, to fully exploit the power of consistency based SSL, we introduce a geometric transformation as strong data augmentation and then align segmentation predictions via a differentiable inverse transformation to enable the computation of pixel-wise consistency. Second, the traditional mean square error (MSE) on unlabelled data is prone to collapsed predictions and this issue exacerbates with severe class imbalance (significantly more background pixels). We propose a N-pair consistency loss to avoid trivial predictions on unlabelled data. We evaluate SemiCurv on six curvilinear segmentation datasets, and find that with no more than 5% of the labelled data, it achieves close to 95% of the performance relative to its fully supervised counterpart.
updated: Thu May 19 2022 05:48:42 GMT+0000 (UTC)
published: Wed May 18 2022 03:52:17 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アソシエイト