arXiv reaDer
教師なしドメイン適応によるAIベースの結核スクリーニングにおけるドメインシフトの緩和
Mitigating domain shift in AI-based tuberculosis screening with unsupervised domain adaptation
ドメイン不変特徴学習(DIFL)が、深層学習結核スクリーニングアルゴリズムのドメイン外の一般化可能性を改善できることを示します。最先端の深層学習アルゴリズムでは、「ドメインシフト」が原因で、目に見えないデータ分布に一般化することが難しい場合が多いことはよく知られています。医用画像のコンテキストでは、これは、ある患者集団から別の患者集団に一般化できないなど、意図しないバイアスにつながる可能性があります。地理的に多様な画像ソースを持つ4つの最も人気のある公開データセットを使用して、結核スクリーニングの目的でResNet-50分類器のパフォーマンスを分析します。ドメイン適応がないと、ResNet-50は、地理的に分散した地域からの画像を使用して、多数の公開結核スクリーニングデータセットからの画像分布間で一般化することが困難であることを示します。ただし、DIFLを組み込むことで、ドメイン外のパフォーマンスが大幅に向上します。分析基準には、ベースラインとDIFL拡張アルゴリズムの両方にわたる精度、感度、特異性、およびAUCの比較が含まれます。 DIFLは、さまざまな公開データセットに適用された場合に、ソースドメイン画像全体で許容可能な精度を維持しながら、結核スクリーニングの一般化可能性を向上させると結論付けています。
We demonstrate that Domain Invariant Feature Learning (DIFL) can improve the out-of-domain generalizability of a deep learning Tuberculosis screening algorithm. It is well known that state of the art deep learning algorithms often have difficulty generalizing to unseen data distributions due to "domain shift". In the context of medical imaging, this could lead to unintended biases such as the inability to generalize from one patient population to another. We analyze the performance of a ResNet-50 classifier for the purposes of Tuberculosis screening using the four most popular public datasets with geographically diverse sources of imagery. We show that without domain adaptation, ResNet-50 has difficulty in generalizing between imaging distributions from a number of public Tuberculosis screening datasets with imagery from geographically distributed regions. However, with the incorporation of DIFL, the out-of-domain performance is greatly enhanced. Analysis criteria includes a comparison of accuracy, sensitivity, specificity and AUC over both the baseline, as well as the DIFL enhanced algorithms. We conclude that DIFL improves generalizability of Tuberculosis screening while maintaining acceptable accuracy over the source domain imagery when applied across a variety of public datasets.
updated: Tue Nov 09 2021 00:45:52 GMT+0000 (UTC)
published: Tue Nov 09 2021 00:45:52 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アソシエイト