基于门控循环单元的图像描述方法

王少晖, 莫建文

桂林电子科技大学学报 ›› 2020, Vol. 40 ›› Issue (03) : 178-182. DOI: 10.16725/j.cnki.cn45-1351/tn.2020.03.002

基于门控循环单元的图像描述方法

  • 王少晖, 莫建文
作者信息 +
History +

摘要

为了使图像描述更快地生成语句和更高效地利用计算机资源,提出一种基于门控循环单元的图像描述方法。该方法通过门控循环单元优化RNN子网内部结构和运算过程,降低了RNN子网的时间复杂度和空间复杂度。实验结果表明,该方法生成语句的时间显著减少,其中CPU模式下RNN子网平均生成语句的时间减少0.03 s(相当于减少3.15%);GPU模式下RNN子网平均生成语句的时间减少0.01 s(相当于减少2.5%);RNN子网的参数大约减少了77万个,所占内存降低了25%。

关键词

图像描述 / 神经网络 / 长短期记忆网络 / 门控循环单元

中图分类号

TP391.41 / TP183

引用本文

导出引用
王少晖, 莫建文. 基于门控循环单元的图像描述方法. 桂林电子科技大学学报. 2020, 40(03): 178-182 https://doi.org/10.16725/j.cnki.cn45-1351/tn.2020.03.002

基金

国家自然科学基金(61661017); 广西自然科学基金(GXNSFAA380149)

评论

Accesses

Citation

Detail

段落导航
相关文章

/