基于通道注意力损失的深度图像分类方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
基于通道注意力损失的深度图像分类方法
申请号:CN202410729684
申请日期:2024-06-06
公开号:CN118674981A
公开日期:2024-09-20
类型:发明专利
摘要
本发明公开了一种基于通道注意力损失的深度图像分类方法,旨在加强网络对图像中突出区域的关注度,并加强其学习判别性特征的能力。首先构建图像的深度特征分布图,通过卷积网络与特征提取层,得到图像所有深度特征通道信息;然后在主分支对深度特征分布图进行全局池化处理,再通过一个完全连接的层,计算SoftMax损失;接着,在额外分支将通道信息公平分配给分类任务的每个类别,提取更具判别性的特征信息,增强网络对不同类别细节的关注度,计算通道注意力损失,推导特定类别的概率;最后将SoftMax损失和通道注意力损失相结合,得到整个网络的最终损失函数,进行联合监督训练与参数优化,提升深度图像分类任务性能。
技术关键词
图像分类方法 深度神经网络 通道注意力机制 分支 参数 卷积网络模型 失活方法 样本通道 处理器 标签 存储装置 可读存储介质 符号 程序 矩阵