arXiv reaDer
網膜画像の生成とセグメンテーションのためのノイズ除去拡散確率モデル
Denoising Diffusion Probabilistic Model for Retinal Image Generation and Segmentation
専門家は網膜画像と血管樹を使用して、さまざまな目、血液循環、脳関連の病気を検出および診断します。ただし、網膜画像の手動セグメンテーションは時間のかかるプロセスであり、高度な専門知識が必要であり、プライバシーの問題により困難です。画像をセグメント化するために多くの方法が提案されていますが、大規模な網膜画像データセットの必要性により、これらの方法のパフォーマンスは制限されます。いくつかの方法では、敵対的生成ネットワーク (GAN) に基づいて深層学習モデルを合成し、限られた種類のサンプルを生成します。この論文では、画像合成において GAN を上回る新しいノイズ除去拡散確率モデル (DDPM) を提案します。私たちは、網膜画像、対応する血管ツリー、および ReTree データセットの画像でトレーニングされた DDPM に基づくセグメンテーション ネットワークで構成される Retinal Trees (ReTree) データセットを開発しました。最初の段階では、標準正規分布に属する乱数から血管ツリーを生成する 2 段階の DDPM を開発します。その後、モデルは、指定された血管ツリーとランダムな分布から眼底画像を生成するように誘導されます。提案されたデータセットは定量的および定性的に評価されています。定量的な評価指標には、Frechet Inception Distance (FID) スコア、Jaccard 類似性係数、Cohen のカッパ、Matthew の相関係数 (MCC)、精度、再現率、F1 スコア、精度が含まれます。データセットの効率を検証するために合成データを使用して血管セグメンテーション モデルをトレーニングし、本物のデータでテストしました。私たちが開発したデータセットとソースコードは、https://github.com/AAleka/retree で入手できます。
Experts use retinal images and vessel trees to detect and diagnose various eye, blood circulation, and brain-related diseases. However, manual segmentation of retinal images is a time-consuming process that requires high expertise and is difficult due to privacy issues. Many methods have been proposed to segment images, but the need for large retinal image datasets limits the performance of these methods. Several methods synthesize deep learning models based on Generative Adversarial Networks (GAN) to generate limited sample varieties. This paper proposes a novel Denoising Diffusion Probabilistic Model (DDPM) that outperformed GANs in image synthesis. We developed a Retinal Trees (ReTree) dataset consisting of retinal images, corresponding vessel trees, and a segmentation network based on DDPM trained with images from the ReTree dataset. In the first stage, we develop a two-stage DDPM that generates vessel trees from random numbers belonging to a standard normal distribution. Later, the model is guided to generate fundus images from given vessel trees and random distribution. The proposed dataset has been evaluated quantitatively and qualitatively. Quantitative evaluation metrics include Frechet Inception Distance (FID) score, Jaccard similarity coefficient, Cohen's kappa, Matthew's Correlation Coefficient (MCC), precision, recall, F1-score, and accuracy. We trained the vessel segmentation model with synthetic data to validate our dataset's efficiency and tested it on authentic data. Our developed dataset and source code is available at https://github.com/AAleka/retree.
updated: Wed Aug 16 2023 13:01:13 GMT+0000 (UTC)
published: Wed Aug 16 2023 13:01:13 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アソシエイト