Tilbake til Nyheter
InnovasjonAI Understanding orientering

DataKernelBench tester om LLM-er kan optimalisere GPU-databasespørringer

En ny benchmark evaluerer om språkmodeller kan generere og forbedre GPU-kode for spørringer i databasestil, med papirrapporteringshastigheter på opptil 2,11× på én H100 GPU og 2,54× over fire H100 GPUer.

5 min readRead the primary source
Primary-source image accompanying DataKernelBench tests whether LLMs can optimize GPU database queries
PrimærkildedokumentKilde registrert
Utgiver
arxiv.org
Kilde lenke
arxiv.orghttps://arxiv.org/abs/2608.25061
Kildetype
Primærdokument – en offisiell kunngjøring, papir, arkivering eller førstepartsside vi leser direkte.
KontekstForstå dette på 60 sekunder

Start her

Nøkkelord

Stor språkmodell (LLM)
En språkmodell trent på massive tekstkorpus for å generere og analysere tekst.
Minne (agentminne)
Lagret kontekst en AI-agent bruker på tvers av trinn eller økter for å forbedre kontinuiteten.
Generalisering
Hvor godt en modell presterer på nye, usynlige data utenfor treningssettet.
Test deg selvQuiz for forklaring av AI-modeller

Hva skjedde

Forskere introduserte DataKernelBench, en målestokk for å teste om store språkmodeller kan optimere uregelmessige, databevegelsestunge databaseoperasjoner på GPUer. Systemet oversetter SQL-spørringer til validerte PyTorch TorchPlan-programmer, og evaluerer deretter modeller etter hvert som de optimerer enten en sentral tensor-avgrenset kodeseksjon eller hele spørringen i CUDA eller Triton.

Papiret, sendt til arXiv 25. august 2026 og identifisert i kilden som akseptert på EMNLP 2026, presenterer DataKernelBench som en evaluering spesifikt for AI-generert optimalisering av databasespørringer på GPUer. Forfatterne hevder at eksisterende LLM-kjernereferanser ikke tester operatører i databasestil tilstrekkelig, som kan være uregelmessige, heterogene og dominert av databevegelse. Det gjør referansemålet annerledes enn de mer vanlige operatørene som ofte brukes til å vurdere generert GPU-kode.

DataKernelBench konverterer SQL til validerte PyTorch TorchPlan-programmer. Modellene blir deretter bedt om å optimalisere enten en kjernetensor-begrenset kodebit eller hele spørringen, ved å bruke CUDA eller Triton. Evalueringen inkluderer utførelsesveiledet reparasjon, noe som betyr at de genererte programmene blir testet og revidert gjennom tilbakemelding fra utførelsen. Sammendraget sier at studien dekker ti proprietære og åpenvektsmodeller på TPC-H SF10-arbeidsmengden ved bruk av en H100 GPU.

I følge avisens rapporterte resultater oppnådde den sterkeste fullsøkende CUDA-konfigurasjonen en 2,11× hastighetsøkning over sammenligningens grunnlinje ved full beståtthastighet. Kilden avslører ikke grunnlinjens navn fordi abstraktet inneholder en misformet lenke i den posisjonen, så det nøyaktige referansepunktet kan ikke identifiseres fra det medfølgende materialet. Forfatterne rapporterer også om en utvidelse i større skala: TorchPlan ble kombinert med Dask-cuDF for data større enn GPU-minne, og på TPC-H SF100 ved bruk av fire H100 GPUer, oppnådde systemet en rapportert 2,54× speedup.

Til sammen definerer oppsettet en sekvens fra spørrepresentasjon til generert programkjøring og rapportert ytelse. SQL-inngangen er representert som et validert TorchPlan-program, mens optimaliseringsmålet kan være enten en sentral tensor-avgrenset seksjon eller hele spørringen. Det modellgenererte resultatet blir ikke behandlet som fullstendig bare fordi det er skrevet; evalueringen bruker utførelsesveiledet reparasjon for å teste og revidere programmer gjennom tilbakemelding fra utførelse. Maskinvare- og arbeidsbelastningsinnstillingene er også en del av det rapporterte eksperimentet: sammendraget beskriver ti proprietære og åpenvektsmodeller, TPC-H SF10-arbeidsbelastningen og en H100 GPU. Den beskriver separat den større TorchPlan- og Dask-cuDF-konfigurasjonen på TPC-H SF100 med fire H100 GPUer. Innenfor denne strukturen er de rapporterte hastighetsøkningene utfall av de angitte konfigurasjonene og passtilstanden, mens den misformede grunnlinjekoblingen etterlater sammenligningsreferansen uten navn i den oppgitte kilden. Dette er omfanget av det abstraktet gir om referanseindeksen og dens rapporterte evaluering. Beskrivelsen identifiserer følgelig objektene som optimaliseres, programmeringsvalgene, reparasjonsmekanismen, den evaluerte arbeidsbelastningen, maskinvareinnstillingene og de to rapporterte ytelsesresultatene, men den legger ikke til detaljer utover de som er gitt i abstraktet.

Kildedetaljer: arxiv.org ↗

Hvorfor det betyr noe

