DOI QR코드

DOI QR Code

Region and Global-Specific PatchCore based Anomaly Detection from Chest X-ray Images

  • Hyunbin Kim (ADDD) ;
  • Junchul Chun (Department of Computer Science, Kyoggi University)
  • 투고 : 2023.12.21
  • 심사 : 2024.08.01
  • 발행 : 2024.08.31

초록

This paper introduces a method aimed at diagnosing the presence or absence of lesions by detecting anomalies in Chest X-ray images. The proposed approach is based on the PatchCore anomaly detection method, which extracts a feature vector containing location information of an image patch from normal image data and calculates the anomaly distance from the normal vector. However, applying PatchCore directly to medical image processing presents challenges due to the possibility of diseases occurring only in specific organs and the presence of image noise unrelated to lesions. In this study, we present an image alignment method that utilizes affine transformation parameter prediction to standardize already captured X-ray images into a specific composition. Additionally, we introduce a region-specific abnormality detection method that requires affine-transformed chest X-ray images. Furthermore, we propose a method to enhance application efficiency and performance through feature map hard masking. The experimental results demonstrate that our proposed approach achieved a maximum AUROC (Area Under the Receiver Operating Characteristic) of 0.774. Compared to a previous study conducted on the same dataset, our method shows a 6.9% higher performance and improved accuracy.

키워드

과제정보

This research was supported by Basic Science Research Program through the National Research Foundation of Korea(NRF) funded by the Ministry of Education (No. 2020R1A6A1A03040583).

참고문헌

  1. Tang, Y.-X. et al., "Automated abnormality classification of chest radiographs using deep convolutional neural networks," npj Digital Medicine, vol.3, no.1, pp.1-8, 2020.
  2. Visuna, L. et al., "Computer-aided diagnostic for classifying chest X-ray images using deep ensemble learning," BMC Medical Imaging, vol.22, no.1, pp.1-16, 2022.
  3. Ligueran, R. J. S.D. et al., "Applied Computer Vision on 2-Dimensional Lung X-Ray Images for Assisted Medical Diagnosis of Pneumonia," International Journal of Computing Sciences Research (IJCSR), vol.7, pp.1239-1254, 2022.
  4. Liu, J. et al., "Align, Attend and Locate: Chest X-ray Diagnosis via Contrast Induced Attention Network With Limited Supervision," in Proc. of the IEEE/CVF International Conference on Computer Vision (ICCV), pp.10631-10640, 2019.
  5. Park, S. et al., "Self-evolving vision transformer for chest X-ray diagnosis through knowledge distillation," Nature Communications, vol.13, no.1, pp.1-11, 2022.
  6. Hyun-bin Kim and Jun-Chul Chun, "Leision Detection in Chest X-ray Images based on Coreset of Patch Feature," Journal of Internet Computing and Services, vol.23, no.3, pp.35-45, 2022.
  7. Bergmann, P. et al., "MVTec AD - A Comprehensive Real-world Dataset for Unsupervised Anomaly Detection," in Proc. of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp.9584-9592, 2019.
  8. Salah, R., Vincent, P., Muller, X., Glorot, X., Bengio, Y., "Contractive Auto-Encoders: Explicit Invariance During Feature Extraction," in Proc. of ICML'11: Proceedings of the 28th International Conference on International Conference on Machine Learning, pp.833-840, 2011.
  9. Schlegl T. et al., "Unsupervised Anomaly Detection with Generative Adversarial Networks to Guide Marker Discovery," in Proc. of International Conference on Information Processing in Medical Imaging (IPMI), Springer, Cham, pp.146-157, 2017.
  10. Ruff, L. et al., "Deep One-class Classification," in Proc. of 35th International Conference on Machine Learning, PMLR, vol.80, pp.4393-4402, 2018.
  11. Bergman, L., Niv, C., and Yedid, H., "Deep Nearest Neighbor Anomaly Detection," arXiv:2002.10445, 2020.
  12. Defard, T. et al., "PaDiM: A Patch Distribution Modeling Framework for Anomaly Detection and Localization," in Proc. of Pattern Recognition. ICPR International Workshops and Challenges. ICPR 2021, Springer, Cham, pp.475-489, 2021.
  13. Roth, K. et al., "Towards Total Recall in Industrial Anomaly Detection," in Proc. of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp.14298-14308, 2022.
  14. Cohen, N., Yedid H., "Sub-Image Anomaly Detection with Deep Pyramid Correspondences," arXiv:2005.02357, 2020.
  15. Deng, J., Dong, W., Socher, R., Li, L.J., Li, K. and Fei-Fei, L., "ImageNet: A large-scale hierarchical image database," in Proc. of IEEE Conference on Computer Vision and Pattern Recognition, pp.248-255, 2009.
  16. Zagoruyko, S., Komodakis, N., "Wide Residual Networks," arXiv:1605.07146, 2016.
  17. Sener, O., Savarese, S., "Active Learning for Convolutional Neural Networks: A Core-Set Approach," in Proc. of 6th International Conference on Learning Representations (ICLR 2018), 2018.
  18. He, K. et al., "Deep Residual Learning for Image Recognition," in Proc. of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp.770-778, 2016.
  19. Zhu, J.-Y. et al., "Unpaired Image-to-Image Translation Using Cycle-Consistent Adversarial Networks," in Proc. of the 2017 IEEE International Conference on Computer Vision (ICCV), pp.2242-2251, 2017.
  20. Ledig, C. et al., "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network," in Proc. of 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp.105-114, 2017.
  21. Yi, J., Yoon, S., "Patch SVDD: Patch-Level SVDD for Anomaly Detection and Segmentation," in Proc. of 15th Asian Conference on Computer Vision - ACCV 2020, 2021.
  22. Bustos, A. et al., "PadChest: A large chest x-ray image dataset with multi-label annotated reports," Medical Image Analysis, vol.66, 2020.
  23. Schlegl, T. et al., "f-AnoGAN: Fast unsupervised anomaly detection with generative adversarial networks," Medical Image Analysis, vol.54, pp.30-44, 2019.