-
题名基于类型辅助引导的代码注释生成模型
- 1
-
-
作者
刘利
吕韦岑
汪洋
-
机构
泸州职业技术学院人工智能与大数据学院
成都工贸职业技术学院信息工程学院
-
出处
《无线电通信技术》
北大核心
2024年第4期807-814,共8页
-
基金
泸州市科技计划项目(2021-JYJ-96)。
-
文摘
代码注释生成方法通常基于结构-序列(Structure-Sequence, Struct2Seq)框架,但忽略了代码注释的类型信息,例如操作符、字符串等。由于类型信息之间的层次具有依赖性,将类型信息引入已有的Struct2Seq框架并不适用。为了解决上述问题,提出一种基于类型辅助引导的代码注释生成(Code Comment Generation based on Type-assisted Guidance, CCG-TG)模型,将源代码视为带有类型信息的n元树。该模型包含一个关联类型编码器和一个限制类型解码器,可以对源代码进行自适应总结。此外,提出一种多级强化学习(Multi-level Reinforcement Learning, MRL)方法来优化所提模型的训练过程。在多个数据集上进行实验,与多种基准模型对比,证明所提CCG-TG模型在所有评价指标上的性能最优。
-
关键词
代码注释生成
类型信息
结构序列框架
类型辅助引导
强化学习
-
Keywords
code comment generation
type information
Struct2Seq framework
type-assisted guidance
reinforcement learning
-
分类号
TP311
[自动化与计算机技术—计算机软件与理论]
-