Evaluating In-Context Learning of Libraries for Code Generation

التفاصيل البيبلوغرافية
العنوان: Evaluating In-Context Learning of Libraries for Code Generation
المؤلفون: Patel, Arkil, Reddy, Siva, Bahdanau, Dzmitry, Dasigi, Pradeep
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computation and Language
الوصف: Contemporary Large Language Models (LLMs) exhibit a high degree of code generation and comprehension capability. A particularly promising area is their ability to interpret code modules from unfamiliar libraries for solving user-instructed tasks. Recent work has shown that large proprietary LLMs can learn novel library usage in-context from demonstrations. These results raise several open questions: whether demonstrations of library usage is required, whether smaller (and more open) models also possess such capabilities, etc. In this work, we take a broader approach by systematically evaluating a diverse array of LLMs across three scenarios reflecting varying levels of domain specialization to understand their abilities and limitations in generating code based on libraries defined in-context. Our results show that even smaller open-source LLMs like Llama-2 and StarCoder demonstrate an adept understanding of novel code libraries based on specification presented in-context. Our findings further reveal that LLMs exhibit a surprisingly high proficiency in learning novel library modules even when provided with just natural language descriptions or raw code implementations of the functions, which are often cheaper to obtain than demonstrations. Overall, our results pave the way for harnessing LLMs in more adaptable and dynamic coding environments.
Comment: NAACL 2024
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2311.09635
رقم الأكسشن: edsarx.2311.09635
قاعدة البيانات: arXiv