arXiv reaDer
下肢の筋骨格セグメンテーションにおけるベイジアンアクティブラーニングのためのハイブリッド表現強化サンプリング
Hybrid Representation-Enhanced Sampling for Bayesian Active Learning in Musculoskeletal Segmentation of Lower Extremities
目的: 自動セグメンテーション用にディープ ラーニング (DL) モデルをトレーニングするために手動でアノテーションを取得するには、多くの場合時間がかかります。不確実性ベースのベイジアンアクティブラーニング(BAL)は、アノテーションの労力を軽減するために広く採用されている方法です。この研究では、BAL に基づいて、密度と多様性の基準を統合するハイブリッド表現強化サンプリング戦略を導入し、最も有益なサンプルを効率的に選択することで手動によるアノテーションのコストを節約します。方法: 実験は、ベイジアン U-net に基づく BAL フレームワークによって、MRI および CT 画像の 2 つの下肢 (LE) データセットに対して実行されます。私たちの方法では、手動修正用に高密度で多様性のある不確かなサンプルを選択し、ラベルのないインスタンスとの類似性を最大限に高め、既存のトレーニング データとの類似性を最小限に抑えるように最適化します。 Dice とアノテーション削減コスト (RAC) と呼ばれる提案された指標をそれぞれ使用して、精度と効率を評価します。さらに、BAL のパフォーマンスに対するさまざまな取得ルールの影響を評価し、有効性を推定するためのアブレーション研究を設計します。結果: 提案された手法は、2 つの取得ルールにわたって両方のデータセットで他の手法に対して優位性または非劣性を示し、定量的な結果により取得ルールの長所と短所が明らかになりました。ボリュームワイズ収集におけるアブレーション研究では、筋骨格セグメンテーションにおいて密度と多様性の基準を組み合わせた方が、どちらか一方のみを使用した場合よりも優れたパフォーマンスを発揮することが示されています。結論: 私たちのサンプリング方法は、画像セグメンテーション タスクにおけるアノテーション コストの削減に効果的であることが証明されています。提案された方法と BAL フレームワークを組み合わせることで、医療画像データセットに効率的にアノテーションを付けるための半自動の方法が提供されます。
Purpose: Obtaining manual annotations to train deep learning (DL) models for auto-segmentation is often time-consuming. Uncertainty-based Bayesian active learning (BAL) is a widely-adopted method to reduce annotation efforts. Based on BAL, this study introduces a hybrid representation-enhanced sampling strategy that integrates density and diversity criteria to save manual annotation costs by efficiently selecting the most informative samples. Methods: The experiments are performed on two lower extremity (LE) datasets of MRI and CT images by a BAL framework based on Bayesian U-net. Our method selects uncertain samples with high density and diversity for manual revision, optimizing for maximal similarity to unlabeled instances and minimal similarity to existing training data. We assess the accuracy and efficiency using Dice and a proposed metric called reduced annotation cost (RAC), respectively. We further evaluate the impact of various acquisition rules on BAL performance and design an ablation study for effectiveness estimation. Results: The proposed method showed superiority or non-inferiority to other methods on both datasets across two acquisition rules, and quantitative results reveal the pros and cons of the acquisition rules. Our ablation study in volume-wise acquisition shows that the combination of density and diversity criteria outperforms solely using either of them in musculoskeletal segmentation. Conclusion: Our sampling method is proven efficient in reducing annotation costs in image segmentation tasks. The combination of the proposed method and our BAL framework provides a semi-automatic way for efficient annotation of medical image datasets.
updated: Wed Jul 26 2023 06:52:29 GMT+0000 (UTC)
published: Wed Jul 26 2023 06:52:29 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アソシエイト