CHS-SQL: Ricerca Euristica Guidata dalla Confidenza per il Collegamento dello Schema in Text-to-SQL
Un nuovo framework chiamato CHS-SQL migliora l'accuratezza di Text-to-SQL bilanciando precisione e richiamo durante il Collegamento dello Schema. L'approccio utilizza una ricerca euristica combinata con la confidenza del modello per selezionare sottoinsiemi di schema candidati, affrontando il compromesso ignorato dai metodi precedenti. Consente un fine-tuning efficiente di Small Language Models (SLM) su una singola GPU NVIDIA RTX 4090, raggiungendo prestazioni vicine a quelle dei modelli grandi garantendo al contempo la sicurezza dei dati. La ricerca, pubblicata su arXiv, dimostra che sia la precisione che il richiamo del Collegamento dello Schema influenzano direttamente l'accuratezza finale di SQL.
Fatti principali
- CHS-SQL è un nuovo framework per il fine-tuning di SLM su compiti Text-to-SQL
- Bilancia precisione e richiamo nel Collegamento dello Schema
- Utilizza ricerca euristica combinata con la confidenza del modello
- Funziona su una singola GPU NVIDIA RTX 4090
- Raggiunge prestazioni vicine a quelle dei modelli grandi
- Garantisce la sicurezza dei dati
- Pubblicato su arXiv con ID 2607.22624
- La precisione e il richiamo del Collegamento dello Schema influenzano direttamente l'accuratezza di SQL
Entità
Istituzioni
- arXiv