arXiv reaDer
TAAL: 医用画像セグメンテーションにおけるアクティブ ラーニングのためのテスト時間増強
TAAL: Test-time Augmentation for Active Learning in Medical Image Segmentation
深層学習の方法は通常、ラベル付きデータの可用性に依存しますが、これは取得に費用と時間がかかります。アクティブ ラーニングは、タスク モデルのパフォーマンスを最大化するために、どのサンプルに注釈を付けるのが最適かを優先することで、このような作業に対処します。アクティブ ラーニングのフレームワークは、自然画像の分類のコンテキストで広く調査されていますが、医療画像のセグメンテーションではまばらにしか使用されていません。課題は、注釈に最適な候補データを明らかにする不確実性の尺度を取得することにあります。この論文では、データ変換によって提供される不確実性情報を活用するセグメンテーションのための新しい半教師付きアクティブ学習アプローチであるアクティブ学習のテスト時間増強 (TAAL) を提案します。私たちの方法は、トレーニングと推論中にクロスオーグメンテーションの一貫性を適用して、半教師ありの方法でモデル学習を改善し、次に注釈を付ける最も関連性の高いラベルのないサンプルを特定します。さらに、一貫性の喪失は、モデルのパフォーマンスをさらに向上させるために、修正されたバージョンの JSD を使用します。不確実性に基づく戦略で一般的に使用される外部モジュールや単純なヒューリスティックではなく、データ変換に依存することで、TAAL は医療分野に適用可能なシンプルでありながら強力な、タスクに依存しない半教師付き能動学習アプローチとして浮上します。公開されている心臓画像のデータセットに関する我々の結果は、TAAL が完全教師あり設定と半教師あり設定の両方で既存のベースライン方法よりも優れていることを示しています。私たちの実装は、https://github.com/melinphd/TAAL で公開されています。
Deep learning methods typically depend on the availability of labeled data, which is expensive and time-consuming to obtain. Active learning addresses such effort by prioritizing which samples are best to annotate in order to maximize the performance of the task model. While frameworks for active learning have been widely explored in the context of classification of natural images, they have been only sparsely used in medical image segmentation. The challenge resides in obtaining an uncertainty measure that reveals the best candidate data for annotation. This paper proposes Test-time Augmentation for Active Learning (TAAL), a novel semi-supervised active learning approach for segmentation that exploits the uncertainty information offered by data transformations. Our method applies cross-augmentation consistency during training and inference to both improve model learning in a semi-supervised fashion and identify the most relevant unlabeled samples to annotate next. In addition, our consistency loss uses a modified version of the JSD to further improve model performance. By relying on data transformations rather than on external modules or simple heuristics typically used in uncertainty-based strategies, TAAL emerges as a simple, yet powerful task-agnostic semi-supervised active learning approach applicable to the medical domain. Our results on a publicly-available dataset of cardiac images show that TAAL outperforms existing baseline methods in both fully-supervised and semi-supervised settings. Our implementation is publicly available on https://github.com/melinphd/TAAL.
updated: Mon Jan 16 2023 22:19:41 GMT+0000 (UTC)
published: Mon Jan 16 2023 22:19:41 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アソシエイト