Arbeidet adresserer et gap i eksisterende LLM-kodingsreferanser, som forfatterne sier i stor grad har fokusert på maskinlæringsoperatører i stedet for databasearbeidsbelastninger. Hvis de rapporterte resultatene holder seg på tvers av bredere arbeidsbelastninger, kan språkmodeller bli nyttige assistenter for den spesialiserte kjerneteknikken som trengs for å gjøre GPU-akselererte databaser raskere.

Den praktiske betydningen er at databaseytelse ofte avhenger av arbeidsbelastningsspesifikke implementeringsvalg, ikke bare av å velge et raskere generellt system. Artikkelens sentrale påstand er at LLM-er kan delta i denne spesialiserte optimaliseringsprosessen ved å generere og reparere GPU-programmer for komplette spørsmål. Det setter modellen nærmere strukturen til en faktisk databasearbeidsbelastning enn en benchmark som evaluerer isolerte maskinlæringskjerner.

De rapporterte funnene peker også på en arbeidsdeling mellom modellkapasitet og informasjon om arbeidsbelastning. Forfatterne sier at implementeringer med høyere ytelse vanligvis bruker kjernefusjon og endringer i utførelsesstrategi. De rapporterer også at arbeidsbelastningskontekst er viktigere enn maskinvarekontekst, og at sterkere modeller drar mest nytte av full-søkespesialisering. Rent praktisk antyder resultatet at det å gi modellen en detaljert beskrivelse av spørringen og dens data kan ha mer betydning enn bare å beskrive GPUen som koden skal kjøres på.

Resultatene er konsekvente som en forskningsretning, men de er ikke bevis på at databaseteknikk har blitt automatisert i produksjonen. Kilden beskriver en benchmark og kontrollerte eksperimenter, ikke distribusjon i en live databasetjeneste. Den fastslår heller ikke at den genererte koden er konsekvent korrekt utenfor de testede spørringene, at den er billigere å produsere enn menneskeskrevne kjerner, eller at hastighetsoppgraderinger ville overleve endrede datadistribusjoner og driftskrav. Disse grensene er viktige fordi en rask spørring som mislykkes på en kantsak ikke er en brukbar databaseoptimalisering.

Interactive Mechanism

Interaktiv mekanisme: Hvordan det faktisk fungerer

Utforsk den underliggende teknologien bak denne utviklingen interaktivt.

Document Size:128K tokens
Needle Placement Depth (Location in document):50% into text
Attention Context Buffer Map:
Target Fact (50%)
Equivalent Pages~320Standard book pages
Retrieval Accuracy99.9%Needle recall score
RAM / KV Cache5.1 GBMemory overhead
Prompt CachingActive~80% discount on reuse
Core takeaway: Million-token context windows allow querying whole codebases or legal archives in one prompt. However, KV cache memory scales with context length, making prompt caching crucial for real-time production.
Interaktiv konseptsjekk+10 Points
AI Models Explained Quiz

Which component of an AI application is the machine-learning model itself?

Hva du skal se neste

Hovedspørsmålene er om resultatene generaliserer utover TPC-H-arbeidsbelastningene som er testet, hvordan referansen definerer en full bestått, og om de rapporterte speedupene forblir etter å ha tatt hensyn til utviklings-, validerings- og maskinvarekostnader. Den medfølgende kilden er et abstrakt, så disse detaljene og uavhengig replikering forblir uløst.

Det første problemet å se er reproduserbarhet. Sammendraget identifiserer antall og brede typer modeller, maskinvaren og TPC-H-skalafaktorene, men det gir ikke navn til modellene, beskriver spørsmålene deres, spesifiserer beregningen av bestått rate eller gir den fullstendige sammenligningsgrunnlinjen. Disse detaljene vil avgjøre hvor rettferdig systemene ble evaluert og hvor lett andre kan gjenta eksperimentene.

Generalisering er et annet åpent spørsmål. De rapporterte testene bruker TPC-H SF10 på én H100 GPU og TPC-H SF100 på fire H100 GPUer. Kilden sier ikke om DataKernelBench dekker andre spørringsfamilier, databasemotorer, datadistribusjoner, GPU-generasjoner eller blandede CPU-GPU-distribusjoner. Den fastslår heller ikke hvordan tilnærmingen oppfører seg når dataene overskrider minnet på mindre strukturerte måter, eller når korrekthet og latens må opprettholdes under skiftende produksjonsbelastninger.

Til slutt bør fremtidige evalueringer skille rå utførelseshastighet fra hele kostnaden ved å bruke en LLM-basert optimizer. Relevante tiltak vil inkludere genereringstid, antall reparasjonsforsøk, valideringsoverhead, GPU-bruk, minnebruk og kostnadene for mislykkede eller avviste programmer. Kilden sier at utførelsesveiledet reparasjon er en del av metoden, men abstraktet gir ingen tall for den prosessen. Inntil disse ukjente er rapportert og uavhengig testet, bør tallene 2,11× og 2,54× behandles som resultater som hevdes av denne artikkelen under de angitte eksperimentelle forholdene, ikke som generelle ytelsesgarantier.

Relaterte guider og quizer

AI-modeller forklartAI treningTransformatorerTest det du vet – prøv en gratis AI-quizSlå opp et AI-begrep i ordlisten vårFølg AI-modellutgivelsessporeren
Fant du dette nyttig?