【0から学ぶAI】第236回:クラウド環境でのモデル実行 〜AWSやGCPを使ったモデルの運用方法を解説

目次

前回のおさらいと今回のテーマ

こんにちは!前回は、APIの作成と公開について解説し、Flaskを使って学習済みモデルをAPI化し、Herokuにデプロイして公開する方法を紹介しました。APIを使うことで、他のアプリケーションやシステムからモデルの予測機能を利用できるようになりましたね。

今回は、クラウド環境でのモデル実行について、AWSやGCPを活用したモデルの運用方法を解説します。クラウドサービスを利用することで、モデルをスケーラブルに運用し、大量のリクエストに対応したり、計算リソースを効率的に管理することが可能になります。この記事では、AWSとGCPを例に、クラウド環境でのモデルデプロイから運用までの手順を紹介します。

クラウド環境でモデルを運用するメリット

クラウド環境を活用することで、以下のメリットがあります。

  1. スケーラビリティ: トラフィックに応じてリソースを自動的に調整し、大量のリクエストにも対応できます。
  2. コスト効率: 実際に利用した分だけ料金が発生するため、無駄なコストを削減できます。
  3. 高可用性: データセンターの冗長構成により、障害が発生しても他のサーバーに自動で切り替えることでサービスの継続性を確保します。
  4. セキュリティ: セキュリティ管理がしっかりしているクラウドサービスを使うことで、データやサービスを保護することができます。

AWS(Amazon Web Services)を使ったモデルデプロイ

まずは、AWSを使ってモデルを運用する方法を説明します。AWSでは、Amazon SageMakerというサービスがあり、モデルのデプロイや訓練、チューニングまでを一貫してサポートしてくれます。また、AWS Lambdaを使ってサーバーレスでモデルをAPIとしてデプロイすることも可能です。

1. Amazon SageMakerによるモデルデプロイ

Amazon SageMakerは、機械学習モデルの開発からデプロイまでを一貫してサポートするマネージドサービスです。以下の手順で、SageMakerを使ったモデルのデプロイ方法を解説します。

(1) AWSアカウントの作成と設定

まず、AWSアカウントを作成し、SageMakerを利用するためのアクセス権限を設定します。

  1. AWSにアクセスし、アカウントを作成します。
  2. AWSマネジメントコンソールで、IAM(Identity and Access Management)を使って新しいユーザーを作成し、SageMakerの管理権限を付与します。

(2) SageMaker Notebookインスタンスの作成

次に、SageMakerのNotebookインスタンスを作成して、モデルの訓練やデプロイを行います。

  1. AWSマネジメントコンソールで、SageMakerにアクセスします。
  2. Notebook instancesから新しいインスタンスを作成します。必要に応じて、インスタンスのタイプ(計算能力)やストレージサイズを選択します。
  3. Notebookインスタンスが起動したら、Jupyter Notebookの環境が利用可能になります。

(3) モデルの訓練とデプロイ

SageMakerでは、モデルの訓練からデプロイまでをJupyter Notebook上で行えます。以下のコードは、SageMakerで学習済みモデルをデプロイするシンプルな例です。

import sagemaker
from sagemaker.tensorflow import TensorFlow

# SageMakerのセッションとロールの設定
sagemaker_session = sagemaker.Session()
role = 'your-sagemaker-execution-role'

# モデルの学習(または既存のモデルを使用する)
tensorflow_estimator = TensorFlow(entry_point='train.py',  # トレーニングスクリプト
                                  role=role,
                                  instance_count=1,
                                  instance_type='ml.m5.large',
                                  framework_version='2.4.1')

# モデルのデプロイ
predictor = tensorflow_estimator.deploy(initial_instance_count=1,
                                        instance_type='ml.m5.large')
  • entry_point: トレーニングスクリプト(例:train.py)のパスを指定します。
  • instance_type: モデルの訓練に使用するインスタンスのタイプを指定します。
  • deploy: モデルをデプロイし、エンドポイント(API)を公開します。

これにより、SageMaker上にデプロイされたモデルがエンドポイントとして公開され、他のアプリケーションからAPI経由でアクセス可能になります。

2. AWS Lambdaによるサーバーレスデプロイ

Amazon Lambdaを使えば、サーバーを管理することなくAPIをデプロイできます。Lambdaはイベント駆動型のサーバーレスサービスで、コードの実行にのみ料金が発生します。

