Fakerを超えて:LLMとPythonでコンテキストを考慮した合成データをスケーリングする

AI tutorial - IT technology blog
AI tutorial - IT technology blog

午前2時の悪夢:本番データがステージング環境に漏洩した時

PagerDuty의警告音が鳴り響いたのは、午前2時14分のことでした。ジュニア開発者がクエリのタイムアウトをデバッグするために、誤って本番環境から50GBのデータベースダンプをステージング環境にプッシュしてしまったのです。

数分以内に、ロギングシステムがELKスタックに流れ込む平文の社会保障番号とクレジットカードのハッシュを検知しました。私たちはその後の6時間を、ログの削除とキーのローテーションに費やしました。さらに重要なことに、なぜ15,000件もの機密顧客レコードが暗号化されていないテスト用バケットに置かれていたのかをCTOに説明しなければなりませんでした。

これは現代のソフトウェア開発における「隠れた代償」です。機能をテストするためには現実的なデータが必要ですが、実際のユーザー情報を使用することはコンプライアンス上の地雷原を歩くようなものです。Fakerのような標準的なライブラリは、ランダムな名前を生成するのには適しています。しかし、コンテキスト(文脈)を考慮したデータが必要な場合には力不足です。例えば、医学的に筋の通った既往歴や、特定の不正パターンに従う一連の金融取引が必要な場合、Fakerでは対応できません。ここで、大規模言語モデル(LLM)とPythonがその状況を一変させます。

従来のデータマスキングが現代のアプリで通用しない理由

単純なマスキングでは、期待するほど保護されません。「山田太郎」を「佐藤花子」に置き換えても、レコードの残りの部分を維持すれば、基礎となるPII(個人識別情報)のパターンが残ってしまうことがよくあります。さらに、従来のスクリプトには意味論的なインテリジェンスが欠けています。ヘルスケアアプリをテストしていると想像してください。データ生成器が5歳の患者に「慢性老年性関節炎」という診断を下した場合、ビジネスロジックのテストは失敗します。さらに悪いことに、重大なバグを見逃してしまう偽陽性の結果を招く可能性もあります。

LLMは関係性を理解しています。「サンフランシスコ」の「シニアソフトウェアエンジニア」は、「デモイン」の「バリスタ」とは異なる給与プロファイルを持つべきであることを知っています。これらのモデルを Python でオーケストレーションすることで、何千ものユニークで有効なレコードを生成できます。これらは本番データのように見え、感じられますが、法的なリスクはゼロです。

アーキテクチャ:Pydantic、OpenAI、そしてバッチ処理

スケーラブルな合成データエンジンを構築するには、3つの主要なコンポーネントが必要です。

私は昨年、フィンテックのクライアントのためにこのフレームワークを実装しました。彼らのステージングデータベース全体を合成データに置き換えたのです。この移行により、コンプライアンス監査の対象範囲が90%削減され、複雑なデータスクラビング(洗浄)スクリプトの必要性がなくなりました。

ハンズオン:合成データエンジンの構築

まず、環境をセットアップしましょう。instructorライブラリを使用します。これはOpenAIのSDKをラップした優れたライブラリで、モデルがPydanticモデルに適合する有効なJSONを返すことを保証します。

pip install openai instructor pydantic

ステップ1:データモデルの定義

ネストされたトランザクションを持つ「ユーザープロファイル」のモデルを作成します。このようなリレーショナルな詳細は、従来のランダム生成器が通常破綻する部分です。

from pydantic import BaseModel, Field
from typing import List
import instructor
from openai import OpenAI

class Transaction(BaseModel):
    amount: float = Field(..., gt=0)
    merchant: str
    category: str = Field(description="例:食費、技術、旅行")
    is_suspicious: bool

class UserProfile(BaseModel):
    full_name: str
    job_title: str
    email: str
    bio: str = Field(description="現実的なプロフィールの自己紹介文")
    recent_transactions: List[Transaction]

ステップ2:生成ロジック

次に、LLMを呼び出す関数を作成します。ここではgpt-4o-miniモデルを使用します。この種の構造化されたタスクにおいて、高速かつ非常に安価だからです。

# パッチを適用したクライアントの初期化
client = instructor.from_openai(OpenAI(api_key="your_api_key"))

def generate_synthetic_user(industry: str):
    return client.chat.completions.create(
        model="gpt-4o-mini",
        response_model=UserProfile,
        messages=[
            {"role": "system", "content": "あなたは高精度な合成データを生成するデータアーキテクトです。"},
            {"role": "user", "content": f"{industry}セクター向けの現実的なユーザープロファイルを、3件のトランザクションと共に生成してください。"}
        ]
    )

ステップ3:並行処理によるスケーリング

1つのレコードを生成するのは簡単ですが、実際には数千件必要になるでしょう。これらを逐次実行するのは時間の無駄です。asyncioを使用して、APIのレート制限内に収めつつ、複数のリクエストを同時に実行します。

import asyncio

async def batch_generate(count: int, industry: str):
    # タスクのリストを作成
    tasks = [asyncio.to_thread(generate_synthetic_user, industry) for _ in range(count)]
    # 並行して実行
    results = await asyncio.gather(*tasks)
    return results

if __name__ == "__main__":
    users = asyncio.run(batch_generate(5, "Eコマース"))
    for user in users:
        print(f"{user.full_name} | {user.job_title}")

参照整合性の維持

大きな悩みの種の一つは、テーブル間でIDの一貫性を保つことです。Userを生成し、その後にOrderを生成する場合、user_idが正しくリンクしていなければなりません。これを解決するために、2パス戦略を推奨します。

  1. まず、ユーザーや製品などの「プライマリ」エンティティを生成します。これらをローカルのJSONファイルやSQLiteデータベースに保存します。
  2. 注文などの「セカンダリ」エンティティを生成する際に、既存のIDからランダムに選択してLLMのプロンプトに渡します。

プロンプトは次のようになります:「以下のユーザーIDのいずれかに対して注文を作成してください:[USR-99, USR-102]。アイテムはユーザーの過去の購入習慣に合わせてください。」

コストとパフォーマンスの最適化

LLMのトークンは無料ではありませんが、データ漏洩の罰金よりは安上がりです。gpt-4o-miniを使用すると、1,000件の複雑なユーザープロファイルの生成コストは約0.05ドルから0.10ドルです。数百万行が必要な場合は、ハイブリッドアプローチをとります。LLMを使用して500件の高品質な「シード」レコードを生成し、その後、標準的なPythonロジックを使用してそれらのシードをわずかに変化させ、50,000件のバリエーションを作成します。これにより、膨大なAPI費用をかけずに、リアルなデータの「感触」を維持できます。

最後に

ちょっとしたバグ修正のために本番データを「借用」する習慣はやめるべきです。GDPRやCCPAの施行により、「だいたい同じ」レベルのセキュリティではリスクが高すぎます。Pythonの柔軟性とLLMの意味論的な力を組み合わせることで、本番環境のクローンよりも優れたテスト環境を構築できます。500件のアクティブなサブスクリプションを持つユーザーや、特殊文字を含む名前など、実際のデータにはまだ存在しないかもしれないエッジケースをプログラムで生成できるのです。

このセットアップには午後一回の時間があれば十分です。ステージング環境の漏洩が夕方のニュースにならないという安心感を得られるなら、その労力に見合う価値は十分にあります。LLMのトークン費用を抑えつつ、より安全な開発フローを構築しましょう。

Share: