Web ワーカーを使用してモデルをトレーニングする,Web ワーカーを使用してモデルをトレーニングする

このチュートリアルでは、 Web ワーカーを使用してリカレントニューラルネットワーク(RNN) をトレーニングして整数の加算を行うサンプル Web アプリケーションを検討します。サンプルアプリでは、加算演算子を明示的に定義していません。代わりに、サンプルの合計を使用して RNN をトレーニングします。

もちろん、これは 2 つの整数を加算する最も効率的な方法ではありません。ただし、このチュートリアルでは、Web ML における重要なテクニック、つまり UI ロジックを処理するメインスレッドをブロックせずに長時間実行される計算を実行する方法を示しています。

このチュートリアルのサンプルアプリケーションはオンラインで入手できるため、コードをダウンロードしたり、開発環境をセットアップしたりする必要はありません。コードをローカルで実行する場合は、「サンプルをローカルで実行する」のオプションの手順を実行します。開発環境をセットアップしたくない場合は、「サンプルを調べる」に進んでください。

サンプルコードはGitHubで入手できます。

(オプション) サンプルをローカルで実行する

前提条件

サンプルアプリをローカルで実行するには、開発環境に次のものをインストールする必要があります。

Node.js (ダウンロード)
糸（取り付け）

サンプルアプリをインストールして実行する

tfjs-examplesリポジトリのクローンを作成するか、ダウンロードします。
addition-rnn-webworkerディレクトリに移動します。
```
cd tfjs-examples/addition-rnn-webworker
```
依存関係をインストールします。
```
yarn
```
開発サーバーを起動します。
```
yarn run watch
```

例を見てみる

サンプルアプリを開きます。 (または、サンプルをローカルで実行している場合は、ブラウザでhttp://localhost:1234に移動します。)

TensorFlow.js: Addition RNNというタイトルのページが表示されるはずです。指示に従ってアプリを試してください。

Web フォームを使用すると、モデルのトレーニングに使用される次のようなパラメーターの一部を更新できます。

桁数: 追加される項の最大桁数。
トレーニングサイズ: 生成するトレーニングサンプルの数。
RNN タイプ: SimpleRNN 、 GRU 、またはLSTMのいずれか。
RNN Hidden Layer Size : 出力空間の次元数 (正の整数である必要があります)。
バッチサイズ: 勾配更新ごとのサンプル数。
Train Iterations : model.fit()を呼び出してモデルをトレーニングする回数
# of test Example : 生成するサンプル文字列の数 (たとえば、 27+41 )。

さまざまなパラメーターを使用してモデルをトレーニングしてみて、さまざまな数値セットの予測の精度を向上できるかどうかを確認してください。また、モデルの適合時間がさまざまなパラメーターによってどのように影響されるかにも注目してください。

コードを調べる

サンプルアプリでは、RNN をトレーニングするために構成できるパラメーターのいくつかを示します。また、Web ワーカーを使用してメインスレッドからモデルをトレーニングする方法も示します。 Web ワーカーは、計算量の多いトレーニングタスクをバックグラウンドスレッドで実行できるため、Web ML で重要です。これにより、メインスレッドでユーザーに影響を与える可能性のあるパフォーマンスの問題を回避できます。メインスレッドとワーカースレッドは、メッセージイベントを通じて相互に通信します。

Web ワーカーの詳細については、「Web ワーカー API 」および「Web ワーカーの使用」を参照してください。

サンプルアプリのメインモジュールは、 index.jsです。 index.jsスクリプトは、 worker.jsモジュールを実行するWeb ワーカーを作成します。

const worker =
    new Worker(new URL('./worker.js', import.meta.url), {type: 'module'});

index.js主に、フォームの送信を処理し、フォームデータを処理し、ワーカーにフォームデータを渡し、ワーカーがモデルをトレーニングして結果を返すのを待ち、結果をページに表示する単一の関数runAdditionRNNDemoで構成されています。。

フォームデータをワーカーに送信するために、スクリプトはワーカー上でpostMessageを呼び出します。

worker.postMessage({
  digits,
  trainingSize,
  rnnType,
  layers,
  hiddenSize,
  trainIterations,
  batchSize,
  numTestExamples
});

ワーカーはこのメッセージをリッスンし、データを準備してトレーニングを開始する関数にフォームデータを渡します。

self.addEventListener('message', async (e) => {
  const { digits, trainingSize, rnnType, layers, hiddenSize, trainIterations, batchSize, numTestExamples } = e.data;
  const demo = new AdditionRNNDemo(digits, trainingSize, rnnType, layers, hiddenSize);
  await demo.train(trainIterations, batchSize, numTestExamples);
})

トレーニング中に、ワーカーは 2 つの異なるメッセージタイプを送信できます。1つはisPredictがtrueに設定されています。

self.postMessage({
  isPredict: true,
  i, iterations, modelFitTime,
  lossValues, accuracyValues,
});

もう 1 つはisPredictがfalseに設定されています。

self.postMessage({
  isPredict: false,
  isCorrect, examples
});

UI スレッド ( index.js ) はメッセージイベントを処理するときに、 isPredictフラグをチェックしてワーカーから返されたデータの形式を決定します。 isPredictが true の場合、データは予測を表す必要があり、スクリプトはtfjs-visを使用してページを更新します。 isPredictが false の場合、スクリプトはデータが例を表していると想定してコードブロックを実行します。データを HTML でラップし、その HTML をページに挿入します。

次は何ですか

このチュートリアルでは、長時間実行されるトレーニングプロセスによる UI スレッドのブロックを回避するために Web ワーカーを使用する例を示しました。バックグラウンドスレッドで負荷の高い計算を実行する利点の詳細については、「Web ワーカーを使用してブラウザのメインスレッドから JavaScript を実行する」を参照してください。

TensorFlow.js モデルのトレーニングの詳細については、「モデルのトレーニング」を参照してください。