arXiv reaDer
肺塞栓症注釈のための機能強化された敵対的半教師ありセマンティックセグメンテーションネットワーク
Feature-enhanced Adversarial Semi-supervised Semantic Segmentation Network for Pulmonary Embolism Annotation
この研究は、コンピューター断層撮影肺血管造影(CTPA)画像の肺塞栓症病変領域に自動的に注釈を付けるための機能強化された敵対的半監視セマンティックセグメンテーションモデルを確立しました。現在の研究では、すべてのPE CTPA画像セグメンテーション方法は、教師あり学習によってトレーニングされています。ただし、CTPA画像がさまざまな病院からのものである場合は、教師あり学習モデルを再トレーニングし、画像のラベルを変更する必要があります。この研究では、ラベルのない画像を少量追加することで、モデルをさまざまなデータセットに適用できるようにする半教師あり学習方法を提案しました。ラベル付き画像とラベルなし画像の両方を使用してモデルをトレーニングすることにより、ラベルなし画像の精度を向上させ、ラベル付けコストを削減できます。半教師ありセグメンテーションモデルには、セグメンテーションネットワークとディスクリミネーターネットワークが含まれます。セグメンテーションネットワークのエンコーダーから生成された特徴情報をディスクリミネーターに追加し、予測マスクとグラウンドトゥルースマスクの類似性を学習できるようにしました。このHRNetベースのアーキテクチャは、畳み込み演算の解像度を高く維持できるため、小さなPE病変領域の予測を改善できます。ラベル付きのオープンソースデータセットとラベルなしの国立成功大学病院(NCKUH)(IRB番号:B-ER-108-380)データセットを使用して、半教師あり学習モデルと、結果として得られる結合の平均交差(mIOU)をトレーニングしました。 )、ダイススコア、および感度は、NCKUHデータセットでそれぞれ0.3510、0.4854、および0.4253を達成しました。次に、中国医薬大学病院(CMUH)(IRB番号:CMUH110-REC3-173)データセットからの少量のラベルなしPE CTPA画像を使用して、モデルを微調整してテストしました。半教師ありモデルの結果を教師ありモデルと比較すると、mIOU、ダイススコア、および感度は、それぞれ0.2344、0.3325、および0.3151から0.3721、0.5113、および0.4967に向上しました。
This study established a feature-enhanced adversarial semi-supervised semantic segmentation model to automatically annotate pulmonary embolism lesion areas in computed tomography pulmonary angiogram (CTPA) images. In current studies, all of the PE CTPA image segmentation methods are trained by supervised learning. However, the supervised learning models need to be retrained and the images need to be relabeled when the CTPA images come from different hospitals. This study proposed a semi-supervised learning method to make the model applicable to different datasets by adding a small amount of unlabeled images. By training the model with both labeled and unlabeled images, the accuracy of unlabeled images can be improved and the labeling cost can be reduced. Our semi-supervised segmentation model includes a segmentation network and a discriminator network. We added feature information generated from the encoder of segmentation network to the discriminator so that it can learn the similarity between predicted mask and ground truth mask. This HRNet-based architecture can maintain a higher resolution for convolutional operations so the prediction of small PE lesion areas can be improved. We used the labeled open-source dataset and the unlabeled National Cheng Kung University Hospital (NCKUH) (IRB number: B-ER-108-380) dataset to train the semi-supervised learning model, and the resulting mean intersection over union (mIOU), dice score, and sensitivity achieved 0.3510, 0.4854, and 0.4253, respectively on the NCKUH dataset. Then, we fine-tuned and tested the model with a small amount of unlabeled PE CTPA images from China Medical University Hospital (CMUH) (IRB number: CMUH110-REC3-173) dataset. Comparing the results of our semi-supervised model with the supervised model, the mIOU, dice score, and sensitivity improved from 0.2344, 0.3325, and 0.3151 to 0.3721, 0.5113, and 0.4967, respectively.
updated: Fri Apr 08 2022 04:21:02 GMT+0000 (UTC)
published: Fri Apr 08 2022 04:21:02 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アソシエイト