摘要
本发明提供一种基于元注意力机制的火星表面岩石分类系统和方法,涉及图像分类识别技术领域。本发明提供的系统通过为神经网络模型配置元注意力机制,以为火星图像的每个图像块分配注意力掩码,以实现对神经网络模型的每个transformer单元的自注意力值进行调整。同时元注意力机制还能够确定每个神经元对应的权重值,并通过权重值调整神经网络模型的前馈神经网络的权重矩阵,这样一来,在面对不同的火星岩石分类任务时,能够有效提升分类精度,并且,由于部署在火星的计算设备的资源有限,本系统能够避免模型在面对不同的分类任务和不同种类的岩石区域时,进行大规模的参数调节和模型再训练,节约计算资源和时间。