计算机工程与应用 ›› 2023, Vol. 59 ›› Issue (1): 187-198.DOI: 10.3778/j.issn.1002-8331.2106-0116

• 模式识别与人工智能 • 上一篇    下一篇

双重注意力机制下的跨光谱虹膜识别优化算法

任家润,沈文忠   

  1. 上海电力大学 电子与信息工程学院,上海 201200
  • 出版日期:2023-01-01 发布日期:2023-01-01

Optimization Algorithm of Cross Spectral Iris Recognition Based on Dual Attention Mechanism

REN Jiarun, SHEN Wenzhong   

  1. School of Electronic and Information Engineering, Shanghai University of Electric Power, Shanghai 201200, China
  • Online:2023-01-01 Published:2023-01-01

摘要: 跨光谱虹膜识别任务,通常是指匹配不同光谱下采集的虹膜图像,针对光谱域变化对虹膜识别率影响的问题,提出一种基于双重注意力机制下的跨光谱虹膜识别优化算法。该算法首先通过浅层网络的特征提取,将可见光与近红外光谱域中的浅层特征送入各自的专有外部注意力模块进行优化,以分别存储各自光谱域中虹膜样本数据集的特有信息,随后将优化后的浅层特征送入共享的深度特征提取网络,再将深度特征先后送入改进的空间注意力模块和外部注意力模块,强化虹膜关键特征信息的表达,同时以改进的加性角间距损失函数ArcFace Loss(additive angular margin loss)动态调整对于困难样本的优化力度。该算法在PolyU数据集上进行了实验验证,等错误率EER(equal error rate)达到了0.23%,分离度达到了8.56,该实验结果与跨光谱虹膜识别SOTA(state-of-the-art)算法进行了比较,远高于其他主流算法。

关键词: 跨光谱, 虹膜识别, 注意力机制, ArcFace Loss, 困难样本

Abstract: The task of cross spectral iris recognition usually refers to matching iris images collected under different spectra. Aiming at the problem of the influence of spectral domain change on iris recognition rate, an optimization algorithm of cross spectral iris recognition based on dual attention mechanism is proposed. Firstly, through the feature extraction of shallow network, the shallow features in the visible and near-infrared spectral domain are sent to their own dedicated external attention module for optimization, so as to store the unique information of iris sample datasets in their respective spectral domain, and then the optimized shallow features are sent to the shared deep feature extraction network, Then, the deep features are sent to the improved spatial attention module and external attention module to enhance the expression of iris key feature information. At the same time, the improved additive angular margin loss function(ArcFace Loss) is used to dynamically adjust the optimization strength for difficult samples. Experimental results on PolyU dataset show that EER(equal error rate) is 0.23% and resolution is 8.56. The experimental results are compared with SOTA(state-of-the-art) algorithm, which is much higher than other mainstream algorithms.

Key words: cross spectrum, iris recognition, attention mechanism, ArcFace loss, difficult sample