Subscribe Now
Trending News
Shorts

QORL učí malý model hledat rychlejší plány Postgresu

QORL učí malý model hledat rychlejší plány Postgresu

Rohan Bansal zveřejnil QORL, výzkumný harness pro trénování agenta nad optimalizátorem PostgreSQL. Místo přepisování planneru nechá model navrhnout hinty pro rozšíření pg_hint_plan, které PostgreSQL nasměrují k jinému fyzickému plánu. Databáze pak kandidáta spustí a porovná jeho dobu běhu s vlastním výchozím plánem. Smysl to má hlavně u náročných analytických dotazů, které se vracejí pořád dokola; cena hledání lepšího plánu se pak může rozložit mezi mnoho spuštění.

Autor začal se čtyřmiliardovým modelem z rodiny Qwen. Nejdřív jej pomocí supervised fine-tuningu naučil pracovat s nástroji QORL, potom jej doladil agentním reinforcement learningem. Odměnou není známka za hezký SQL výstup, ale naměřený čas dotazu. Repozitář obsahuje postupy pro kalibraci měření, vyhodnocení, SFT i RL a přikládá pracovní zátěže JOB a CEB nad databázovou sadou IMDb.

Ve svém testu na 113 joinově náročných dotazech JOB Bansal uvádí pro nejlepší kandidát vybraný napříč třemi běhy geometrický průměr zrychlení 1,81×, 68 zlepšení a žádnou regresi. Jde o výsledek autorova experimentu, nikoli o příslib pro libovolnou produkční databázi. QORL má otevřený zdrojový kód; pro trénování ale počítá s Linuxem x86-64, Pythonem 3.12 a pro GPU běh s vLLM, FlashAttention a CUDA závislostmi.

Zdroje:

Related posts