(1) AWS Lambdaのセットアップ

  1. AWSマネジメントコンソールで、Lambdaにアクセスし、新しい関数を作成します。
  2. ランタイムとしてPythonを選択し、関数名を指定して作成します。

(2) Lambda関数のコード例

以下は、Lambda関数内でTensorFlowモデルを使って予測を行うシンプルな例です。

import json
import tensorflow as tf
import numpy as np

def lambda_handler(event, context):
    # モデルの読み込み
    model = tf.keras.models.load_model('/mnt/data/model.h5')

    # リクエストから入力データを取得
    input_data = np.array(event['input']).reshape(1, -1)

    # 予測
    prediction = model.predict(input_data)
    predicted_class = int(np.argmax(prediction))

    # 結果を返す
    return {
        'statusCode': 200,
        'body': json.dumps({'prediction': predicted_class})
    }

このコードをLambdaにデプロイし、API Gatewayと連携することで、APIとして公開できます。

GCP(Google Cloud Platform)を使ったモデルデプロイ

Google Cloud Platform(GCP)でも、機械学習モデルのデプロイに特化したサービスが提供されています。ここでは、Google AI Platformを使ったモデルデプロイ方法を解説します。

1. Google Cloud AI Platformの設定

GCPでは、Google Cloud AI Platformを使ってモデルのデプロイや運用が可能です。

(1) GCPアカウントの作成と設定

  1. GCPのアカウントを作成し、Google Cloud Consoleにアクセスします。
  2. プロジェクトを作成し、AI Platformを有効にします。

(2) AI Platformの使用準備

次に、Google Cloud SDKをインストールし、ローカル環境からGCPにアクセスできるようにします。

gcloud init

(3) モデルのデプロイ

以下のコマンドでモデルをGCPにアップロードし、AI Platformでデプロイします。

# モデルファイルをGoogle Cloud Storageにアップロード
gsutil cp model.h5 gs://your-bucket-name/models/model.h5

# AI Platformにモデルをデプロイ
gcloud ai-platform models create my_model --regions us-central1

gcloud ai-platform versions create v1 \
    --model my_model \
    --origin gs://your-bucket-name/models/ \
    --runtime-version 2.5 \
    --framework tensorflow \
    --python-version 3.7
  • gsutil: Google Cloud Storageにファイルをアップロードするツールです。
  • gcloud ai-platform: AI Platformにモデルをデプロイするコマンドです。

デプロイが完了すると、GCP上でAPIエンドポイントが作成され、HTTPリクエストを通じてモデルの予測機能が利用可能になります。

クラウド環境でのベストプラクティス

  1. スケーラビリティの確保: トラフィックが増加した際に自動でリソースを拡張するオートスケーリングの設定を行います。
  2. セキュリティの強化: APIの認証機能や、アクセス制限を設けることで、外部

からの不正なアクセスを防ぎます。

  1. 監視とログの活用: クラウドの監視ツール(AWS CloudWatch、GCP Monitoringなど)を活用し、APIのパフォーマンスやエラーをリアルタイムで確認します。
  2. コスト管理: クラウドのリソースは従量課金制のため、使用量に応じたコスト管理を行い、無駄な支出を防ぎます。

まとめ

今回は、AWSやGCPを活用したクラウド環境でのモデル実行について解説しました。クラウドサービスを使うことで、スケーラブルで高可用性なモデルの運用が可能になります。これを基礎として、さらなる機械学習モデルの展開や高度なシステム開発に挑戦してください。

次回予告

次回は、Dockerによる環境構築として、コンテナ技術を用いた環境の再現性確保方法について解説します。開発環境の統一やデプロイの効率化に役立つDockerを学びましょう!


注釈

  • Amazon SageMaker: AWSが提供する機械学習モデルの開発、訓練、デプロイを支援するサービス。
  • Google AI Platform: GCPの機械学習モデル運用サービスで、デプロイから運用、モニタリングまでを一元管理可能。
よかったらシェアしてね!
  • URLをコピーしました!
  • URLをコピーしました!

この記事を書いた人

株式会社PROMPTは生成AIに関する様々な情報を発信しています。
記事にしてほしいテーマや調べてほしいテーマがあればお問合せフォームからご連絡ください。
---
PROMPT Inc. provides a variety of information related to generative AI.
If there is a topic you would like us to write an article about or research, please contact us using the inquiry form.

コメント

コメントする

目次