迁移学习与对抗生成网络结合的图像分类方法

扫码查看

原文链接

万方数据
维普

中文摘要：在深度图像分类任务中,传统方法依赖于预训练的卷积神经网络(CNN)进行特征提取并结合标准分类损失函数进行训练.然而,数据集样本不足和特征表示不足会影响模型的分类性能.为解决这一问题,一种结合迁移学习和对抗生成网络(GAN)的创新方法被提出.首先,利用在大规模数据集上预训练的ResNet模型提取图像的高层特征,通过微调使其适应新的分类任务;然后,训练GAN生成高质量的图像数据,增强训练数据集的多样性,从而提升模型的泛化能力.实验在CIFAR-10和ImageNet数据集上进行,结果表明,结合迁移学习和GAN的方法显著提高了图像分类的准确性和鲁棒性.所提出的方法为解决数据样本不足问题提供了有效的解决方案.

作者：

孙勇

展开 >

作者单位：

南京审计大学江苏南京 211815

关键词：

图像分类迁移学习对抗生成网络卷积神经网络特征提取

出版年：

2024

DOI：

10.3969/j.issn.1672-9528.2024.11.007

信息技术与信息化

山东电子学会

信息技术与信息化

影响因子：0.29

ISSN：1672-9528

年,卷(期)：2024.(11)