arXiv reaDer
血管混合下での一貫した事後分布:クロスドメイン網膜動脈/静脈分類の正則化
Consistent Posterior Distributions under Vessel-Mixing: A Regularization for Cross-Domain Retinal Artery/Vein Classification
網膜動脈/静脈(A / V)分類は、糖尿病と心血管疾患を診断するための重要な手法です。ディープラーニングベースの方法はA / V分類で印象的な結果を達成しますが、たとえばイメージングプロトコルのバリエーションによって引き起こされるドメインシフトのために、別のデータベースに直接適用すると、通常、パフォーマンスが大幅に低下します。この論文では、網膜A / V分類におけるクロスドメイン学習のために、新しい血管混合ベースの整合性正則化フレームワークを提案します。特に、ラベルのスムーズな事前に基づいて、ソースドメインへの深刻なバイアスを軽減するために、モデルは正則化され、摂動下にあるラベルのないターゲットドメイン入力に対して一貫した予測を提供します。この一貫性の正則化は、モデルと摂動が互いに反対であるメカニズムを暗黙的に導入します。このメカニズムでは、モデルは摂動に対処するのに十分な堅牢性を持つようにプッシュされます。したがって、血管混合摂動と呼ばれる網膜A / Vのシナリオで、モデルの堅牢性をさらに刺激するために、より困難な相手を調査します。特に、2つの画像を局所的に混合することにより、眼底画像、特に血管構造を効果的に乱します。さまざまな機関や画像機器によって収集された4つの公開データセットを使用して、クロスドメインA / V分類に関する広範な実験を実施しています。結果は、私たちの方法が最先端のクロスドメインパフォーマンスを達成していることを示しています。これは、ターゲットドメインでの完全な教師あり学習によって得られる上限にも近いものです。
Retinal artery/vein (A/V) classification is a critical technique for diagnosing diabetes and cardiovascular diseases. Although deep learning based methods achieve impressive results in A/V classification, their performances usually degrade severely when being directly applied to another database, due to the domain shift, e.g., caused by the variations in imaging protocols. In this paper, we propose a novel vessel-mixing based consistency regularization framework, for cross-domain learning in retinal A/V classification. Specially, to alleviate the severe bias to source domain, based on the label smooth prior, the model is regularized to give consistent predictions for unlabeled target-domain inputs that are under perturbation. This consistency regularization implicitly introduces a mechanism where the model and the perturbation is opponent to each other, where the model is pushed to be robust enough to cope with the perturbation. Thus, we investigate a more difficult opponent to further inspire the robustness of model, in the scenario of retinal A/V, called vessel-mixing perturbation. Specially, it effectively disturbs the fundus images especially the vessel structures by mixing two images regionally. We conduct extensive experiments on cross-domain A/V classification using four public datasets, which are collected by diverse institutions and imaging devices. The results demonstrate that our method achieves the state-of-the-art cross-domain performance, which is also close to the upper bound obtained by fully supervised learning on target domain.
updated: Fri Jun 18 2021 12:55:59 GMT+0000 (UTC)
published: Tue Mar 16 2021 14:18:35 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アソシエイト