基于跨域知识迁移的地下目标识别

    Underground Target Recognition Based on Cross-domain Knowledge Transfer

    • 摘要: 在地下目标识别领域,小样本问题限制了深度学习模型的训练效果和泛化能力。尽管生成对抗网络能够扩充数据集,但生成图像与真实图像间的差异易加剧过拟合。为此,文中提出了一种跨域知识迁移模型。模型以YOLOv8为基础,首先提取生成图像(源域)的特征,然后通过知识蒸馏技术,将源域中的知识有效传递到真实图像(目标域),最后结合改进的掩码生成蒸馏与逻辑损失相结合的损失函数,进一步优化了迁移过程。实验结果表明,该模型在小样本条件下,识别准确率从80.60% 提升至92.62%,交并比为0.5时的平均精确度平均值从0.864提升至0.908,不仅有效缓解了小样本问题,还显著提升了模型的识别准确率和泛化能力。

       

      Abstract: In the field of underground target recognition, the small sample problem limits the training effectiveness and generalization ability of deep learning models. Although generative adversarial networks can augment datasets, the discrepancy between generated images and real images tends to exacerbate overfitting. To address this issue, a cross-domain knowledge transfer model is proposed in this paper. Based on YOLOv8, the proposed model first extracts features from generated images (source domain) and then effectively transfers knowledge from the source domain to real images (target domain) through knowledge distillation techniques. Furthermore, an optimized transfer process is achieved by incorporating an improved loss function that combines masked generative distillation and Logit loss. Experimental results demonstrate that under small sample conditions, the proposed model enhances recognition accuracy from 80.60% to 92.62% and improves the mean average precision at an intersection over union of 0.5 from 0.864 to 0.908. This not only effectively mitigates the small sample problem but also significantly boosts the recognition accuracy and generalization ability of the model.

       

    /

    返回文章
    返回