arXiv reaDer
医療画像セグメンテーションのための数ショット設定での暗黙の勾配によるメタ学習
Meta-learning with implicit gradients in a few-shot setting for medical image segmentation
広く使用されている従来の教師あり深層学習方法では、多数のトレーニングサンプルが必要ですが、多くの場合、目に見えないデータセットで一般化できません。したがって、訓練されたモデルのより一般的なアプリケーションは、臨床診療用の医用画像処理にはかなり限定されています。固有の病変カテゴリまたは固有の患者集団ごとに個別にトレーニングされたモデルを使用するには、十分に大きなキュレーションされたデータセットが必要になります。これは、実際の臨床設定で使用するには実用的ではありません。数ショットの学習アプローチは、労働集約的で高価な膨大な数の信頼できるグラウンドトゥルースラベルの必要性を最小限に抑えるだけでなく、新しい母集団からのデータセットをモデル化するためにも使用できます。この目的のために、医療画像セグメンテーションの数ショット設定の下で、最適化ベースの暗黙的なモデルにとらわれないメタ学習(iMAML)アルゴリズムを活用することを提案します。私たちのアプローチでは、多様であるが小さいトレーニングサンプルから学習した重みを活用して、見えないデータセットの分析を高精度で実行できます。従来の数ショット学習アプローチとは異なり、私たちの方法は一般化能力を向上させることを示します。私たちの知る限り、これは医療画像のセグメンテーションにiMAMLを活用し、病変データセットの一意のインスタンスと混合インスタンスでのメタトレーニングなどのシナリオでモデルの強度を調査する最初の作業です。公開されている皮膚とポリープのデータセットに関する定量的な結果は、提案された方法が、ナイーブな監視ベースラインモデルと最近の2つのショットセグメンテーションアプローチを大幅に上回っていることを示しています。さらに、私たちのiMAMLアプローチは、ほとんどの実験で、対応するMAMLと比較してダイススコアが2%〜4%向上していることを示しています。
Widely used traditional supervised deep learning methods require a large number of training samples but often fail to generalize on unseen datasets. Therefore, a more general application of any trained model is quite limited for medical imaging for clinical practice. Using separately trained models for each unique lesion category or a unique patient population will require sufficiently large curated datasets, which is not practical to use in a real-world clinical set-up. Few-shot learning approaches can not only minimize the need for an enormous number of reliable ground truth labels that are labour-intensive and expensive but can also be used to model on a dataset coming from a new population. To this end, we propose to exploit an optimization-based implicit model agnostic meta-learning (iMAML) algorithm under few-shot settings for medical image segmentation. Our approach can leverage the learned weights from diverse but small training samples to perform analysis on unseen datasets with high accuracy. We show that, unlike classical few-shot learning approaches, our method improves generalization capability. To our knowledge, this is the first work that exploits iMAML for medical image segmentation and explores the strength of the model on scenarios such as meta-training on unique and mixed instances of lesion datasets. Our quantitative results on publicly available skin and polyp datasets show that the proposed method outperforms the naive supervised baseline model and two recent few-shot segmentation approaches by large margins. In addition, our iMAML approach shows an improvement of 2%-4% in dice score compared to its counterpart MAML for most experiments.
updated: Mon Jan 31 2022 03:42:44 GMT+0000 (UTC)
published: Sun Jun 06 2021 19:52: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アソシエイト