arXiv reaDer
Unsupervised Domain Adaptation for Retinal Vessel Segmentation with Adversarial Learning and Transfer Normalization
網膜血管のセグメンテーションは、さまざまな心血管疾患および眼科疾患のコンピューター支援スクリーニング、診断、および治療において重要な役割を果たします。最近、深層学習ベースの網膜血管セグメンテーションアルゴリズムが驚くべきパフォーマンスを達成しました。ただし、ドメインシフトの問題により、これらのアルゴリズムは、トレーニングデータとは異なる新しいデータに適用されると、パフォーマンスが低下することがよくあります。テストドメインごとに新しいデータに手動でラベルを付けることは、多くの場合、時間と労力を要する作業です。この作業では、エントロピーベースの敵対的学習と転送正規化レイヤーを使用して、ドメイン間で十分に一般化され、ターゲットドメインの注釈を必要としないセグメンテーションネットワークをトレーニングすることにより、網膜血管セグメンテーションにおける教師なしドメイン適応を調査します。具体的には、最初に、エントロピーベースの敵対的学習戦略を開発して、ソースドメインとターゲットドメイン間の分布の不一致を減らし、ターゲットドメインでのエントロピー最小化の目的も達成します。さらに、ディープネットワークの転送可能性をさらに高めるために、新しい転送正規化層が提案されています。各ドメインの機能を個別に正規化して、ドメインの分散ギャップを補正します。さらに、ドメイン間でより転送可能な機能チャネルを適応的に選択するため、ネットワークの一般化パフォーマンスがさらに向上します。 3つの通常の眼底画像データセットと超広視野眼底画像データセットで広範な実験を実施しました。その結果は、私たちのアプローチが他の最先端の方法と比較して大幅なパフォーマンスの向上をもたらすことを示しています。
Retinal vessel segmentation plays a key role in computer-aided screening, diagnosis, and treatment of various cardiovascular and ophthalmic diseases. Recently, deep learning-based retinal vessel segmentation algorithms have achieved remarkable performance. However, due to the domain shift problem, the performance of these algorithms often degrades when they are applied to new data that is different from the training data. Manually labeling new data for each test domain is often a time-consuming and laborious task. In this work, we explore unsupervised domain adaptation in retinal vessel segmentation by using entropy-based adversarial learning and transfer normalization layer to train a segmentation network, which generalizes well across domains and requires no annotation of the target domain. Specifically, first, an entropy-based adversarial learning strategy is developed to reduce the distribution discrepancy between the source and target domains while also achieving the objective of entropy minimization on the target domain. In addition, a new transfer normalization layer is proposed to further boost the transferability of the deep network. It normalizes the features of each domain separately to compensate for the domain distribution gap. Besides, it also adaptively selects those feature channels that are more transferable between domains, thus further enhancing the generalization performance of the network. We conducted extensive experiments on three regular fundus image datasets and an ultra-widefield fundus image dataset, and the results show that our approach yields significant performance gains compared to other state-of-the-art methods.
updated: Wed Aug 04 2021 02:45:37 GMT+0000 (UTC)
published: Wed Aug 04 2021 02:45:37 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト