AB
AiBoss
project

OmniSQL - 自然言語をSQLクエリ文に変換するオープンソースのテキストtoSQLモデル。

OmniSQLは、自然言語の問題を効率的にSQLクエリ文に変換するオープンソースのテキストtoSQLモデルです。革新的なデータ合成フレームワークを通じて、初の100万レベルのテキストtoSQLデータセットであるSynSQL-2.5Mを生成します。

OmniSQLとは何ですか?

OmniSQLは、自然言語の問題を効率的にSQLクエリに変換するオープンソースのテキストtoSQLモデルです。革新的なデータ合成フレームワークを活用し、16,000を超えるクロスドメインデータベースを網羅し、さまざまな複雑さと言語スタイルを包含する250万件の高品質サンプルを含む、初の100万件規模のテキストtoSQLデータセット「SynSQL-2.5M」を生成しました。OmniSQLは、7B、14B、32Bのモデルバージョンを提供しており、その微調整にはSpiderおよびBIRDデータセットからの高品質なラベル付きデータが組み込まれています。

OmniSQLの主な機能

  • テキストからSQLへの変換OmniSQLは、ユーザーが自然言語で入力した質問を理解し、それに対応するSQLクエリ文に変換することができます。
  • 複数のデータベースと複雑なクエリをサポートしますOmniSQLは複数のデータベースタイプをサポートしており、単純な単一テーブルクエリから、複雑な複数テーブル結合、サブクエリ、関数呼び出し、共通テーブル式(CTE)まで、さまざまな複雑さのSQLクエリを処理できます。
  • マインドチェーンソリューションを提供するOmniSQLは、SQLクエリを生成するだけでなく、各サンプルに対して思考プロセスに基づいたソリューションを提供します。この思考プロセスは、自然言語の問題を理解してSQLクエリを生成するまでの論理的な推論プロセスを示し、ユーザーがモデルの意思決定経路をより深く理解するのに役立ち、開発者にとってモデルのデバッグと最適化を容易にします。
  • 複数モデルバージョンの選択OmniSQLは、7B、14B、32Bの3種類のモデルサイズを提供しています。ユーザーは、ニーズとコンピューティングリソースに基づいて適切なモデルサイズを選択できます。これらの異なるサイズは、パフォーマンスとリソース消費のバランスを取るように設計されています。小さいモデルは実行速度が速く、リソース消費量も少なく、一方、大きいモデルは特定の複雑なクエリシナリオにおいて優れたパフォーマンスを発揮する場合があります。

OmniSQL の技術原則

  • データベースは自動的に生成されますOmniSQLは、Webテーブルを分析してビジネスシナリオを推測し、大規模な言語モデルを使用して、複数テーブル間の関係と主キー/外部キー制約を備えたデータベース構造を自動的に構築します。拡張戦略を採用することで、列数を増やし、構造を最適化することで、生成されたデータベースを実際のアプリケーションにより適したものにします。
  • 複雑性を考慮したSQLクエリ生成この機能は4つの複雑度レベルを定義し、集計関数(SUM、AVGなど)やウィンドウ関数(ROW_NUMBER、RANKなど)といったSQLite関数ライブラリと組み合わせることで、様々なSQLクエリを生成します。ユーザーの問題に基づいて適切な複雑度レベルをインテリジェントに選択し、適切なクエリ文を提供します。
  • 逆翻訳における様式化の問題このシステムは、SQLクエリを9種類の異なる言語スタイルで自然言語の質問に変換するSQL-to-Question戦略を採用しています。意味解析により、翻訳前後の意味の一貫性が確保され、自然言語からSQLへの変換の効率と精度が向上し、様々なユーザーの言語習慣に対応できます。
  • CoT溶液の合成このアプローチでは、段階的な推論ジェネレーターを通して、サンプルに中間的な導出ステップを追加します。トレーニング中に、モデルは質問からSQLへの変換と各推論ステップのロジックを学習し、推論の精度と信頼性を向上させ、ユーザーに透明な推論プロセスを提示し、信頼性を高めます。
  • 大規模データ合成とトレーニングOmniSQLは、独自のデータ合成フレームワークに基づき、大規模かつ高品質なトレーニングデータセット「SynSQL-2.5M」を作成しました。このデータセットには250万を超えるサンプルが含まれており、16,000以上の異なるドメインのデータベースを網羅しています。このような大規模で多様なデータセットを用いてトレーニングを行うことで、OmniSQLは様々なドメインやスタイルにおける自然言語表現とSQLクエリのマッピング関係を学習し、より優れた汎化能力と適応性を発揮します。

OmniSQLプロジェクトのアドレス

OmniSQLアプリケーションのシナリオ

  • エンタープライズデータ分析OmniSQLは、技術的な知識を持たないユーザーでも、自然言語によるクエリ機能を通じて、データベースから必要な情報を簡単に取得できるようにします。
  • 教育SQLの指導において、OmniSQLの連鎖思考(CoT)ソリューションは、初心者が自然言語による質問からSQLクエリへの変換プロセスをより深く理解するのに役立ちます。講師はOmniSQLを使用してクエリ例を生成することで、学生が実践的な演習を通してSQLの概念とスキルを習得できるようにすることができます。
  • ドメイン間適応OmniSQLは、独自のデータ合成フレームワークに基づいて、ドメイン固有のデータセットを迅速に生成できます。医療分野では、医学研究を支援するEHRSQLデータセットを生成でき、科学研究分野では、科学データ分析を支援するScienceBenchmarkデータセットを生成できます。