arXiv reaDer
適応可能なタスクベースの優先順位付けを使用したアクティブ ラーニング
Active learning using adaptable task-based prioritisation
教師あり機械学習ベースの医用画像コンピューティング アプリケーションでは、専門家によるラベルのキュレーションが必要ですが、ラベルのない画像データは比較的豊富にある可能性があります。アクティブな学習方法は、ラベル効率の高いモデル トレーニングのために、エキスパート アノテーション用に利用可能な画像データのサブセットに優先順位を付けることを目的としています。マルチクラス セグメンテーション タスクのために、バッチ モードの能動学習のように、一連のバッチで画像の優先度を測定するコントローラー ニューラル ネットワークを開発します。コントローラーは、タスク予測子も最適化するマルコフ決定プロセス (MDP) 環境内で、正のタスク固有のパフォーマンス向上に報酬を与えることによって最適化されます。この作業では、タスク予測子はセグメンテーション ネットワークです。複数の MDP を使用したメタ強化学習アルゴリズムが提案されているため、事前にトレーニングされたコントローラーを、さまざまな機関からのデータを含む新しい MDP に適応させたり、腹部内のさまざまな臓器や構造のセグメンテーションを必要としたりすることができます。 1,000 人以上の患者からの複数の CT データセットを使用した実験結果を提示し、9 つの異なる腹部臓器のセグメンテーション タスクを使用して、学習した優先順位付けコントローラー機能とその機関間および臓器間の適応性の有効性を実証します。提案された適応可能な優先順位付けメトリックは、他のヒューリスティックまたはランダムな優先順位付けメトリックで必要とされるラベルの約 40% から 60% を使用して、トレーニングでは見られない新しいクラスの腎臓のセグメンテーション精度を収束させることを示します。限られたサイズの臨床データセットの場合、提案された適応可能な優先順位付けは、ランダムな優先順位付けおよび代替のアクティブサンプリング戦略と比較して、腎臓および肝臓の血管セグメンテーションのタスクで、それぞれダイススコアで 22.6% および 10.2% のパフォーマンス向上を提供します。
Supervised machine learning-based medical image computing applications necessitate expert label curation, while unlabelled image data might be relatively abundant. Active learning methods aim to prioritise a subset of available image data for expert annotation, for label-efficient model training. We develop a controller neural network that measures priority of images in a sequence of batches, as in batch-mode active learning, for multi-class segmentation tasks. The controller is optimised by rewarding positive task-specific performance gain, within a Markov decision process (MDP) environment that also optimises the task predictor. In this work, the task predictor is a segmentation network. A meta-reinforcement learning algorithm is proposed with multiple MDPs, such that the pre-trained controller can be adapted to a new MDP that contains data from different institutes and/or requires segmentation of different organs or structures within the abdomen. We present experimental results using multiple CT datasets from more than one thousand patients, with segmentation tasks of nine different abdominal organs, to demonstrate the efficacy of the learnt prioritisation controller function and its cross-institute and cross-organ adaptability. We show that the proposed adaptable prioritisation metric yields converging segmentation accuracy for the novel class of kidney, unseen in training, using between approximately 40% to 60% of labels otherwise required with other heuristic or random prioritisation metrics. For clinical datasets of limited size, the proposed adaptable prioritisation offers a performance improvement of 22.6% and 10.2% in Dice score, for tasks of kidney and liver vessel segmentation, respectively, compared to random prioritisation and alternative active sampling strategies.
updated: Sat Dec 03 2022 22:37:38 GMT+0000 (UTC)
published: Sat Dec 03 2022 22:37:38 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アソシエイト