2018年度  第7期


标题:基于标签相关性的卷积神经网络多标签分类算法
作者:蒋俊钊 程良伦 李全杰
作者单位:广东工业大学自动化学院,广东 广州 510006
关键字:标签相关性,卷积神经网络,后验概率,多标签分类
摘要:针对卷积神经网络对于多标签分类中未考虑标签相关性的问题,提出了一种考虑标签相关性的卷积神经网络分类方法。在AlexNet网络结构的基础上进行了两处改进:首先,通过共享卷积层的通用特征,降低神经网络的训练时间;其次,根据贝叶斯模型计算待预测标签的后验概率,通过在神经网络的目标函数中引入后验概率与预测概率的近似度作为损失项,提升多标签分类的准确率。实验结果表明,所提算法在Cars车辆数据集上取得了更好的分类效果。