用专家知识校准模型决策,提升稀有树种分类准确率
Expert Knowledge-Guided Decision Calibration for Accurate Fine-Grained Tree Species Classification
- 引入专家知识引导模块,聚焦关键判别特征
- 通过不确定性校准使分类精度提升11.46%、准确率增6.42%
- 轻量级插件设计,仅需0.08M参数即可显著增强模型
精准的细粒度树种分类对森林清查和生物多样性监测至关重要。现有方法多聚焦于复杂架构以适配局部数据分布,但常忽略有限数据中的长尾分布与高类别相似性问题,难以区分少样本或易混淆类别。受人类知识传播中主动求助专家的启发,本文提出外部‘领域专家’机制,构建专家知识引导的分类决策校准网络(EKDC-Net)。该框架解决两个核心问题:专家知识提取与利用。首先设计局部先验引导的知识提取模块(LPKEM),基于类激活图(CAM)分析引导专家关注关键判别特征;随后提出不确定性引导的决策校准模块(UDCM),动态结合全局类别不确定性和实例级预测不确定性来修正本地模型决策。此外,构建涵盖102个树种的大规模数据集CU-Tree102,以缓解现有基准数据多样性不足的问题。在三个基准数据集上的实验表明,本方法达当前最优性能。作为轻量级即插即用模块,仅增加0.08M可学习参数,即可使主干网络准确率提升6.42%、精确率提升11.46%。代码与预训练模型已开源。
原文摘要 · Abstract (English)
Accurate fine-grained tree species classification is critical for forest inventory and biodiversity monitoring. Existing methods predominantly focus on designing complex architectures to fit local data distributions. However, they often overlook the long-tailed distributions and high inter-class similarity inherent in limited data, thereby struggling to distinguish between few-shot or confusing categories. In the process of knowledge dissemination in the human world, individuals will actively seek expert assistance to transcend the limitations of local thinking. Inspired by this, we introduce an external "Domain Expert" and propose an Expert Knowledge-Guided Classification Decision Calibration Network (EKDC-Net) to overcome these challenges. Our framework addresses two core issues: expert knowledge extraction and utilization. Specifically, we first develop a Local Prior Guided Knowledge Extraction Module (LPKEM). By leveraging Class Activation Map (CAM) analysis, LPKEM guides the domain expert to focus exclusively on discriminative features essential for classification. Subsequently, to effectively integrate this knowledge, we design an Uncertainty-Guided Decision Calibration Module (UDCM). This module dynamically corrects the local model's decisions by considering both overall category uncertainty and instance-level prediction uncertainty. Furthermore, we present a large-scale classification dataset covering 102 tree species, named CU-Tree102 to address the issue of scarce diversity in current benchmarks. Experiments on three benchmark datasets demonstrate that our approach achieves state-of-the-art performance. Crucially, as a lightweight plug-and-play module, EKDC-Net improves backbone accuracy by 6.42% and precision by 11.46% using only 0.08M additional learnable parameters. The dataset, code, and pre-trained models are available at https://github.com/WHU-USI3DV/TreeCLS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。