Abstract
Histopathological analysis is the present gold standard for cancer diagnosis. Accurate classification of histopathology images has great clinical significance and application value for assisting pathologists in diagnosis. However, the performance of histopathology image classification is greatly affected by data imbalance. To address this problem, we propose a novel data augmentation framework based on the diffusion model, DiscrimDiff, which expands the dataset by synthesizing images of rare classes. To compensate for the lack of discrimination ability of the diffusion model for synthesized images, we design a post-discrimination mechanism to provide image quality assurance for data augmentation. Our method significantly improves classification performance on multiple datasets. Furthermore, histomorphological features of different classes concerned by the diffusion model may provide guiding significance for pathologists in clinical diagnosis. Therefore, we visualize histomorphological features related to classification, which can be used to assist pathologist-in-training education and improve the understanding of histomorphology.
X. Guan, Y. Wang and Y. Lin—Co-first authors.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Fuchs, T.J., Buhmann, J.M.: Computational pathology: challenges and promises for tissue analysis. Comput. Med. Imaging Graph. 35(7–8), 515–530 (2011)
Lu, M.Y., Williamson, D.F., Chen, T.Y., Chen, R.J., Barbieri, M., Mahmood, F.: Data-efficient and weakly supervised computational pathology on whole-slide images. Nat. Biomed. Eng. 5(6), 555–570 (2021)
Cui, M., Zhang, D.Y.: Artificial intelligence and computational pathology. Lab. Invest. 101(4), 412–422 (2021)
Abada, E., Anaya, I.C., Abada, O., Lebbos, A., Beydoun, R.: Colorectal adenocarcinoma with enteroblastic differentiation: diagnostic challenges of a rare case encountered in clinical practice. J. Pathol. Transl. Med. 56(2), 97–102 (2022)
Abbasniya, M.R., Sheikholeslamzadeh, S.A., Nasiri, H., Emami, S.: Classification of breast tumors based on histopathology images using deep features and ensemble of gradient boosting methods. Comput. Electr. Eng. 103, 108382 (2022)
Ho, J., Jain, A., Abbeel, P.: Denoising diffusion probabilistic models. In: Advances in Neural Information Processing Systems, vol. 33, pp. 6840–6851 (2020)
Dhariwal, P., Nichol, A.: Diffusion models beat GANs on image synthesis. In: Advances in Neural Information Processing Systems, vol. 34, pp. 8780–8794 (2021)
Moghadam, P.A., et al.: A morphology focused diffusion probabilistic model for synthesis of histopathology images. In: Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision, pp. 2000–2009 (2023)
Carrillo-Perez, F., Pizurica, M., Zheng, Y., Shen, J., Gevaert, O.: RNA-to-image multi-cancer synthesis using cascaded diffusion models. bioRxiv (2023)
Jeong, J., Kim, K.D., Nam, Y., Cho, C.E., Go, H., Kim, N.: Stain normalization using score-based diffusion model through stain separation and overlapped moving window patch strategies. Comput. Biol. Med. 152, 106335 (2023)
Xue, Y., et al.: Synthetic augmentation and feature-based filtering for improved cervical histopathology image classification. In: Shen, D., et al. (eds.) Medical Image Computing and Computer Assisted Intervention - MICCAI 2019. Lecture Notes in Computer Science(), vol. 11764, pp. 387–396. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32239-7_43
Xue, Y., et al.: Selective synthetic augmentation with HistoGAN for improved histopathology image classification. Med. Image Anal. 67, 101816 (2021)
Dravid, A., Schiffers, F., Gong, B., Katsaggelos, A.K.: medXGAN: visual explanations for medical classifiers through a generative latent space. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 2936–2945(2022)
Dolezal, J.M., et al.: Deep learning generates synthetic cancer histology for explainability and education. arXiv preprint: arXiv:2211.06522 (2022)
McInnes, L., Healy, J., Melville, J.: UMAP: uniform manifold approximation and projection for dimension reduction. arXiv preprint: arXiv:1802.03426 (2018)
Wei, J., et al.: A petri dish for histopathology image analysis. In: Tucker, A., Henriques Abreu, P., Cardoso, J., Pereira Rodrigues, P., Riano, D. (eds.) Artificial Intelligence in Medicine. Lecture Notes in Computer Science(), vol. 12721, pp. 11–24. Springer, Cham (2021). https://doi.org/10.1007/978-3-030-77211-6_2
Kather, J. N., Halama, N., Marx, A.: 100,000 histological images of human colorectal cancer and healthy tissue (v0.1) [Data set]. Zenodo (2018). https://doi.org/10.5281/zenodo.1214456
Spanhol, F.A., Oliveira, L.S., Petitjean, C., Heutte, L.: A dataset for breast cancer histopathological image classification. IEEE Trans. Biomed. Eng. 63(7), 1455–1462 (2015)
Leavey, P., Sengupta, A., Rakheja, D., Daescu, O., Arunachalam, H.B., Mishra, R.: Osteosarcoma data from UT Southwestern/UT Dallas for Viable and Necrotic Tumor Assessment [Data set]. The Cancer Imaging Archive (2019). https://doi.org/10.7937/tcia.2019.bvhjhdas
Han, C., et al.: WSSS4LUAD: grand challenge on weakly-supervised tissue semantic segmentation for lung adenocarcinoma. arXiv preprint: arXiv:2204.06455 (2022)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Zhang, H., Cisse, M., Dauphin, Y.N., Lopez-Paz, D.: mixup: beyond empirical risk minimization. arXiv preprint: arXiv:1710.09412 (2017)
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China (62031023), in part by the Shenzhen Science and Technology Project (JCYJ20200109142808034 &GXWD20220818170353009), and in part by Guangdong Special Support (2019TX05X187).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
1 Electronic supplementary material
Below is the link to the electronic supplementary material.
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Guan, X., Wang, Y., Lin, Y., Zhang, Y. (2024). Data Augmentation Based on DiscrimDiff for Histopathology Image Classification. In: Xue, Y., Chen, C., Chen, C., Zuo, L., Liu, Y. (eds) Data Augmentation, Labelling, and Imperfections. MICCAI 2023. Lecture Notes in Computer Science, vol 14379. Springer, Cham. https://doi.org/10.1007/978-3-031-58171-7_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-58171-7_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-58170-0
Online ISBN: 978-3-031-58171-7
eBook Packages: Computer ScienceComputer Science (R0)