HyperAIHyperAI

Command Palette

Search for a command to run...

CHASE-SQL: テキストからSQLへの変換における多経路推論と嗜好最適化された候補選択

Mohammadreza Pourreza extsuperscript1* Hailong Li extsuperscript1* Ruoxi Sun extsuperscript1 Yeounoh Chung extsuperscript1 Shayan Talaei extsuperscript2 Gaurav Tarlok Kakkar extsuperscript1 Yu Gan extsuperscript1 Amin Saberi extsuperscript2 Fatma Özcan extsuperscript1 Sercan Ö. Arik extsuperscript1

概要

大規模言語モデル(LLM)のテキストからSQLへの変換タスク(Text-to-SQL)における性能向上の課題に取り組むため、CHASE-SQLという新しいフレームワークを提案します。CHASE-SQLは、テスト時の計算リソースを多エージェントモデリングに活用する革新的な戦略を採用し、候補の生成と選択を改善します。CHASE-SQLは、LLMが持つ内在的な知識を利用して、異なるLLMジェネレーターによって多様で高品質なSQL候補を生成します。具体的には以下の方法を使用しています:(1) 単一のLLM呼び出しで複雑なクエリーを管理可能なサブクエリーに分解する分割統治法;(2) データベースエンジンが実行時に取る手順を反映したクエリー実行計画に基づく思考連鎖推論;(3) テスト問題に特化した具体的な少ショットデモンストレーションを提供する独自のインスタンス認識型合成例生成技術。最適な候補を選定するために、選択エージェントが微調整された二値候補選択用LLMを使用して、候補間のペアワイズ比較を行いランキング付けを行います。この選択手法は他の手法よりも堅牢であることが示されています。提案されるジェネレーター-セレクター・フレームワークは、SQLクエリーの質と多様性を向上させるだけでなく、従来の方法よりも優れた性能を発揮します。全体として、我々が提案するCHASE-SQLは著名なBIRD Text-to-SQLデータセットベンチマークのテストセットおよび開発セットにおいて、最先端の実行精度73.0%および73.01%を達成しており、論文提出時点ではリーダーボードでの最高提出となっています。


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています