Dopamin: Transformer-based Comment Classifiers through Domain Post-Training and Multi-level Layer Aggregation

Code comments provide important information for understanding the source code. They can help developers understand the overall purpose of a function or class, as well as identify bugs and technical debt. However, an overabundance of comments is meaningless and counterproductive. As a result, it is c...

Full description

Saved in:
Bibliographic Details
Published in:2024 IEEE/ACM International Workshop on Natural Language-Based Software Engineering (NLBSE) pp. 61 - 64
Main Authors: Hai, Nam Le, Bui, Nghi D. Q.
Format: Conference Proceeding
Language:English
Published: ACM 20.04.2024
Subjects:
Adaptation models
Codes
Computer languages
Conferences
Semantics
Source coding
Transformers
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Code comments provide important information for understanding the source code. They can help developers understand the overall purpose of a function or class, as well as identify bugs and technical debt. However, an overabundance of comments is meaningless and counterproductive. As a result, it is critical to automatically filter out these comments for specific purposes. In this paper, we present Dopamin, a Transformer-based tool for dealing with this issue. Our model excels not only in presenting knowledge sharing of common categories across multiple languages, but also in achieving robust performance in comment classification by improving comment representation. As a result, it outperforms the STACC baseline by 3 \% on the NLBSE'24 Tool Competition dataset in terms of average F1 score, while maintaining a comparable inference time for practical use. The source code is publicity available at https://github.com/FSoft-AI4Code/Dopamin.
DOI:10.1145/3643787.3648044