现代电子技术2024，Vol.47Issue(7)：54-60,7.DOI:10.16652/j.issn.1004-373x.2024.07.009
轻量级空间移位MLP用于指静脉分割
Lightweight spatial-shift MLP for finger vein segmentation
摘要
Abstract
The CNN and Transformer based architectures' image segmentation network models are characterized by excessive parameters,intricate computational demands,and a substantial requirement for memory resources,which makes them unable to meet the demand for fast and effective finger vein image segmentation,and difficult to be deployed in embedded platforms with limited arithmetic power.Therefore,a lightweight finger vein segmentation algorithm based on MLP is proposed.The information flow is acquired by moving the feature maps in different axes to capture local dependencies and improve local information extraction.The feature maps are labeled and the features are projectively convolved by tokenized MLP blocks.A lightweight attention module is added to improve the segmentation performance before both down-sampling and up-sampling,shifting the inputs while inputting them to the MLP,so that the network model focuses more on learning the local dependencies.The algorithm was experimented on three publicly available finger vein datasets of SDU-FV,HKPU and UTFVP.The results show that the method uses only 346.949K Params,1.835G Flops,and 11.023M computational complexity,and the segmentation performance metrics Dice,AUC,and Acc reach 0.5156,0.8959,and 91.68%,respectively.The algorithmic method achieves optimal performance for both Dice and AUC metrics on three NVIDIA embedded platforms.关键词
手指静脉分割/CNN/Transformer/轻量级/嵌入式平台/标记MLPKey words
finger vein segmentation/CNN/Transformer/lightweight/embedded platform/tokenized MLP分类
电子信息工程引用本文复制引用
曾军英,秦传波,田慧明,陈宇聪,顾亚谨,邓森耀,尹永宏,尤吴杭,黄国林,甘俊英..轻量级空间移位MLP用于指静脉分割[J].现代电子技术,2024,47(7):54-60,7.基金项目
国家自然科学基金项目(61771347) （61771347）
广东普通高校人工智能重点领域专项(2019KZDZX1017) （2019KZDZX1017）
广东普通高校重点领域专项(2020ZDZX3031) （2020ZDZX3031）
广东省基础与应用基础研究基金(2021A1515011576) （2021A1515011576）
2022年广东省教育厅研究生教育创新计划项目(粤教研函[2022]1号,五邑大学-江门妇幼保健院联合培养研究生示范基地) （粤教研函[2022]1号,五邑大学-江门妇幼保健院联合培养研究生示范基地）
