国内刊号:51-1196/TP
国际刊号:1001-3695
发布日期:
作者:李林,刘政,南海,张泽崴,魏晔,
单位:重庆理工大学计算机科学与工程学院,重庆400054;
关键词:深度强化学习,主动学习,图像语义分割,决斗网络,门控循环单元,
基金:重庆市教育委员会科学技术研究项目(KJQN202101149);重庆市基础研究与前沿探索专项资助项目(CSTB2022NSCQ-MSX0918,CSTB2022NSCQ-MSX0493);重庆理工大学研究生创新资助项目(gzlcx20233251);;
针对在图像语义分割任务中获取像素标签困难和分割数据集类别不平衡的问题,提出了一种基于改进双重深度Q网络的主动学习语义分割模型CG_D3QN。引入了一种结合决斗网络结构以及门控循环单元的混合网络结构,通过减轻Q值过估计问题和有效地利用历史状态信息,提高了策略评估的准确性和计算效率。在CamVid和Cityscapes数据集上,该模型相较于基线方法,所需的样本标注量减少了65.0%,同时对于少样本标签的类别,其平均交并比提升了约1%~3%。实验结果表明,该模型能够显著减少样本标注成本并有效地缓解了类别不平衡问题,且对于不同的分割网络也具有适用性。
来源:2024年第11期
《计算机应用研究》期刊编辑部