arXiv:2409.04218cs.CV2024-09

轻量级模型MpoxMamba高效识别猴痘皮损,适合资源有限地区部署。

MpoxMamba: A Grouped Mamba-based Lightweight Hybrid Network for Mpox Detection

  • 融合深度可分离卷积与分组Mamba模块,兼顾局部特征与长程依赖建模。
  • 参数量仅0.77M,计算量0.53G,远低于同类模型。
  • 在两个公开数据集上超越现有方法,支持在线免费检测服务。

由于缺乏有效的猴痘检测工具,猴痘病毒持续全球传播,并再度被世界卫生组织宣布为国际公共卫生紧急事件。基于轻量级深度学习的检测系统对缓解疫情至关重要,尤其适用于资源有限的场景。然而,其成功应用的关键在于模型能否在保持轻量化的同时,有效捕捉猴痘皮损的局部特征与长程依赖关系。受Mamba在建模长程依赖方面的成功及其线性复杂度启发,本文提出一种名为MpoxMamba的轻量级混合架构,用于高效猴痘检测。该模型采用深度可分离卷积提取皮损局部特征,通过分组Mamba模块显著增强全局上下文信息建模能力。实验结果表明,MpoxMamba参数量为0.77M,FLOPs为0.53G,优于当前最先进的轻量级模型及现有猴痘检测方法。此外,研究团队开发了基于Web的在线检测应用(http://5227i971s5.goho.co:30290),并开源代码(https://github.com/YubiaoYue/MpoxMamba)。

原文摘要 · Abstract (English)

Due to the lack of effective mpox detection tools, the mpox virus continues to spread worldwide and has once again been declared a public health emergency of international concern by the World Health Organization. Lightweight deep learning model-based detection systems are crucial to alleviate mpox outbreaks since they are suitable for widespread deployment, especially in resource-limited scenarios. However, the key to its successful application depends on ensuring that the model can effectively model local features and long-range dependencies in mpox lesions while maintaining lightweight. Inspired by the success of Mamba in modeling long-range dependencies and its linear complexity, we proposed a lightweight hybrid architecture called MpoxMamba for efficient mpox detection. MpoxMamba utilizes depth-wise separable convolutions to extract local feature representations in mpox skin lesions and greatly enhances the model's ability to model the global contextual information by grouped Mamba modules. Notably, MpoxMamba's parameter size and FLOPs are 0.77M and 0.53G, respectively. Experimental results on two widely recognized benchmark datasets demonstrate that MpoxMamba outperforms state-of-the-art lightweight models and existing mpox detection methods. Importantly, we developed a web-based online application to provide free mpox detection (http://5227i971s5.goho.co:30290). The source codes of MpoxMamba are available at https://github.com/YubiaoYue/MpoxMamba.

猴痘检测轻量模型Mamba医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。