arXiv reaDer
MetaMedSeg:少数ショットの臓器セグメンテーションのための体積メタ学習
MetaMedSeg: Volumetric Meta-learning for Few-Shot Organ Segmentation
十分な注釈付き画像データの欠如は、医療画像セグメンテーションの一般的な問題です。一部の臓器と密度では、注釈が不足しているため、モデルトレーニングの収束が不十分になる可能性がありますが、他の臓器には注釈付きのデータがたくさんあります。この作業では、スライス間の多様性をキャプチャすることを目的として、体積医療データのメタ学習タスクを再定義する勾配ベースのメタ学習アルゴリズムであるMetaMedSegを紹介します。また、勾配集約のさまざまな重み付けスキームについても検討し、タスクごとに複雑さが異なる可能性があるため、初期化への寄与が異なる可能性があると主張します。モデルをトレーニングするために、重要度を意識した重み付けスキームを提案します。実験では、さまざまな臓器のCTおよびMRIボリュームから2Dスライスを抽出し、セマンティックセグメンテーションを実行することにより、医療用十種競技データセットの評価を示します。結果は、提案されたボリュームタスク定義が、関連するベースラインと比較して、IoUに関して最大30%の改善につながることを示しています。提案された更新ルールは、ターゲット臓器のデータ分布がソース臓器と大きく異なる複雑なシナリオのパフォーマンスを改善することも示されています。
The lack of sufficient annotated image data is a common issue in medical image segmentation. For some organs and densities, the annotation may be scarce, leading to poor model training convergence, while other organs have plenty of annotated data. In this work, we present MetaMedSeg, a gradient-based meta-learning algorithm that redefines the meta-learning task for the volumetric medical data with the goal to capture the variety between the slices. We also explore different weighting schemes for gradients aggregation, arguing that different tasks might have different complexity, and hence, contribute differently to the initialization. We propose an importance-aware weighting scheme to train our model. In the experiments, we present an evaluation of the medical decathlon dataset by extracting 2D slices from CT and MRI volumes of different organs and performing semantic segmentation. The results show that our proposed volumetric task definition leads to up to 30% improvement in terms of IoU compared to related baselines. The proposed update rule is also shown to improve the performance for complex scenarios where the data distribution of the target organ is very different from the source organs.
updated: Sat Sep 18 2021 11:13:45 GMT+0000 (UTC)
published: Sat Sep 18 2021 11:13:45 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アソシエイト