Nowadays, the iris has become a major instrument, promising to replace fingerprints in identification. In addition, the iris is also a promising biomarker for non-invasive pre-diagnostic problems. To improve the iris recognition ability, in this study, we propose to use Vision Transformer (ViT) combined with Cross-Entropy Loss and Triplet Loss to support the classification. Different from traditional training methods, we conduct a comparison between the ViT model trained from scratch and the ViT-B/16 model fine-tuned from pre-trained weights. The results show that, in the context of limited training data, the fine-tuned model from ViT-B/16 gives significantly higher performance and converges faster than the model trained from scratch. The system is evaluated on various iris databases, including CASIA-Iris-Thousand, CASIA-Iris-Lamp, and CASIA-Interval, demonstrating good stability and generalization ability. The proposed method achieves classification accuracy ranging over 99%, confirming the effectiveness of combining two loss functions and leveraging features from the pretrained ViT model.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Dual-Loss Vision Transformer Framework for Robust Iris Recognition

  • Van-Huu Tran,
  • Van-Len Vo,
  • The-Bao Nguyen,
  • Quoc H. Nguyen,
  • Hung Ho-Dac,
  • Trong-Thua Huynh

摘要

Nowadays, the iris has become a major instrument, promising to replace fingerprints in identification. In addition, the iris is also a promising biomarker for non-invasive pre-diagnostic problems. To improve the iris recognition ability, in this study, we propose to use Vision Transformer (ViT) combined with Cross-Entropy Loss and Triplet Loss to support the classification. Different from traditional training methods, we conduct a comparison between the ViT model trained from scratch and the ViT-B/16 model fine-tuned from pre-trained weights. The results show that, in the context of limited training data, the fine-tuned model from ViT-B/16 gives significantly higher performance and converges faster than the model trained from scratch. The system is evaluated on various iris databases, including CASIA-Iris-Thousand, CASIA-Iris-Lamp, and CASIA-Interval, demonstrating good stability and generalization ability. The proposed method achieves classification accuracy ranging over 99%, confirming the effectiveness of combining two loss functions and leveraging features from the pretrained ViT model.