arXiv reaDer
MS-KD: Multi-Organ Segmentation with Multiple Binary-Labeled Datasets
3D医用画像で複数の臓器に注釈を付けることは、時間とコストがかかります。一方、1つの特定の臓器に注釈が付けられた単一臓器のデータセットが多数存在します。このホワイトペーパーでは、バイナリラベルの付いたデータセットのセットを活用して多臓器セグメンテーションモデルを学習する方法を調査します。新規のマルチティーチャーシングルスチューデントナレッジディスティレーション(MS-KD)フレームワークが提案され、ティーチャーモデルは事前にトレーニングされたシングルオルガンセグメンテーションネットワークであり、スチューデントモデルはマルチオルガンセグメンテーションネットワークです。各教師が異なる器官に焦点を合わせていることを考慮して、ロジットごとの監視と機能ごとの監視からなる地域ベースの監視方法が提案されています。各教師は、教師が専門家と見なされる器官領域とすべての教師が同意する背景領域の2つの領域で生徒を監督します。 3つの公開された単一臓器データセットと多臓器データセットに関する広範な実験により、提案されたMS-KDフレームワークの有効性が実証されました。
Annotating multiple organs in 3D medical images is time-consuming and costly. Meanwhile, there exist many single-organ datasets with one specific organ annotated. This paper investigates how to learn a multi-organ segmentation model leveraging a set of binary-labeled datasets. A novel Multi-teacher Single-student Knowledge Distillation (MS-KD) framework is proposed, where the teacher models are pre-trained single-organ segmentation networks, and the student model is a multi-organ segmentation network. Considering that each teacher focuses on different organs, a region-based supervision method, consisting of logits-wise supervision and feature-wise supervision, is proposed. Each teacher supervises the student in two regions, the organ region where the teacher is considered as an expert and the background region where all teachers agree. Extensive experiments on three public single-organ datasets and a multi-organ dataset have demonstrated the effectiveness of the proposed MS-KD framework.
updated: Thu Aug 05 2021 12:29:26 GMT+0000 (UTC)
published: Thu Aug 05 2021 12:29:26 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト