arXiv reaDer
USCL:ビデオ対比表現学習による深部超音波画像診断モデルの事前トレーニング
USCL: Pretraining Deep Ultrasound Image Diagnosis Model through Video Contrastive Representation Learning
ほとんどの深部神経ネットワーク(DNN)ベースの超音波(US)医療画像分析モデルは、モデルの一般化を改善するために、事前にトレーニングされたバックボーン(ImageNetなど)を使用します。ただし、自然画像と医用画像の間のドメインのギャップは、避けられないパフォーマンスのボトルネックを引き起こします。この問題を軽減するために、US-4という名前のUSデータセットが、同じドメインで直接事前トレーニングするために構築されています。これには、4つの米国のビデオサブデータセットからの23,000を超える画像が含まれています。 US-4から堅牢な機能を学習するために、事前トレーニング用にUSCLという名前の米国の半教師あり対照学習法を提案します。ネガティブペア間の高い類似性を回避し、限られた米国のビデオから豊富な視覚的特徴をマイニングするために、USCLはサンプルペア生成方法を採用して、対照的な最適化の単一ステップに含まれる特徴を強化します。いくつかのダウンストリームタスクに関する広範な実験は、ImageNet事前トレーニングおよびその他の最先端(SOTA)事前トレーニングアプローチに対するUSCL事前トレーニングの優位性を示しています。特に、USCLの事前トレーニングされたバックボーンは、POCUSデータセットで94%を超える微調整精度を達成します。これは、ImageNetの事前トレーニングされたモデルの84%よりも10%高くなります。この作品のソースコードはhttps://github.com/983632847/USCLで入手できます。
Most deep neural networks (DNNs) based ultrasound (US) medical image analysis models use pretrained backbones (e.g., ImageNet) for better model generalization. However, the domain gap between natural and medical images causes an inevitable performance bottleneck. To alleviate this problem, an US dataset named US-4 is constructed for direct pretraining on the same domain. It contains over 23,000 images from four US video sub-datasets. To learn robust features from US-4, we propose an US semi-supervised contrastive learning method, named USCL, for pretraining. In order to avoid high similarities between negative pairs as well as mine abundant visual features from limited US videos, USCL adopts a sample pair generation method to enrich the feature involved in a single step of contrastive optimization. Extensive experiments on several downstream tasks show the superiority of USCL pretraining against ImageNet pretraining and other state-of-the-art (SOTA) pretraining approaches. In particular, USCL pretrained backbone achieves fine-tuning accuracy of over 94% on POCUS dataset, which is 10% higher than 84% of the ImageNet pretrained model. The source codes of this work are available at https://github.com/983632847/USCL.
updated: Thu Sep 02 2021 09:56:23 GMT+0000 (UTC)
published: Wed Nov 25 2020 23:44: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アソシエイト