-
公开(公告)号:CN114004333A
公开(公告)日:2022-02-01
申请号:CN202111251327.2
申请日:2021-10-26
Applicant: 桂林理工大学
IPC: G06N3/04 , G06N3/08 , G06K9/62 , G06V10/764 , G06V10/82
Abstract: 一种基于多假类生成对抗网络的过采样方法。首先通过一个条件变分自编码器学习不平衡图像数据的特征;然后使用变分自编码器中的解码器初始化GAN中的生成器,帮助分别器更好的确定类别分布;然后使用不平衡数据集来训练多假类生成对抗网络,同时为多假类生成对抗网络添加梯度惩罚项,以提升GAN训练的稳定性,并保证样本生成多样性;将多假类生成对抗网络中的分类损失替换为焦点损失,使GAN在训练时更加侧重那些难于分对的样本;最后为训练后的GAN模型输入少数类别标签及随机噪声,即可生成高质量的少数类样本。本发明能够有效地为不平衡图像数据集中的少数类生成高质量的样本,使数据成为平衡数据集,并帮助分类器提高不平衡场景下的分类性能。