Neural Architecture Search based Global-local Vision Mamba for Palm-Vein Recognition

TL;DR

This paper introduces a novel hybrid neural network architecture called Global-local Vision Mamba (GLVM) for palm-vein recognition, combining local and global feature learning with neural architecture search to improve accuracy.

Contribution

The paper proposes a new hybrid network structure with Multi-head Mamba and ConvMamba blocks, and introduces a neural architecture search method to optimize the model for vein recognition.

Findings

Outperforms existing methods on public palm-vein datasets

Achieves state-of-the-art recognition accuracy

Demonstrates robustness and effectiveness of the proposed architecture

Abstract

Due to the advantages such as high security, high privacy, and liveness recognition, vein recognition has been received more and more attention in past years. Recently, deep learning models, e.g., Mamba has shown robust feature representation with linear computational complexity and successfully applied for visual tasks. However, vision Manba can capture long-distance feature dependencies but unfortunately deteriorate local feature details. Besides, manually designing a Mamba architecture based on human priori knowledge is very time-consuming and error-prone. In this paper, first, we propose a hybrid network structure named Global-local Vision Mamba (GLVM), to learn the local correlations in images explicitly and global dependencies among tokens for vein feature representation. Secondly, we design a Multi-head Mamba to learn the dependencies along different directions, so as to improve the feature representation ability of vision Mamba. Thirdly, to learn the complementary features, we propose a ConvMamba block consisting of three branches, named Multi-head Mamba branch (MHMamba), Feature Iteration Unit branch (FIU), and Convolutional Neural Network (CNN) branch, where the Feature Iteration Unit branch aims to fuse convolutional local features with Mamba-based global representations. Finally, a Globallocal Alternate Neural Architecture Search (GLNAS) method is proposed to search the optimal architecture of GLVM alternately with the evolutionary algorithm, thereby improving the recognition performance for vein recognition tasks. We conduct rigorous experiments on three public palm-vein databases to estimate the performance. The experimental results demonstrate that the proposed method outperforms the representative approaches and achieves state-of-the-art recognition accuracy.