Self-distillation Augmented Masked Autoencoders for Histopathological Image Classification

التفاصيل البيبلوغرافية
العنوان: Self-distillation Augmented Masked Autoencoders for Histopathological Image Classification
المؤلفون: Luo, Yang, Chen, Zhineng, Zhou, Shengtian, Gao, Xieping
سنة النشر: 2022
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computer Vision and Pattern Recognition
الوصف: Self-supervised learning (SSL) has drawn increasing attention in histopathological image analysis in recent years. Compared to contrastive learning which is troubled with the false negative problem, i.e., semantically similar images are selected as negative samples, masked autoencoders (MAE) building SSL from a generative paradigm is probably a more appropriate pre-training. In this paper, we introduce MAE and verify the effect of visible patches for histopathological image understanding. Moreover, a novel SD-MAE model is proposed to enable a self-distillation augmented MAE. Besides the reconstruction loss on masked image patches, SD-MAE further imposes the self-distillation loss on visible patches to enhance the representational capacity of the encoder located shallow layer. We apply SD-MAE to histopathological image classification, cell segmentation and object detection. Experiments demonstrate that SD-MAE shows highly competitive performance when compared with other SSL methods in these tasks.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2203.16983
رقم الأكسشن: edsarx.2203.16983
قاعدة البيانات: arXiv