GAN 기반 의료영상 생성 모델에 대한 품질 및 다양성 평가 및 분석
Assessment and Analysis of Fidelity and Diversity for GAN-based Medical Image Generative Model
0
조회
0
다운로드
3
인용
0
참고문헌
키워드
초록
최근 의료영상의 발전에 따라 의료 영상 생성에 대한 다양한 연구가 제안되고 있는데, 이와 관련하여 생성된의료 영상의 품질과 다양성을 정확하게 평가하는 것이 중요해지고 있다. 생성된 의료 영상을 평가하는 방법으로는 전문가의 시각적 튜링 테스트(visual turing test), 특징 분포 시각화, IS, FID를 통한 정량적 평가를 통해평가하고 있으나 의료 영상을 품질(fidelity)과 다양성(diversity) 측면에서 정량적으로 평가 하는 방법은 거의 이루어지고 있지 않다. 본 논문에서는 DCGAN과 PGGAN 생성 모델을 통해 비소세포폐암 환자의 흉부 CT 데이터셋을 학습하여 영상을 생성하고, 이를 품질(fidelity)과 다양성(diversity) 측면에서 두 생성 모델의 성능을 평가한다. 1차원 점수 기반 평가방법인 IS, FID와 2차원 점수 기반 평가방법인 Precision 및 Recall, 개선된 Precision 및 Recall을 통해 성능을 정량적으로 평가하고, 의료영상에서의 각 평가방법들의 특징과 한계점에 대해서도 분석한다.
Abstract
Recently, various researches on medical image generation have been suggested, and it becomes crucial to accurately evaluate the quality and diversity of the generated medical images. For this purpose, the expert’s visual turing test, feature distribution visualization, and quantitative evaluation through IS and FID are evaluated. However, there are few methods for quantitatively evaluating medical images in terms of fidelity and diversity. In this paper, images are generated by learning a chest CT dataset of non-small cell lung cancer patients through DCGAN and PGGAN generative models, and the performance of the two generative models are evaluated in terms of fidelity and diversity. The performance is quantitatively evaluated through IS and FID, which are one-dimensional score-based evaluation methods, and Precision and Recall, Improved Precision and Recall, which are two-dimensional score-based evaluation methods, and the characteristics and limitations of each evaluation method are also analyzed in medical imaging.
참고문헌 (30)
- D. P. Kingma, "Auto-encoding variational bayes", arXiv preprint arXiv : 1312. 6114, ~ (2013)
- I. J. Goodfellow, "Generative adversarial networks", ~ (2014)
- 김민규, "딥러닝 기반 의료영상 분석을 위한 데이터 증강 기법", 대한영상의학회지, 81(6), pp.1290~1304 (2020), 대한영상의학회[DOI]
- V. Sandfort, "Data augmentation using generative adversarial networks(cyclegan)to improve generalizability in ct segmentation tasks", Scientific reports, 9(1), pp.19~ (2019)
- G. -P. Diller, "Utility of deep learning networks for the generation of artificial cardiac magnetic resonance images in congenital heart disease", BMC Medical Imaging, 20(1), pp.1~8 (2020)
- H. Y. Park, "Realistic high-resolution body computed tomography image synthesis by using progressive growing generative adversarial network : Visual turing test", JMIR Medical Informatics, 9(3), pp.e23328~ (2021)
- C. Han, "Gan-based synthetic brain mr image generation", 2018IEEE 15th International Symposium on Biomedical Imaging (ISBI2018). IEEE, pp.734~738 (2018)
- M. J. Chuquicusma, "How to fool radiologists with generative adversarial networks? a visual turing test for lung cancer diagnosis", 2018 IEEE 15th international symposium on biomedical imaging (ISBI2018). IEEE, pp.240~244 (2018)
- C. Zheng, "Assessment of generative adversarial networks model for synthetic optical coherence tomography images of retinal disorders", Translational Vision Science & Technology, 9(2), pp.29~29 (2020)
- H. Lee, "Classification of focal liver lesions in ct images using convolutional neural networks with lesion information augmented patches and synthetic data augmentation", Medical physics, 48(9), pp.5029~5046 (2021)
- M. Frid-Adar, "Gan-based synthetic medical image augmentation for increased cnn performance in liver lesion classification", Neurocomputing, 321, pp.321~331 (2018)
- C. Han, "Synthesizing diverse lung nodules wherever massively: 3d multi-conditional gan-based ct image augmentation for object detection", 2019 International Conference on 3D Vision(3DV). IEEE, pp.729~737 (2019)
- A. Borji, "Pros and cons of gan evaluation measures", ~ (2018)
- M. S. Sajjadi, "Assessing generative models via precision and recall", Advances in Neural Information Processing Systems, 31 (2018)
- T. Koga, "General-to-detailed gan for infrequent class medical images", arXiv preprint arXiv : 1812. 01690, ~ (2018)
- Skandarani, "Gans for medical image synthesis: An empirical study", arXiv preprint arXiv:2105.05318, ~ (2021)
- A. Radford, "Unsupervised representation learning with deep convolutional generative adversarial networks", ~ (2016)
- T. Karras, "Progressive growing of gans for improved quality, stability, and variation", International Conference on Learning Representations (2018)
- I. Gulrajani, "Improved training of wasserstein gans", Advances in neural information processing systems, 30 (2017)
- T. Salimans, "Improved techniques for training gans", ~ (2016)
- M. Heusel, "Gans trained by a two time-scale update rule converge to a local nash equilibrium", Advances in neural information processing systems, 30 (2017)
- A. Borji, "Pros and cons of gan evaluation measures: New developments", ~ (2021)
- T. Kynkäänniemi, "Improved precision and recall metric for assessing generative models", ~
- "The Cancer Imaging Archive", 2019, ~, https:
- C. Szegedy, "Rethinking the inception architecture for computer vision", Proceedings of the IEEE conference on computer vision and pattern recognition, pp.2818~2826 (2016)
- M. F. Naeem, "Reliable fidelity and diversity metrics for generative models", International Conference on Machine Learning. PMLR, pp.7176~7185 (2020)
- R. Geirhos, "Imagenet-trained cnns are biased to-wards texture; increasing shape bias improves accuracy and robustness", International Conference on Learning Representations (2018)
- T. Karras, "A style-based generator architecture for generative adversarial networks", Proceedings of the IEEE, 2019, 4401~4410
- T. Karras, "Analyzing and improving the image quality of stylegan", Proceedings of the IEEE, 2020, 8110~8119
- A. Brock, "Large scale gan training for high fidelity natural image synthesis", International Conference on Learning Representations (2018)