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...

Descripción completa

Detalles Bibliográficos
Autores principales: Gao, Wenjian, Xu, Chuanyun, Li, Gang, Zhang, Yang, Bai, Nanlan, Li, Mengwei
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