arXiv reaDer
DiffMIC: 医用画像分類のためのデュアル ガイダンス拡散ネットワーク
DiffMIC: Dual-Guidance Diffusion Network for Medical Image Classification
拡散確率モデルは、最近、生成画像モデリングで目覚ましいパフォーマンスを発揮し、コンピューター ビジョン コミュニティで大きな注目を集めています。ただし、かなりの量の拡散ベースの研究が生成タスクに焦点を当てている一方で、一般的な医療画像分類に拡散モデルを適用した研究はほとんどありません。この論文では、医用画像の予期しないノイズと摂動を排除し、意味表現をロバストにキャプチャすることにより、一般的な医用画像分類に対処する最初の拡散ベースのモデル (DiffMIC と呼ばれる) を提案します。この目標を達成するために、各拡散ステップを複数の粒度で条件付けて、段階的な地域の注意を向上させるデュアル条件付きガイダンス戦略を考案します。さらに、拡散フォワードプロセス中に最大平均不一致正則化を適用することにより、各粒度の相互情報を学習することを提案します。超音波画像での胎盤成熟度のグレーディング、皮膚鏡画像を使用した皮膚病変の分類、眼底画像を使用した糖尿病性網膜症のグレーディングなど、異なる画像モダリティを使用した 3 つの医療分類タスクに対する DiffMIC の有効性を評価します。私たちの実験結果は、DiffMICが最先端の方法よりも大幅に優れていることを示しており、提案されたモデルの普遍性と有効性を示しています。私たちのコードは、https://github.com/scott-yjyang/DiffMIC で公開されます。
Diffusion Probabilistic Models have recently shown remarkable performance in generative image modeling, attracting significant attention in the computer vision community. However, while a substantial amount of diffusion-based research has focused on generative tasks, few studies have applied diffusion models to general medical image classification. In this paper, we propose the first diffusion-based model (named DiffMIC) to address general medical image classification by eliminating unexpected noise and perturbations in medical images and robustly capturing semantic representation. To achieve this goal, we devise a dual conditional guidance strategy that conditions each diffusion step with multiple granularities to improve step-wise regional attention. Furthermore, we propose learning the mutual information in each granularity by enforcing Maximum-Mean Discrepancy regularization during the diffusion forward process. We evaluate the effectiveness of our DiffMIC on three medical classification tasks with different image modalities, including placental maturity grading on ultrasound images, skin lesion classification using dermatoscopic images, and diabetic retinopathy grading using fundus images. Our experimental results demonstrate that DiffMIC outperforms state-of-the-art methods by a significant margin, indicating the universality and effectiveness of the proposed model. Our code will be publicly available at https://github.com/scott-yjyang/DiffMIC.
updated: Thu Mar 23 2023 06:12:48 GMT+0000 (UTC)
published: Sun Mar 19 2023 09:15:45 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アソシエイト