Resolution-based distillation for efficient histology image classification.
Resolution-based distillation for efficient histology image classification.
复制标题
DOI:
10.1016/j.artmed.2021.102136
复制
发表时间:
2021-09
影响因子:
7.5
通讯作者:
Hassanpour S
中科院分区:
文献类型:
--
作者:
DiPalma J;Suriawinata AA;Tafe LJ;Torresani L;Hassanpour S
Developing deep learning models to analyze histology images has been computationally challenging, as the massive size of the images causes excessive strain on all parts of the computing pipeline. This paper proposes a novel deep learning-based methodology for improving the computational efficiency of histology image classification. The proposed approach is robust when used with images that have reduced input resolution, and it can be trained effectively with limited labeled data. Moreover, our approach operates at either the tissue- or slide-level, removing the need for laborious patch-level labeling. Our method uses knowledge distillation to transfer knowledge from a teacher model pre-trained at high resolution to a student model trained on the same images at a considerably lower resolution. Also, to address the lack of large-scale labeled histology image datasets, we perform the knowledge distillation in a self-supervised fashion. We evaluate our approach on three distinct histology image datasets associated with celiac disease, lung adenocarcinoma, and renal cell carcinoma. Our results on these datasets demonstrate that a combination of knowledge distillation and self-supervision allows the student model to approach and, in some cases, surpass the teacher model’s classification accuracy while being much more computationally efficient. Additionally, we observe an increase in student classification performance as the size of the unlabeled dataset increases, indicating that there is potential for this method to scale further with additional unlabeled data. Our model outperforms the high-resolution teacher model for celiac disease in accuracy, F1-score, precision, and recall while requiring 4 times fewer computations. For lung adenocarcinoma, our results at 1.25x magnification are within 1.5% of the results for the teacher model at 10x magnification, with a reduction in computational cost by a factor of 64. Our model on renal cell carcinoma at 1.25x magnification performs within 1% of the teacher model at 5x magnification while requiring 16 times fewer computations. Furthermore, our celiac disease outcomes benefit from additional performance scaling with the use of more unlabeled data. In the case of 0.625x magnification, using unlabeled data improves accuracy by 4% over the tissue-level baseline. Therefore, our approach can improve the feasibility of deep learning solutions for digital pathology on standard computational hardware and infrastructures.
登录
查看更多内容
DOI:
10.1097/pas.0b013e3181b8cf03
发表时间:
2009-12
期刊:
The American journal of surgical pathology
影响因子:
--
作者:
Girard N;Deshpande C;Lau C;Finley D;Rusch V;Pao W;Travis WD
通讯作者:
Travis WD
影响因子:
3.5
作者:
Griebel L;Prokosch HU;Köpcke F;Toddenroth D;Christoph J;Leb I;Engel I;Sedlmayr M
通讯作者:
Sedlmayr M
DOI:
10.1016/j.patol.2019.02.004
发表时间:
2019-10-01
期刊:
Revista espanola de patologia : publicacion oficial de la Sociedad Espanola de Anatomia Patologica y de la Sociedad Espanola de Citologia
影响因子:
--
作者:
Gutierrez Olivares, Victor Manuel;Gonzalez Torres, Luz Mery;Niebles De la Hoz, Maria Camila
通讯作者:
Niebles De la Hoz, Maria Camila
影响因子:
--
作者:
Chen, Pingjun;Yang, Lin
通讯作者:
Yang, Lin
影响因子:
10.6
作者:
Ge, Shiming;Zhao, Shengwei;Li, Jia
通讯作者:
Li, Jia