为地理空间代码生成构建首个专用评估基准,助力开发更精准的编程助手。
Evaluation of Code LLMs on Geospatial Code Generation
- 基于任务复杂度分类构建地理空间编码问题数据集
- 设计可自动验证正确性的测试场景,覆盖空间推理与工具使用
- 公开数据与代码,推动地理空间专用LLM模型发展
软件开发辅助工具研究已久,近期基于大语言模型(LLMs)的代码生成方法在数据科学和机器学习领域表现优异。然而,地理空间数据科学面临独特挑战,现有模型通常未在此类任务上进行评估。本文构建了一个针对地理空间代码生成的评估基准,依据任务复杂度和所需工具对地理空间任务进行分类,并人工创建高质量编码问题,涵盖空间推理、空间数据处理及地理空间工具使用。每个问题配备可自动验证正确性的测试场景。我们评估了若干现有代码生成LLMs在该领域的表现,并将数据集与可复现的评估代码开源至公共GitHub仓库,旨在为未来新模型提供标准化评测基准。该数据集有望推动高精度地理空间编码模型的发展,支持定制化地理空间编程助手的实现。
原文摘要 · Abstract (English)
Software development support tools have been studied for a long time, with recent approaches using Large Language Models (LLMs) for code generation. These models can generate Python code for data science and machine learning applications. LLMs are helpful for software engineers because they increase productivity in daily work. An LLM can also serve as a "mentor" for inexperienced software developers, and be a viable learning support. High-quality code generation with LLMs can also be beneficial in geospatial data science. However, this domain poses different challenges, and code generation LLMs are typically not evaluated on geospatial tasks. Here, we show how we constructed an evaluation benchmark for code generation models, based on a selection of geospatial tasks. We categorised geospatial tasks based on their complexity and required tools. Then, we created a dataset with tasks that test model capabilities in spatial reasoning, spatial data processing, and geospatial tools usage. The dataset consists of specific coding problems that were manually created for high quality. For every problem, we proposed a set of test scenarios that make it possible to automatically check the generated code for correctness. In addition, we tested a selection of existing code generation LLMs for code generation in the geospatial domain. We share our dataset and reproducible evaluation code on a public GitHub repository, arguing that this can serve as an evaluation benchmark for new LLMs in the future. Our dataset will hopefully contribute to the development new models capable of solving geospatial coding tasks with high accuracy. These models will enable the creation of coding assistants tailored for geospatial applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。