| 注册
首页|期刊导航|电子学报|基于坐标重要性池化和解耦类别对齐蒸馏的图像分类算法

基于坐标重要性池化和解耦类别对齐蒸馏的图像分类算法

刘颖 薛家昊 张伟东 许志杰

电子学报2025,Vol.53Issue(3):962-973,12.
电子学报2025,Vol.53Issue(3):962-973,12.DOI:10.12263/DZXB.20240754

基于坐标重要性池化和解耦类别对齐蒸馏的图像分类算法

Image Classification Algorithm Based on Coordinate Importance Pooling and Decoupled Class Alignment Distillation

刘颖 1薛家昊 2张伟东 1许志杰3

作者信息

  • 1. 西安邮电大学图像与信息处理研究所,陕西 西安 710121||无线通信与信息处理技术国际联合研究中心,陕西 西安 710121
  • 2. 西安邮电大学图像与信息处理研究所,陕西 西安 710121
  • 3. 无线通信与信息处理技术国际联合研究中心,陕西 西安 710121||英国哈德斯菲尔德大学,西约克郡 HD13DH
  • 折叠

摘要

Abstract

An image classification algorithm based on coordinate importance pooling and decoupled class alignment distillation is proposed to improve the image classification accuracy of convolutional neural networks while achieving net-work lightweighting.Firstly,a coordinate importance pooling module is designed and embedded it into ResNet34,in order to fully utilize the positional information of image pixels to enhance the ability to discriminate important features.Secondly,BlurPool is used to mitigate the impact on network performance due to shift equivariance during down-sampling,and to construct the teacher network.Finally,the decoupled class alignment distillation algorithm was constructed to efficiently mi-grate image classification knowledge from the teacher network to the lightweight MobileNetV3 network,which considers the knowledge of target and non-target class separately and introduces correlation information between the class.The experi-mental results on different datasets showed that the proposed teacher network effectively improves the classification perfor-mance,and the distillation-trained student network achieves superior overall performance than other networks of the same magnitude,making it better applicable to practical scenarios with limited computational and storage power.

关键词

图像分类/轻量化/知识蒸馏/ResNet34/坐标重要性池化

Key words

image classification/lightweight/knowledge distillation/ResNet34/coordinate importance pooling

分类

信息技术与安全科学

引用本文复制引用

刘颖,薛家昊,张伟东,许志杰..基于坐标重要性池化和解耦类别对齐蒸馏的图像分类算法[J].电子学报,2025,53(3):962-973,12.

基金项目

国家自然科学基金(No.62106195) National Natural Science Foundation of China(No.62106195) (No.62106195)

电子学报

OA北大核心

0372-2112

访问量7
|
下载量0
段落导航相关论文