arXiv reaDer
SegMatch: 手術器具のセグメンテーションのための半教師あり学習方法
SegMatch: A semi-supervised learning method for surgical instrument segmentation
外科器具のセグメンテーションは、高度な外科支援を提供し、コンピューター支援介入を改善するための重要な要素として認識されています。この研究では、腹腔鏡手術画像やロボット手術画像に対する高価なアノテーションの必要性を減らすための半教師あり学習方法である SegMatch を提案します。 SegMatch は、一貫性の正則化と擬似ラベル付けを組み合わせた、広く普及している半教師あり分類パイプラインである FixMatch を基盤として構築されており、それをセグメンテーションの目的に適応させています。私たちが提案する SegMatch では、ラベルのない画像が弱く拡張されてセグメンテーション モデルに入力され、高い信頼スコアを持つピクセル上の敵対的な拡張画像のモデルの出力に対して教師なし損失を強制する疑似ラベルが生成されます。セグメンテーション タスクへの適応には、依存する拡張関数の等変性と不変性の特性を慎重に考慮することが含まれます。拡張の関連性を高めるために、手作りの拡張のみの使用をやめ、訓練可能な敵対的拡張戦略を導入します。私たちのアルゴリズムは、MICCAI Instrument Segmentation Challenge データセットである Robust-MIS 2019 および EndoVis 2017 で評価されました。私たちの結果は、トレーニング目的でラベルなしのデータを追加することで、これらのトレーニング データの利用可能性によって制限される完全に教師ありのアプローチのパフォーマンスを上回ることができることを示しています。課題。また、SegMatch は、さまざまなラベル付きデータとラベルなしデータの比率において、さまざまな最先端の半教師あり学習セマンティック セグメンテーション モデルよりも優れたパフォーマンスを発揮します。
Surgical instrument segmentation is recognised as a key enabler to provide advanced surgical assistance and improve computer assisted interventions. In this work, we propose SegMatch, a semi supervised learning method to reduce the need for expensive annotation for laparoscopic and robotic surgical images. SegMatch builds on FixMatch, a widespread semi supervised classification pipeline combining consistency regularization and pseudo labelling, and adapts it for the purpose of segmentation. In our proposed SegMatch, the unlabelled images are weakly augmented and fed into the segmentation model to generate a pseudo-label to enforce the unsupervised loss against the output of the model for the adversarial augmented image on the pixels with a high confidence score. Our adaptation for segmentation tasks includes carefully considering the equivariance and invariance properties of the augmentation functions we rely on. To increase the relevance of our augmentations, we depart from using only handcrafted augmentations and introduce a trainable adversarial augmentation strategy. Our algorithm was evaluated on the MICCAI Instrument Segmentation Challenge datasets Robust-MIS 2019 and EndoVis 2017. Our results demonstrate that adding unlabelled data for training purposes allows us to surpass the performance of fully supervised approaches which are limited by the availability of training data in these challenges. SegMatch also outperforms a range of state-of-the-art semi-supervised learning semantic segmentation models in different labelled to unlabelled data ratios.
updated: Wed Aug 09 2023 21:30:18 GMT+0000 (UTC)
published: Wed Aug 09 2023 21:30:18 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アソシエイト