arXiv reaDer
クラスが不均衡なかろうじて監視された膝のセグメンテーションのためのラベル分布の較正
Calibrating Label Distribution for Class-Imbalanced Barely-Supervised Knee Segmentation
3D膝MR画像のセグメンテーションは、変形性関節症の評価にとって重要です。他の医療データと同様に、膝のMR画像のボリュームごとのラベリングは、専門知識が要求され、時間がかかります。したがって、半教師あり学習(SSL)、特にほとんど教師あり学習は、ラベル付けされたデータが不十分なトレーニングには非常に望ましいものです。膝のMR画像では、軟骨が前景の体積の6%しか占めておらず、十分なラベル付きデータがないと状況が悪化するため、クラスの不均衡の問題が深刻であることがわかりました。上記の問題に対処するために、ノイズが多く不均衡なラベルを使用した、ほとんど監視されていない膝のセグメンテーションのための新しいフレームワークを提示します。私たちのフレームワークは、ラベル配布を活用して、ネットワークが軟骨部分の学習により多くの努力を払うように促します。具体的には、1。)客観的損失関数をクラス認識加重形式に変更するためのラベル量分布、および2.)ラベル付きとラベルなしの両方から軟骨領域のより多くのサブボリュームをトリミングするためのトリミング確率マスクを構築するためのラベル位置分布を利用します。入力。さらに、効率的な教師なし学習のために信頼性の低いカテゴリの監視を強化するために、二重の不確実性を意識したサンプリング監視を設計します。実験によると、提案されたフレームワークは、ラベルのないデータを組み込み、クラスの不均衡の問題を軽減することにより、大幅な改善をもたらします。さらに重要なことに、私たちの方法は最先端のSSL方法よりも優れており、より困難なSSL設定に対する私たちのフレームワークの可能性を示しています。
Segmentation of 3D knee MR images is important for the assessment of osteoarthritis. Like other medical data, the volume-wise labeling of knee MR images is expertise-demanded and time-consuming; hence semi-supervised learning (SSL), particularly barely-supervised learning, is highly desirable for training with insufficient labeled data. We observed that the class imbalance problem is severe in the knee MR images as the cartilages only occupy 6% of foreground volumes, and the situation becomes worse without sufficient labeled data. To address the above problem, we present a novel framework for barely-supervised knee segmentation with noisy and imbalanced labels. Our framework leverages label distribution to encourage the network to put more effort into learning cartilage parts. Specifically, we utilize 1.) label quantity distribution for modifying the objective loss function to a class-aware weighted form and 2.) label position distribution for constructing a cropping probability mask to crop more sub-volumes in cartilage areas from both labeled and unlabeled inputs. In addition, we design dual uncertainty-aware sampling supervision to enhance the supervision of low-confident categories for efficient unsupervised learning. Experiments show that our proposed framework brings significant improvements by incorporating the unlabeled data and alleviating the problem of class imbalance. More importantly, our method outperforms the state-of-the-art SSL methods, demonstrating the potential of our framework for the more challenging SSL setting.
updated: Sat May 07 2022 12:53:06 GMT+0000 (UTC)
published: Sat May 07 2022 12:53:06 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アソシエイト