计算机科学
修剪
初始化
人工智能
进化算法
上下文图像分类
人口
进化计算
领域(数学)
模式识别(心理学)
机器学习
图像(数学)
数学
社会学
人口学
生物
程序设计语言
纯数学
农学
作者
Jiaqi Zhao,Chengrun Yang,Yong Zhou,Yajie Zhou,Zhujun Jiang,Ying Chen
标识
DOI:10.1109/igarss47720.2021.9553847
摘要
Remote sensing image scene classification has achieved significant breakthroughs in recent years. However, due to the high complexity and expensive computation most of CNNs used in the field of remote sensing imagery scene classification, it has become a challenging task for extracting effective features at restricted hardware conditions. To solve this problem, we present a model compression method by means of evolutionary algorithms. Specifically, we compress the model by pruning filters and transform the compression of the CNN model into a multi-objective optimization problem based on classification accuracy and compression ratio by using the adaptive-BN-based evaluation method. Furthermore, the prior knowledge of ResNet-50 on ImageNet is introduced to reduce the instability of evolutionary algorithm as a result of random population initialization. Experiments are implemented on three datasets with two evolutionary algorithms, and results demonstrate that our method can achieve state-of-the-art performances.
科研通智能强力驱动
Strongly Powered by AbleSci AI