畳み込みニューラルネットワーク(CNN)は、強力な医用画像セグメンテーションモデルであることが示されています。この研究では、これらのモデルに関する未解決の主な問題のいくつかに取り組みます。具体的には、小さな医療画像データセットでこれらのモデルをトレーニングすることは依然として困難であり、転移学習などの手法を促進する多くの研究があります。さらに、これらのモデルは、自信過剰の予測を生成したり、テスト時に配布外(OOD)データが提示されたときに静かに失敗したりすることで悪名高くなっています。この論文では、マルチタスク学習、つまり、複数の異なるデータセットで単一のモデルをトレーニングし、いくつかの異なる臓器と異なる画像モダリティにまたがることを提唱します。 1つのCNNがコンテキストを自動的に認識し、関心のある臓器を正確にセグメント化することを学習するだけでなく、そのような共同モデルは、各データセットで個別にトレーニングされた専用モデルよりも多くの場合、予測がより正確で優れていることを示しています。私たちの実験は、マルチタスク学習が医用画像セグメンテーションタスクの転移学習よりも優れていることを示しています。 OODデータを検出するために、CNNフィーチャマップのスペクトル分析に基づく方法を提案します。テスト画像がモデルのトレーニングに使用された画像と類似しているかどうかを識別するために使用できる、異なるイメージングモダリティや関心のある異なる臓器を表す異なるデータセットが異なるスペクトルシグネチャを持っていることを示します。このアプローチは、予測の不確実性に基づくOOD検出よりもはるかに正確であることを示しています。このホワイトペーパーで提案されている方法は、CNNベースの医用画像セグメンテーションモデルの精度と信頼性の向上に大きく貢献しています。
Convolutional Neural Networks (CNNs) have shown to be powerful medical image segmentation models. In this study, we address some of the main unresolved issues regarding these models. Specifically, training of these models on small medical image datasets is still challenging, with many studies promoting techniques such as transfer learning. Moreover, these models are infamous for producing over-confident predictions and for failing silently when presented with out-of-distribution (OOD) data at test time. In this paper, we advocate for multi-task learning, i.e., training a single model on several different datasets, spanning several different organs of interest and different imaging modalities. We show that not only a single CNN learns to automatically recognize the context and accurately segment the organ of interest in each context, but also that such a joint model often has more accurate and better-calibrated predictions than dedicated models trained separately on each dataset. Our experiments show that multi-task learning can outperform transfer learning in medical image segmentation tasks. For detecting OOD data, we propose a method based on spectral analysis of CNN feature maps. We show that different datasets, representing different imaging modalities and/or different organs of interest, have distinct spectral signatures, which can be used to identify whether or not a test image is similar to the images used to train a model. We show that this approach is far more accurate than OOD detection based on prediction uncertainty. The methods proposed in this paper contribute significantly to improving the accuracy and reliability of CNN-based medical image segmentation models.