• • 下一篇
杜艺,续明进,孔佳仪,王力瑶,赵晨
摘要: 针对复杂场景下目标检测任务中深度学习算法泛化性、鲁棒性受限以及全参数微调(FPFT)计算成本高的问题,提出一种基于YOLOv11(You Only Look Once version 11)的低秩自适应参数高效微调算法。首先,在YOLOv11骨干和颈部网络中嵌入低秩自适应(LrAn)模块;其次,结合低秩自适应(LoRA)、权重分解低秩自适应(DoRA)和主奇异值与奇异向量自适应(PiSSA)三种低秩分解算法,通过权重分解与动态调整机制实现参数的高效更新;最后,在训练过程中,将YOLOv11网络的绝大部分预训练权重保持冻结状态,仅对LrAn模块中由三种低秩分解算法生成的低秩矩阵进行训练,将可训练参数规模缩减至原算法的1.56%。COCO(Common Objects in Context)数据集实验表明,所提算法相较基线YOLOv11算法在精确度、召回率和平均精度均值指标上分别提升4.18、7.11和7.85个百分点。可见,所提算法为资源受限场景下的大型检测算法轻量化与高效微调提供了有效技术路径。
中图分类号: