Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Rethinking Repetition Problems of LLMs in Code Generation

Created by
  • Haebom

저자

Yihong Dong, Yuchen Liu, Xue Jiang, Zhi Jin, Ge Li

개요

본 논문은 신경망 언어 모델을 이용한 코드 생성에서 반복 문제를 해결하기 위해, 기존 연구에서 주로 다루던 내용 반복(content repetition)이 아닌 구조적 반복(structural repetition) 문제에 초점을 맞추고 있습니다. 구조적 반복은 다양한 패턴으로 나타나지만 고정된 구조를 가지는 코드 반복을 의미하며, 문법적으로 반영될 수 있습니다. 본 논문에서는 구조적 반복을 공식적으로 정의하고, 문법 기반 반복 패널티(Repetition Penalization based on Grammar, RPG)라는 효율적인 디코딩 방법을 제안합니다. RPG는 문법 규칙을 활용하여 코드 생성 중 반복 문제를 식별하고, 반복에 기여하는 중요 토큰의 가능성을 전략적으로 감소시켜 코드 생성에서 반복을 완화합니다. 또한, 코드 생성에서 반복 문제 완화 방식을 종합적으로 평가하기 위한 새로운 데이터셋 CodeRepetEval을 구축했습니다. 실험 결과, RPG는 CodeRepetEval 데이터셋뿐만 아니라 HumanEval 및 MBPP 벤치마크에서도 기존 최고 성능 기준 모델을 상당히 능가하며, 반복을 효과적으로 줄이고 생성된 코드의 품질을 향상시키는 것으로 나타났습니다.

시사점, 한계점

시사점:
코드 생성에서 구조적 반복 문제를 명확히 정의하고, 이를 해결하기 위한 효과적인 방법인 RPG를 제시.
문법 기반의 접근 방식을 통해 기존 연구보다 더욱 근본적인 반복 문제 해결에 기여.
새로운 평가 데이터셋 CodeRepetEval을 제공하여 코드 생성에서의 반복 문제 연구를 위한 기반 마련.
RPG를 통해 생성된 코드의 품질 향상 및 반복 감소 효과를 실험적으로 검증.
한계점:
CodeRepetEval 데이터셋의 규모 및 다양성에 대한 추가적인 검토 필요.
RPG의 성능이 특정 유형의 코드 또는 프로그래밍 언어에 편향될 가능성 존재.
다양한 복잡도의 코드 생성에 대한 RPG의 일반화 성능에 대한 추가 연구 필요.
문법 규칙의 정확성 및 완전성이 RPG의 성능에 영향을 미칠 수 있음.
👍