Agentic coding without the cloud: evaluating open-weight large language models on longitudinal data preparation tasks

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics, Software Development & Engineering · Depth: Expert, quick

Summary

An open-source framework has been introduced to evaluate AI agents powered by open-weight large language models (LLMs) for data preparation in governance-restricted research. This framework addresses the challenge of using LLMs with sensitive personal data, where transmitting data to cloud-based services is prohibited. It comprises a curated ground-truth dataset, including cleaning scripts for six sweeps of data from a British cohort study, task definitions like category harmonization and multi-wave merging, and automated routines for evaluating LLM-produced R code and outputted data. Benchmarking 31-35B parameter models on consumer-grade hardware across 20 data preparation tasks (creating 102 variables) showed an "average task completion" up to 87.9%. These results indicate a viable path for AI-assisted data preparation using locally deployable open-weight LLMs in settings with strict data governance.

Key takeaway

For research scientists or data scientists handling sensitive personal data under strict governance, you should consider locally deploying open-weight large language models for data preparation. This approach bypasses restrictions on transmitting data to cloud services, offering a secure and effective alternative. Models in the 31-35B parameter range, running on consumer-grade hardware, demonstrated up to 87.9% task completion, indicating their readiness for real-world application in such environments. Explore the RRBench framework to evaluate and integrate these agentic coding solutions into your workflows.

Key insights

Open-weight LLMs running locally can effectively automate data preparation for sensitive data in governance-restricted research settings.

Principles

Method

Evaluate open-weight LLM agents using a framework with ground-truth cleaning scripts, defined data preparation tasks like harmonization and merging, and automated routines to assess generated R code and outputted data.

In practice

Topics

Code references

Best for: AI Engineer, Machine Learning Engineer, CTO, AI Scientist, Research Scientist, Data Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.