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

王少晖, 莫建文

doi:10.16725/j.cnki.cn45-1351/tn.2020.03.002

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

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

王少晖, 莫建文

作者信息 +

Author information +

文章历史 +

PDF

摘要

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