基于分层表示和上下文增强的类摘要生成技术

来源：计算机研究与发展杂志2024年第2期北京时间：

作者：陈豪伶,虞慧群,范贵生,李明辰,黄子杰,

摘要：代码摘要是源代码的自然语言解释，高质量的代码摘要有助于提高开发人员程序理解效率.近年来，代码自动摘要的研究集中在为方法粒度的代码片段生成摘要.然而，对于面向对象的语言，例如Java，类才是项目的基本组成单元.基于上述问题，提出一种基于分层表示和上下文增强的类摘要生成方法HRCE（hierarchicalrepresentationandcontextenhancement），并构建了一个包含358992个\langleJava类，上下文，摘要\rangle数据对的类摘要数据集.HRCE使用代码精简策略去除类的非关键代码，从而缩短代码长度.然后，对类的层次结构，包括类签名、属性和方法分别进行建模，获得类的语义信息和层次结构信息.此外，从项目中抽取父类的签名及摘要来刻画类在项目中依赖的上下文.实验表明，基于分层表示和上下文增强的类摘要生成模型能够表征代码的语义和层次结构，并可以从目标类的内部和外部获取信息.HRCE在BLEU，METEOR，ROUGE-L等评估指标上超过了所有基准模型.

关键词：代码自动摘要, 分层表示, 上下文增强, 深度学习, 类摘要,