Exploring Language Model’s Code Generation Ability with Auxiliary Functions

Seonghyeon Lee, Sanghwan Jang, Seongbo Jang, Dongha Lee, Hwanjo Yu


Abstract
Auxiliary function is a helpful component to improve language model’s code generation ability. However, a systematic exploration of how they affect has yet to be done. In this work, we comprehensively evaluate the ability to utilize auxiliary functions encoded in recent code-pretrained language models. First, we construct a human-crafted evaluation set, called HumanExtension, which contains examples of two functions where one function assists the other.With HumanExtension, we design several experiments to examine their ability in a multifaceted way. Our evaluation processes enable a comprehensive understanding of including auxiliary functions in the prompt in terms of effectiveness and robustness. An additional implementation style analysis captures the models’ various implementation patterns when they access the auxiliary function. Through this analysis, we discover the models’ promising ability to utilize auxiliary functions including their self-improving behavior by implementing the two functions step-by-step. However, our analysis also reveals the model’s underutilized behavior to call the auxiliary function, suggesting the future direction to enhance their implementation by eliciting the auxiliary function call ability encoded in the models. We release our code and dataset to facilitate this research direction.
Anthology ID:
2024.findings-naacl.181
Volume:
Findings of the Association for Computational Linguistics: NAACL 2024
Month:
June
Year:
2024
Address:
Mexico City, Mexico
Editors:
Kevin Duh, Helena Gomez, Steven Bethard
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
2836–2848
Language:
URL:
https://aclanthology.org/2024.findings-naacl.181
DOI:
10.18653/v1/2024.findings-naacl.181
Bibkey:
Cite (ACL):
Seonghyeon Lee, Sanghwan Jang, Seongbo Jang, Dongha Lee, and Hwanjo Yu. 2024. Exploring Language Model’s Code Generation Ability with Auxiliary Functions. In Findings of the Association for Computational Linguistics: NAACL 2024, pages 2836–2848, Mexico City, Mexico. Association for Computational Linguistics.
Cite (Informal):
Exploring Language Model’s Code Generation Ability with Auxiliary Functions (Lee et al., Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-naacl.181.pdf