arXiv:2512.15931cs.LG2025-12中稿 · the 3rd Workshop o…被引 1

用状态空间模型提升真菌条形码分类,解决数据稀疏难题

BarcodeMamba+: Advancing State-Space Models for Fungal Biodiversity Research

  • 基于状态空间模型构建预训练-微调框架,适应标签稀疏场景
  • 在多个分类层级上超越现有方法,显著提升未见物种识别能力
  • 适合生物多样性研究、基因组学与少样本学习方向的研究者

从DNA条形码进行准确分类是全球生物多样性监测的核心,但真菌因标签稀疏和长尾类群分布而面临巨大挑战。传统监督学习方法在此领域常难以泛化至新物种,且无法有效捕捉数据的层级结构。为此,我们提出BarcodeMamba+,一种基于高效状态空间模型架构的真菌条形码分类基础模型。采用预训练-微调范式,利用部分标注数据,在此数据稀疏环境下表现远优于传统全监督方法。微调阶段系统集成并评估了多项改进——包括层级标签平滑、加权损失函数以及来自MycoAI的多头输出层,专门应对真菌分类难题。实验表明,各项组件均带来显著性能提升。在包含明显分类分布偏移的挑战性基准测试中,最终模型在所有分类层级上均优于多种现有方法。本工作为基于基因组的生物多样性研究提供了强大工具,并建立了一种适用于该复杂领域的有效可扩展训练范式。代码已公开于https://github.com/bioscan-ml/BarcodeMamba。

原文摘要 · Abstract (English)

Accurate taxonomic classification from DNA barcodes is a cornerstone of global biodiversity monitoring, yet fungi present extreme challenges due to sparse labelling and long-tailed taxa distributions. Conventional supervised learning methods often falter in this domain, struggling to generalize to unseen species and to capture the hierarchical nature of the data. To address these limitations, we introduce BarcodeMamba+, a foundation model for fungal barcode classification built on a powerful and efficient state-space model architecture. We employ a pretrain and fine-tune paradigm, which utilizes partially labelled data and we demonstrate this is substantially more effective than traditional fully-supervised methods in this data-sparse environment. During fine-tuning, we systematically integrate and evaluate a suite of enhancements--including hierarchical label smoothing, a weighted loss function, and a multi-head output layer from MycoAI--to specifically tackle the challenges of fungal taxonomy. Our experiments show that each of these components yields significant performance gains. On a challenging fungal classification benchmark with distinct taxonomic distribution shifts from the broad training set, our final model outperforms a range of existing methods across all taxonomic levels. Our work provides a powerful new tool for genomics-based biodiversity research and establishes an effective and scalable training paradigm for this challenging domain. Our code is publicly available at https://github.com/bioscan-ml/BarcodeMamba.

真菌分类状态空间模型少样本学习

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