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

作者：王少晖; 莫建文^*

来源：桂林电子科技大学学报, 2020, 40(03): 178-182.

DOI：10.16725/j.cnki.cn45-1351/tn.2020.03.002

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

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

• 单位
  桂林电子科技大学