arXiv reaDer
半教師あり腹部臓器セグメンテーションのためのハイブリッドアーキテクチャと疑似ラベルの組み合わせ
Combining Hybrid Architecture and Pseudo-label for Semi-supervised Abdominal Organ Segmentation
腹部臓器のセグメンテーションには、臓器の定量化、外科的計画、疾患診断など、多くの重要な臨床応用があります。ただし、CTスキャンから臓器に手動で注釈を付けることは、時間と労力を要します。半教師あり学習は、ラベルのない画像と限られたラベルの付いたサンプルの大規模なセットから学習することにより、この課題を軽減する可能性を示しています。この作業では、自己トレーニング戦略に従い、教師と生徒の両方のモデルにCNNとTransformerを備えたハイブリッドアーキテクチャ(PHTrans)を採用して、正確な疑似ラベルを生成します。その後、効率を維持しながらモデルのパフォーマンスと一般化能力を向上させるためのトレーニングのために、軽量PHTransを備えた2段階のセグメンテーションフレームワークにラベルデータとともにそれらを導入します。 FLARE2022の検証セットでの実験は、私たちの方法が優れたセグメンテーションパフォーマンスと高速で低リソースのモデル推論を達成することを示しています。平均DSCとHSDはそれぞれ0.8956と0.9316です。私たちの開発環境では、平均推論時間は18.62秒、平均最大GPUメモリは1995.04 MB、GPUメモリ時間曲線の下の領域とCPU使用率時間曲線の下の平均領域は23196.84と319.67です。
Abdominal organ segmentation has many important clinical applications, such as organ quantification, surgical planning, and disease diagnosis. However, manually annotating organs from CT scans is time-consuming and labor-intensive. Semi-supervised learning has shown the potential to alleviate this challenge by learning from a large set of unlabeled images and limited labeled samples. In this work, we follow the self-training strategy and employ a hybrid architecture (PHTrans) with CNN and Transformer for both teacher and student models to generate precise pseudo-labels. Afterward, we introduce them with label data together into a two-stage segmentation framework with lightweight PHTrans for training to improve the performance and generalization ability of the model while remaining efficient. Experiments on the validation set of FLARE2022 demonstrate that our method achieves excellent segmentation performance as well as fast and low-resource model inference. The average DSC and HSD are 0.8956 and 0.9316, respectively. Under our development environments, the average inference time is 18.62 s, the average maximum GPU memory is 1995.04 MB, and the area under the GPU memory-time curve and the average area under the CPU utilization-time curve are 23196.84 and 319.67.
updated: Sat Jul 23 2022 13:02:43 GMT+0000 (UTC)
published: Sat Jul 23 2022 13:02:43 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アソシエイト