面向中小企业的基于张量注意力的小样本分类方法和系统

摘要

本发明公开面向中小企业的基于张量注意力的小样本分类方法和系统，属于人工智能领域。包括：划分为支持集和查询集，图像用张量表征，将支持集和查询集输入至分类模型，其包括串联的神经网络骨架和张量注意力模块，张量注意力模块用于对数据特征进行位置和空间上的融合，得到张量自注意力特征，对支持集原型特征张量与查询集自注意力特征张量进行相似度计算，计算损失并更新，将类别未知中小企业监控图像转化为张量数据，输入至微调后模型进行预测。本发明引入张量注意力机制，将传统注意力模块扩展到张量空间，用于维持数据间潜在关联关系；将提取到浅层特征输入到张量注意力模块中，实现数据位置和空间上的融合，提高小样本图像分类的准确率。