arXiv:2512.06613cs.CV2025-12被引 1

用分层网络提升硅藻分类准确率,错误更局部,结果更可信。

Hierarchical Deep Learning for Diatom Image Classification: A Multi-Level Taxonomic Approach

  • 构建五级分层网络,逐级预测分类阶元,共享特征并限制合法后代。
  • 物种级准确率69.4%,上层分类准确率显著提升,错误定位更精准。
  • 适合生态监测中需要多层级分类的场景,结果更符合生物分类逻辑。

硅藻的精确分类对水生态系统监测至关重要,传统方法依赖专家。现有深度学习模型多为扁平分类,仅预测单一分类等级。本文探究将分类层级嵌入神经网络是否可提升精度与错误局部性。提出DiatomCascadeNet(H-COFGS),一个包含五个级联头的分层卷积网络,联合预测门、纲、目、科、属、种。各头接收共享骨干特征及高层概率分布,训练与推理时使用二值掩码限制预测范围至合法后代。基于1,456张图像、82个物种的筛选数据集,在相同设置下对比分层与扁平模型。H-COFGS在物种级达到69.4%准确率,与扁平基线持平,但所有上层分类表现更优。物种预测失败时,92.5%的错误可在属级正确识别,优于扁平模型的67.2%。平均分类距离降低38.2%(1.209 vs. 1.955)。渐进训练揭示双向机制:自顶向下的约束掩码缩小预测空间,自底向上的梯度通过共享骨干反向传播,优化特征表示。使门级准确率从96.2%提升至99.5%,上层获6-8%增益,产出更鲁棒、可解释且符合生物学规律的多级分类结果。

原文摘要 · Abstract (English)

Accurate taxonomic identification of diatoms is essential for aquatic ecosystem monitoring, yet conventional methods depend heavily on expert taxonomists. Recent deep learning approaches improve automation, but most treat diatom recognition as flat classification, predicting only one taxonomic rank. We investigate whether embedding taxonomic hierarchy into neural network architectures can improve both accuracy and error locality. We introduce DiatomCascadeNet (H-COFGS), a hierarchical convolutional network with five cascaded heads that jointly predict class, order, family, genus, and species. Each head receives shared backbone features and probability distributions from higher levels, with binary masks restricting predictions to valid descendants during training and inference. Using a filtered dataset of 1,456 diatom images covering 82 species, we compare hierarchical and flat models under identical settings. H-COFGS matches flat baselines at the species level (69.4% accuracy) while outperforming at all upper taxonomic levels. When species predictions fail, errors remain taxonomically local: 92.5% of misclassified species are correctly predicted at the genus level, versus 67.2% for flat baselines. H-COFGS reduces mean taxonomic distance by 38.2% (1.209 vs. 1.955). Progressive training reveals bidirectional mechanisms: hierarchical constraint masks operate top-down to constrain prediction space, while gradients from fine-grained levels propagate bottom-up through the shared backbone, refining features. This improves class accuracy from 96.2% to 99.5% and yields 6-8% gains at upper levels, producing more robust, interpretable, and biologically aligned predictions for multi-level taxonomic classification.

分类硅藻分层网络生态监测

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