CoSS: leveraging statement semantics for code summarization

Shi, Chaochen; Cai, Borui; Zhao, Yao; Gao, Longxiang; Sood, Keshav; Xiang, Yong

doi:10.1109/tse.2023.3256362

Cited by 1 publication

(1 citation statement)

References 41 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…Code summarization tasks can be categorized into statement-level [1], function-level [2][3][4], and file-level [5] summarization, corresponding to explaining specific statement meanings, function functionalities, and document purposes, respectively. In this paper, we focus on the function-level code summarization.…”

Section: Introductionmentioning

confidence: 99%

CogCol: Code Graph-Based Contrastive Learning Model for Code Summarization

Shi,

Yin,

et al. 2024

Electronics

View full text Add to dashboard Cite

Summarizing source code by natural language aims to help developers better understand existing code, making software development more efficient. Since source code is highly structured, recent research uses code structure information like Abstract Semantic Tree (AST) to enhance the structure understanding rather than a normal translation task. However, AST can only represent the syntactic relationship of code snippets, which can not reflect high-level relationships like control and data dependency in the program dependency graph. Moreover, researchers treat the AST as the unique structure information of one code snippet corresponding to one summarization. It will be easily affected by simple perturbations as it lacks the understanding of code with similar structure. To handle the above problems, we build CogCol, a Code graph-based Contrastive learning model. CogCol is a Transformer-based model that converts code graphs into unique sequences to enhance the model’s structure learning. In detail, CogCol uses supervised contrastive learning by building several kinds of code graphs as positive samples to enhance the structural representation of code snippets and generalizability. Moreover, experiments on the widely used open-source dataset show that CogCol can significantly improve the state-of-the-art code summarization models under Meteor, BLEU, and ROUGE.

show abstract