Cargando…
Cervical Cell Image Classification-Based Knowledge Distillation
Current deep-learning-based cervical cell classification methods suffer from parameter redundancy and poor model generalization performance, which creates challenges for the intelligent classification of cervical cytology smear images. In this paper, we establish a method for such classification tha...
Autores principales: | , , , , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
MDPI
2022
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9680356/ https://www.ncbi.nlm.nih.gov/pubmed/36412723 http://dx.doi.org/10.3390/biomimetics7040195 |
_version_ | 1784834397746233344 |
---|---|
author | Gao, Wenjian Xu, Chuanyun Li, Gang Zhang, Yang Bai, Nanlan Li, Mengwei |
author_facet | Gao, Wenjian Xu, Chuanyun Li, Gang Zhang, Yang Bai, Nanlan Li, Mengwei |
author_sort | Gao, Wenjian |
collection | PubMed |
description | Current deep-learning-based cervical cell classification methods suffer from parameter redundancy and poor model generalization performance, which creates challenges for the intelligent classification of cervical cytology smear images. In this paper, we establish a method for such classification that combines transfer learning and knowledge distillation. This new method not only transfers common features between different source domain data, but also realizes model-to-model knowledge transfer using the unnormalized probability output between models as knowledge. A multi-exit classification network is then introduced as the student network, where a global context module is embedded in each exit branch. A self-distillation method is then proposed to fuse contextual information; deep classifiers in the student network guide shallow classifiers to learn, and multiple classifier outputs are fused using an average integration strategy to form a classifier with strong generalization performance. The experimental results show that the developed method achieves good results using the SIPaKMeD dataset. The accuracy, sensitivity, specificity, and F-measure of the five classifications are 98.52%, 98.53%, 98.68%, 98.59%, respectively. The effectiveness of the method is further verified on a natural image dataset. |
format | Online Article Text |
id | pubmed-9680356 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2022 |
publisher | MDPI |
record_format | MEDLINE/PubMed |
spelling | pubmed-96803562022-11-23 Cervical Cell Image Classification-Based Knowledge Distillation Gao, Wenjian Xu, Chuanyun Li, Gang Zhang, Yang Bai, Nanlan Li, Mengwei Biomimetics (Basel) Article Current deep-learning-based cervical cell classification methods suffer from parameter redundancy and poor model generalization performance, which creates challenges for the intelligent classification of cervical cytology smear images. In this paper, we establish a method for such classification that combines transfer learning and knowledge distillation. This new method not only transfers common features between different source domain data, but also realizes model-to-model knowledge transfer using the unnormalized probability output between models as knowledge. A multi-exit classification network is then introduced as the student network, where a global context module is embedded in each exit branch. A self-distillation method is then proposed to fuse contextual information; deep classifiers in the student network guide shallow classifiers to learn, and multiple classifier outputs are fused using an average integration strategy to form a classifier with strong generalization performance. The experimental results show that the developed method achieves good results using the SIPaKMeD dataset. The accuracy, sensitivity, specificity, and F-measure of the five classifications are 98.52%, 98.53%, 98.68%, 98.59%, respectively. The effectiveness of the method is further verified on a natural image dataset. MDPI 2022-11-10 /pmc/articles/PMC9680356/ /pubmed/36412723 http://dx.doi.org/10.3390/biomimetics7040195 Text en © 2022 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/). |
spellingShingle | Article Gao, Wenjian Xu, Chuanyun Li, Gang Zhang, Yang Bai, Nanlan Li, Mengwei Cervical Cell Image Classification-Based Knowledge Distillation |
title | Cervical Cell Image Classification-Based Knowledge Distillation |
title_full | Cervical Cell Image Classification-Based Knowledge Distillation |
title_fullStr | Cervical Cell Image Classification-Based Knowledge Distillation |
title_full_unstemmed | Cervical Cell Image Classification-Based Knowledge Distillation |
title_short | Cervical Cell Image Classification-Based Knowledge Distillation |
title_sort | cervical cell image classification-based knowledge distillation |
topic | Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9680356/ https://www.ncbi.nlm.nih.gov/pubmed/36412723 http://dx.doi.org/10.3390/biomimetics7040195 |
work_keys_str_mv | AT gaowenjian cervicalcellimageclassificationbasedknowledgedistillation AT xuchuanyun cervicalcellimageclassificationbasedknowledgedistillation AT ligang cervicalcellimageclassificationbasedknowledgedistillation AT zhangyang cervicalcellimageclassificationbasedknowledgedistillation AT bainanlan cervicalcellimageclassificationbasedknowledgedistillation AT limengwei cervicalcellimageclassificationbasedknowledgedistillation |