arXiv reaDer
CNNとTransformer間のクロスティーチングによる半教師あり医療画像セグメンテーション
Semi-Supervised Medical Image Segmentation via Cross Teaching between CNN and Transformer
最近、畳み込みニューラルネットワーク(CNN)とトランスフォーマーを使用した深層学習により、完全に監視された医療画像セグメンテーションで有望な結果が示されました。ただし、トレーニング用の注釈を制限して良好なパフォーマンスを達成することは、依然として困難です。この作業では、CNNとTransformerのクロスティーチングを導入することにより、半教師あり医療画像セグメンテーションのための非常にシンプルで効率的なフレームワークを紹介します。具体的には、一貫性の正則化からクロスティーチングまでの古典的なディープコトレーニングを簡素化します。この場合、ネットワークの予測は、他のネットワークを直接エンドツーエンドで監視するための疑似ラベルとして使用されます。 CNNとTransformerの学習パラダイムの違いを考慮して、CNNを使用するだけでなく、CNNとTransformerのクロスティーチングを紹介します。公開ベンチマークでの実験は、私たちの方法が、より単純なフレームワークだけで、8つの既存の半教師あり学習方法よりも優れていることを示しています。特に、この作業は、半教師あり医療画像セグメンテーションのためにCNNとトランスフォーマーを組み合わせて、公開ベンチマークで有望な結果を達成する最初の試みである可能性があります。コードはhttps://github.com/HiLab-git/SSL4MISでリリースされます。
Recently, deep learning with Convolutional Neural Networks (CNNs) and Transformers has shown encouraging results in fully supervised medical image segmentation. However, it is still challenging for them to achieve good performance with limited annotations for training. In this work, we present a very simple yet efficient framework for semi-supervised medical image segmentation by introducing the cross teaching between CNN and Transformer. Specifically, we simplify the classical deep co-training from consistency regularization to cross teaching, where the prediction of a network is used as the pseudo label to supervise the other network directly end-to-end. Considering the difference in learning paradigm between CNN and Transformer, we introduce the Cross Teaching between CNN and Transformer rather than just using CNNs. Experiments on a public benchmark show that our method outperforms eight existing semi-supervised learning methods just with a simpler framework. Notably, this work may be the first attempt to combine CNN and transformer for semi-supervised medical image segmentation and achieve promising results on a public benchmark. The code will be released at: https://github.com/HiLab-git/SSL4MIS.
updated: Thu Dec 09 2021 13:22:38 GMT+0000 (UTC)
published: Thu Dec 09 2021 13:22: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アソシエイト