本論文では、SQL-PaLMと呼ばれる包括的なフレームワークを提案している。SQL-PaLMは、大規模言語モデルPaLM-2のUnicorn変種を活用し、Text-to-SQLタスクに適応させることを目的としている。
まず、プロンプティングによるアプローチを評価し、その後、より優れたパフォーマンスを発揮するためのファインチューニングに焦点を当てている。既存研究がプロンプティング戦略や比較的小規模な言語モデルのファインチューニングに主に注目しているのに対し、SQL-PaLMは大規模な言語モデルのチューニングに注力している。大規模モデルには、小規模モデルと比べて大幅に向上した理解力と推論能力といった特性がある。
SQL-PaLMでは、Text-to-SQLパフォーマンスに影響を与える重要な側面を系統的に調査・分析している。まず、トレーニングデータの多様性と網羅性が重要であり、タスクやベンチマーク間での知見を提示している。人的コストを抑えつつトレーニングデータの網羅性を向上させるため、大規模な言語モデル生成の合成データも提案している。
次に、入力表現の質が全体的な品質に大きな影響を及ぼすことを示している。データベースの値、列の説明、ヒントなど、さまざまな情報を活用してText-to-SQLをファインチューニングする方法を詳細に検討している。
さらに、実世界のデータベースサイズに対応するために、関連する列のみをインプットとして効率的に選択する手法を提案している。この手法により、コンテキストサイズを大幅に削減しつつ、パフォーマンスへの影響は最小限に抑えられる。特に、プログラムを利用した列選択と検索ベースの列選択アプローチを提案し、それらの統合方法について検討している。
最後に、実行結果に基づくテスト時の精選によって、複数の学習パラダイムの出力を統合し、精度の向上を図っている。
Vers une autre langue
à partir du contenu source
arxiv.org
Questions plus approfondies