A regularization term for slide correlation reduction in whole slide image analysis with deep learningDownload PDF

Jan 21, 2021 (edited Feb 22, 2021)MIDL 2021 Conference SubmissionReaders: Everyone
  • Keywords: deep learning, histopathology, whole slide image, over-fitting, uveal melanoma
  • Abstract: To develop deep learning-based models for automatic analysis of histopathology whole slide images (WSIs), the atomic entities to be directly processed are often the smaller patches cropped from WSIs as it is not always possible to feed a whole WSI to a model given its enormous size. However, a trained model tends to relate the slide-specific characteristics to diagnosis results because a large number of patches cropped from the same WSI will share common slide features and thus have strong correlations between them, resulting in deteriorated generalization capability of the trained model. Current approaches to alleviate this issue include data pre-processing (stain normalization or color augmentation) and adversarial learning, both of which introduce extra complications in computations. Alternatively, we propose to reduce the impact of this issue by introducing a new regularization term to the standard loss function to reduce the correlation of the patches from the same WSI. It is intuitive and easy-to-implement and introduces comparably smaller computation overhead compared to existing approaches. Experimental results prove that the proposed regularization term is able to enhance the generalization capability of learning models and consequently to achieve better performance. The code is available in: \url{https://github.com/hrzhang1123/SlideCorrelationReduction}.
  • Registration: I acknowledge that publication of this at MIDL and in the proceedings requires at least one of the authors to register and present the work during the conference.
  • Source Code Url: https://github.com/hrzhang1123/SlideCorrelationReduction
  • Authorship: I confirm that I am the author of this work and that it has not been submitted to another publication before.
  • Data Set Url: https://camelyon16.grand-challenge.org/
  • Paper Type: methodological development
  • Source Latex: zip
  • Primary Subject Area: Application: Histopathology
  • Secondary Subject Area: Interpretability and Explainable AI
14 Replies