迁移学习分类器

一种面向不平衡数据的分类模型,通过结合主动采样、实时数据增强和预训练DenseNet,提升对静态和实时数据的分类效果。

Liu, Yang · Yang, Guoping · 乔少杰 · Meiqi, Liu · Qu, Lulu · 韩楠 · 吴涛 · Yuan, Guan · Yuzhong, Peng

Engineering Applications of Artificial Intelligence 2023

技术优势

主动采样调整样本分布

新提出的主动采样模块能够动态调整倾斜分布的样本数量,从而缓解类别不平衡对模型的影响。

实时数据增强避免过拟合

数据增强模块扩展实时数据,避免因数据不足导致的过拟合,提升模型泛化能力。

迁移预训练DenseNet降低内存占用

使用在ImageNet上预训练的DenseNet并调整内存使用,使模型更高效,同时继承强大的特征提取能力。

应用场景