[发明专利]一种基于自监督神经网络的指纹快速检索方法

说明书

一种基于自监督神经网络的指纹快速检索方法，构建指纹特征提取器模型并进行预训练；模型主体采用线性投射层加Transformer编码器的结构；将预训练过的指纹特征提取器模型迁移到指纹图片上，用自监督学习的方式再次训练；构建指纹图片重构器模型，然后与自监督学习得到的指纹特征提取器模型连接，利用指纹重构任务使指纹特征提取器模型对指纹特征的抽取能力进一步提升；用指纹图片重构任务训练得到的指纹特征提取器模型对指纹库中的指纹图片逐一进行特征提取，建立指纹索引库；用查询指纹在指纹索引库检索得到候选指纹序列。本发明可在大规模指纹检索中直接利用原始指纹图片进行自动特征提取，并可提高指纹检索的精度和效率。

技术领域

本发明属于生物特征识别技术领域，涉及图像处理和指纹检索，特别涉及一种基于自监督神经网络的指纹快速检索方法。

背景技术

指纹是目前生物特征识别技术领域中应用较为广泛的一种生物特征。由于指纹具有唯一性和不变性等优良特性，使得自动指纹识别系统在身份鉴别领域得到了广泛的应用。但随着自动指纹识别技术应用的领域越来越广，覆盖的人口越来越多，指纹数据库的规模也在不断扩大，有些指纹数据库的容量甚至达到亿级别，随之而来的挑战就是，从数亿个指纹中找出与查询指纹相似度较高的指纹，即大规模指纹图片的检索问题。以警用系统的指纹系统为例，如果每个公民都要在其数据库中登记存储指纹，则数据量动辄以亿计算，如此巨大的指纹数量必然导致指纹检索的速度降低。因此，针对大规模指纹数据库设计一款高效且精准的指纹检索算法对身份鉴别领域具有重要的意义。

传统的指纹图片检索方法都是基于人工设计的特征提取算法，对指纹图片进行特征提取，然后再利用该特征来进行指纹的检索和比对。比较常用的人工设计的指纹特征是“细节点”特征。细节点一般采用三元存储格式：(x,y,θ)，x为细节点的横坐标，y为细节点的纵坐标，θ为细节点处纹线的方向。这种人工设计的特征会因为设计者的先验知识等主观因素而有所局限性，从而忽视部分指纹特征，而这些被忽视掉的特征很可能是会对指纹检索起帮助作用的特征。

发明内容

为了克服上述现有技术的缺点，本发明的目的在于提供一种基于自监督神经网络的指纹快速检索方法，以期在大规模指纹检索中直接利用原始指纹图片进行自动特征提取，并提高指纹检索的精度和效率。

为了实现上述目的，本发明采用的技术方案是：

一种基于自监督神经网络的指纹快速检索方法，包括如下步骤：

步骤1，构建指纹特征提取器模型，然后对其在ImageNet数据集上进行预训练；所述指纹特征提取器模型的主体采用线性投射层加Transformer编码器的结构，将Transformer编码器第一个位置的输出作为全连接层分类器的输入；预训练时，将ImageNet数据集中的每一张图片切割成图像块，当作时间序列输入至线性投射层，然后根据全连接层分类器的输出类别与图片的真实类别计算损失，更新模型参数；

步骤2，将步骤1中预训练过的指纹特征提取器模型迁移到指纹图片上，用自监督学习的方式在指纹图片上进行再次训练，训练时同样将数据集中的每一张指纹图片切割成图像块，当作时间序列输入至线性投射层；

步骤3，构建指纹图片重构器模型，然后与步骤2中自监督学习得到的指纹特征提取器模型相连接，利用指纹重构任务使指纹特征提取器模型对指纹特征的抽取能力进一步提升；

步骤4，用步骤3中指纹图片重构任务所训练得到的指纹特征提取器模型，对指纹库中的指纹图片逐一进行特征提取，用提取的特征向量建立指纹索引库；

步骤5，用查询指纹在步骤4中建立的指纹索引库进行快速检索，得到候选指纹序列。