arXiv reaDer
ウェーブレットベースの正規化フローによる効率的な黒色腫の分布外検出
Efficient Out-of-Distribution Detection of Melanoma with Wavelet-based Normalizing Flows
メラノーマは、後期段階での死亡率が高い皮膚がんの深刻な形態です。幸いなことに、早期に検出された場合、黒色腫の予後は有望であり、悪性黒色腫の発生率は比較的低いです。その結果、データセットは非常に不均衡になり、現在の最先端の教師付き分類 AI モデルのトレーニングが複雑になります。生成モデルを使用して良性のデータ分布を学習し、密度推定により分布外 (OOD) の悪性画像を検出することを提案します。正規化フロー (NF) は、正確な可能性を計算できるため、OOD 検出の理想的な候補です。それにもかかわらず、セマンティック コンテキストではなく、見かけのグラフィック機能に対する誘導バイアスが、正確な OOD 検出を妨げています。この作業では、悪性画像の尤度ベースの OOD 検出を改善するために、メラノーマのドメインレベルの知識でこれらのバイアスを使用することを目指しています。私たちの有望な結果は、NFを使用した黒色腫のOOD検出の可能性を示しています。ウェーブレットベースの NF を使用することで、受信者動作特性の曲線下面積が 9% 増加しました。このモデルでは、推論に必要なパラメーターが大幅に少なくなり、エッジ デバイスでより適用しやすくなります。提案された方法論は、医療専門家による皮膚がん患者の診断を支援し、生存率を継続的に高めることができます。さらに、この研究は、同様のデータの不均衡の問題を抱えた腫瘍学の他の分野への道を開きます。
Melanoma is a serious form of skin cancer with high mortality rate at later stages. Fortunately, when detected early, the prognosis of melanoma is promising and malignant melanoma incidence rates are relatively low. As a result, datasets are heavily imbalanced which complicates training current state-of-the-art supervised classification AI models. We propose to use generative models to learn the benign data distribution and detect Out-of-Distribution (OOD) malignant images through density estimation. Normalizing Flows (NFs) are ideal candidates for OOD detection due to their ability to compute exact likelihoods. Nevertheless, their inductive biases towards apparent graphical features rather than semantic context hamper accurate OOD detection. In this work, we aim at using these biases with domain-level knowledge of melanoma, to improve likelihood-based OOD detection of malignant images. Our encouraging results demonstrate potential for OOD detection of melanoma using NFs. We achieve a 9% increase in Area Under Curve of the Receiver Operating Characteristics by using wavelet-based NFs. This model requires significantly less parameters for inference making it more applicable on edge devices. The proposed methodology can aid medical experts with diagnosis of skin-cancer patients and continuously increase survival rates. Furthermore, this research paves the way for other areas in oncology with similar data imbalance issuesCode available at: https://github.com/A-Vzer/WaveletFlowPytorch
updated: Tue Aug 09 2022 09:57:56 GMT+0000 (UTC)
published: Tue Aug 09 2022 09:57:56 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アソシエイト