EnsorFlow ディープラーニングフレームワーク Google はビッグデータとクラウドコンピューティングのリーダーであるだけでなく、機械学習とディープラーニングの分野でも豊富な経験を持っています。2015 年末には、ディープラーニング フレームワーク TensorFlow を社内向けにオープンソース化しました。 Caffe、Theano、Torch、MXNet などのフレームワークと比較して、TensorFlow は Github 上でフォークとスターの数が最も多く、グラフ分類、オーディオ処理、推奨システム、自然言語処理などのシナリオで幅広く応用されています。最近人気のKerasフレームワークは、デフォルトで最下層にTensorFlowを使用しています。有名なスタンフォード大学のCS231nコースでは、授業や課題のプログラミング言語としてTensorFlowを使用しています。国内外で多くのTensorFlowの本がすでに準備中または販売中です。AlphaGoの開発チームDeepmindも、ニューラルネットワークアプリケーションをTensorFlowに移行する予定です。これらすべてが、業界でのTensorFlowの人気を裏付けています。 TensorFlow は Github でソースコードを公開しただけでなく、論文「TensorFlow: 異機種分散システムでの大規模機械学習」でシステムフレームワークの設計と実装も紹介しました。テストされた 200 ノード規模のトレーニング クラスターは、他の分散型ディープラーニング フレームワークの追随を許しません。 Google はまた、論文「レコメンデーション システムのためのワイド & ディープ ラーニング」と「YouTube 動画レコメンデーション システム」で、Google Play App Store と YouTube 動画レコメンデーションのアルゴリズム モデルを紹介し、TensorFlow に基づくコード例を提供しました。TensorFlow を使用すれば、ImageNet や Kaggle のコンペティションで、誰でも最先端に近い優れた結果を得ることができます。 TensorFlow 入門から応用まで TensorFlow の人気により、ディープラーニングの敷居がどんどん下がったと言っても過言ではありません。Python と機械学習の基本的な知識さえあれば、ニューラル ネットワーク モデルを使い始めるのは非常に簡単になります。 TensorFlow は、Python と C++ の 2 つのプログラミング言語をサポートしています。最も複雑な多層ニューラル ネットワーク モデルでも、Python で実装できます。ビジネスで他のプログラミング言語を使用している場合でも心配はいりません。また、クロス言語 gRPC または HTTP サービスを使用して、TensorFlow でトレーニングされたスマート モデルにアクセスすることもできます。 では、Python を使用して TensorFlow アプリケーションを作成するにはどうすればよいでしょうか? 始めるのはどれくらい難しいのでしょうか? 以下では、文字列を出力し、簡単な計算を実行する Hello World アプリケーションを作成します。
このシンプルなコードから、TensorFlow が非常に使いやすいことがわかります。追加のサービスを開始せずに、Python 標準ライブラリを介してインポートできます。初めて TensorFlow に触れると、戸惑うかもしれません。このロジックは Python でも実装できますが、なぜ tf.constant() と tf.Session() を使用するのでしょうか。実際、TensorFlow は Graph と Session を使用して実行中のモデルとトレーニングを定義します。これは、この記事の後半で紹介する複雑なモデルと分散トレーニングに非常に役立ちます。 前回の Hello world アプリケーションにはトレーニング済みのモデルがありませんでした。次に、ロジスティック回帰の問題とモデルを紹介します。 numpy を使用して線形関係データのセットを構築し、TensorFlow によって実装された確率的勾配アルゴリズムを使用して、十分な時間のトレーニング後に関数の傾きと切片を自動的に解きます。
上記のコードは、tensorflow_examples プロジェクトにあります。トレーニング後、出力の傾き w は約 2、切片 b は約 10 であることがわかります。これは、構築したデータ間の相関と一致しています。TensorFlow コードには最小二乗法やその他のアルゴリズムは実装されておらず、コード ロジックを制御する if-else もありません。これは完全にデータ駆動型であり、勾配降下アルゴリズムに従って Loss 値を動的に調整することで学習されます。このように、他のデータセットに変更したり、画像分類などの他の分野の問題に変更したりしても、コードを修正することなく機械が自動的に学習することができます。これもニューラルネットワークとTensorFlowの力です。
以前のモデルには、w と b の 2 つの変数しかありませんでした。データが非線形関係にある場合、良い結果を得ることは困難です。そのため、TensorFlow が設計に重点を置いているディープラーニング モデルでもあるディープ ニューラル ネットワークを使用することをお勧めします。 Google が 2014 年に Inception モデルで ImageNet グローバル コンペティションに勝利したことは知られています。そのコードは TensorFlow に基づいて実装されています。以下は、より複雑なモデル定義コードです。
TensorFlow がパッケージ化した完全接続ネットワーク、畳み込みニューラル ネットワーク、RNN、LSTM を使用すると、さまざまなネットワーク モデルを組み合わせることができ、Inception などの多層ニューラル ネットワークを実装するのはレゴを組み立てるのと同じくらい簡単です。ただし、最適化アルゴリズムの選択、TFRecord の生成、モデル ファイルのエクスポート、分散トレーニングのサポートには多くの詳細があります。次に、TensorFlow に関連するすべてのコアな使用スキルを 1 つの記事で紹介します。 TensorFlow コアの使用に関するヒント TensorFlowのさまざまな使い方を紹介するために、deep_recommend_system(https://github.com/tobegit3hub/deep_recommend_system)を使用します。 このオープンソース プロジェクトは、TFRecords、QueueRunner、Checkpoint、TensorBoard、推論、GPU サポート、分散トレーニング、多層ニューラル ネットワーク モデルなどの機能を実装しています。また、Wide や Deep などのモデルを実装するように簡単に拡張でき、直接ダウンロードして実際のプロジェクト開発に使用できます。 1. トレーニングデータを準備する 一般的に、TensorFlow アプリケーション コードには、Graph の定義と Session の操作が含まれます。コードの量が多くない場合は、cancer_classifier.py ファイルなどの 1 つのファイルにカプセル化できます。トレーニングの前に、サンプル データとテスト データを準備する必要があります。通常、データ ファイルはスペースまたはカンマで区切られた CSV ファイルですが、TensorFlow では、QueuRunner と Coordinator によるマルチスレッド データ読み取りをサポートするバイナリ TFRecords 形式の使用を推奨しています。バッチ サイズとエポック パラメータを使用して、トレーニング中の単一バッチのサイズと、サンプル ファイルの反復トレーニング ラウンドの数を制御できます。 CSV ファイルを直接読み込む場合、次に読み込むデータのポインタをコード内に記録する必要があり、すべてのサンプルをメモリにロードできない場合には非常に不便です。 データ ディレクトリには、CSV と TFRecords 形式変換ツール generate_csv_tfrecords.py が用意されています。このスクリプトを参照することで、任意の形式の CSV ファイルを解析し、TensorFlow でサポートされている TFRecords 形式に変換できます。ビッグデータでもスモールデータでも、シンプルなスクリプトツールを使って TensorFlow に直接接続できます。また、このプロジェクトでは、API を呼び出して TFRecords ファイルの内容を直接読み取る print_csv_tfrecords.py スクリプトも提供しています。
2. コマンドライン引数を受け入れる TFRecords を使用すると、ニューラル ネットワーク モデルをトレーニングするためのコードを記述できます。ただし、ご存知のとおり、ディープラーニングには調整が必要なハイパーパラメータが多すぎます。最適化アルゴリズム、モデル レイヤー、さまざまなモデルを常に調整する必要があります。このとき、コマンド ライン パラメータを使用すると非常に便利です。 TensorFlow は最下層で python-gflags プロジェクトを使用し、それが tf.app.flags インターフェースにカプセル化されます。使い方は非常にシンプルで直感的です。実際のプロジェクトでは、特に後述する Cloud Machine Learning サービスでは、コマンドライン パラメータが事前に定義されているのが一般的で、パラメータによるハイパーパラメータのチューニングが簡素化されます。
3. ニューラルネットワークモデルを定義する データとパラメータを準備した後、最も重要なことはネットワーク モデルを定義することです。モデル パラメータの定義は、複数の変数を作成するだけの簡単なものから、tf.variable_scope() および tf.get_variables() インターフェイスを使用するなどの複雑なものまであります。各変数に一意の名前が付けられ、隠しノードとネットワーク層の数を簡単に変更できるようにするには、変数を定義するときに CPU のバインドに特に注意しながら、プロジェクト内のコードを参照することをお勧めします。TensorFlow はデフォルトで GPU を使用するため、パラメータの更新が遅くなる可能性があります。
上記のコードは実稼働環境でも非常に一般的であり、トレーニング、推論の実装、モデルの精度と AUC の検証に使用されます。プロジェクトでは、このコードに基づいて、Google Play App Storeのレコメンデーション業務で広く使用されているワイド&ディープモデルも実装しました。これは、従来のロジスティック回帰モデルとディープラーニングニューラルネットワークモデルを有機的に組み合わせたものであり、一般的なレコメンデーションシステムにも適しています。 4. 異なる最適化アルゴリズムを使用する ネットワーク モデルを定義した後、モデル パラメータを最適化するためにどのオプティマイザーを使用するかを考える必要があります。Sgd、Rmsprop、Adagrad、Ftrl のどれを選択すればよいでしょうか。さまざまなシナリオやデータ セットに対して決まった答えはありません。最善の方法は実践することです。上記で定義したコマンド ライン パラメータを使用すると、さまざまな最適化アルゴリズムを使用してモデルを簡単にトレーニングできます。
実際の運用では、最適化アルゴリズムによってトレーニング結果とトレーニング速度に大きな違いがあります。ネットワークパラメータの過剰最適化は、他の最適化アルゴリズムを使用する場合ほど効果的ではない可能性があります。したがって、適切な最適化アルゴリズムを選択することも、ハイパーパラメータのチューニングにおいて非常に重要なステップです。このロジックを TensorFlow コードに追加することで、対応する機能も適切に実装できます。 5. オンライン学習と継続学習 多くの機械学習ベンダーは、自社製品がオンライン学習をサポートしていると主張しています。実際、これはオンライン データによるモデルの継続的な最適化をサポートするという TensorFlow の基本機能にすぎません。 TensorFlow は、tf.train.Saver() を通じてモデルを保存し、モデル パラメータを復元できます。Python を使用してモデル ファイルをロードした後、オンラインで要求されたデータを継続的に受け入れ、モデル パラメータを更新し、次の最適化またはオンライン サービスのために Saver を通じてチェックポイントとして保存できます。
継続的なトレーニングとは、トレーニングが中断された場合でも、前回のトレーニングの結果に基づいてモデルを最適化できることを意味します。TensorFlow では、これも Saver とチェックポイント ファイルを通じて実現されます。 deep_recommend_system プロジェクトでは、デフォルトで前回のトレーニングからモデルの最適化を継続できます。また、コマンドラインで train_from_scratch を指定することもできます。トレーニングプロセスが中断されることを心配する必要がないだけでなく、トレーニングと推論をしながらオンラインサービスを提供することもできます。 6. TensorFlowを使用してパラメータを最適化する TensorFlow には、強力なグラフィカル ツールである TensorBoard も統合されています。通常は、必要なトレーニング インジケーターをコードに追加するだけで、TensorBoard がこれらのパラメーターに基づいてグラフを自動的に描画するため、モデル トレーニングの状態を視覚的に把握できます。
7. 分散TensorFlowアプリケーション 最後に、TensorFlow の強力な分散コンピューティング機能を紹介する必要があります。Caffe などの従来のコンピューティング フレームワークは、分散トレーニングをネイティブにサポートしておらず、データの量が膨大な場合にマシンを追加してスケールアウトできないことがよくあります。 TensorFlow は、Google のさまざまな事業から PB レベルのデータを運びます。分散コンピューティングのニーズは設計当初から考慮されており、ニューラル ネットワーク モデルの分散コンピューティングは、gRPC や Protobuf などの高性能ライブラリを通じて実装されています。 分散 TensorFlow アプリケーションを実装するのは難しくありません。グラフを構築するためのコードは、スタンドアロン バージョンと同じです。分散 cancer_classifier.py の例を実装しました。次のコマンドを使用して、マルチ ps マルチワーカー トレーニング クラスターを起動できます。
コードを詳しく読む前に、分散 TensorFlow における ps、ワーカー、グラフ内、グラフ間、同期トレーニング、非同期トレーニングの概念を理解する必要があります。まず、ps はトレーニング クラスター全体のパラメータ サーバーであり、モデルの変数を保存します。ワーカーはモデルの勾配を計算するノードであり、得られた勾配ベクトルは ps に渡され、モデルを更新します。イングラフはビトウィーングラフに対応しますが、どちらも同期トレーニングと非同期トレーニングを実現できます。イングラフとは、クラスター全体を 1 つのクライアントで構築し、クライアントがグラフをクラスターに送信することを意味します。他のワーカーは、勾配計算タスクの処理のみを担当します。ビトウィーングラフとは、クラスター内の複数のワーカーが複数のグラフを作成できることを意味しますが、ワーカーは同じコードを実行するため、構築されたグラフも同じであり、パラメーターは同じ PS に保存され、同じモデルのトレーニングが保証されます。このようにして、複数のワーカーがグラフを構築し、トレーニング データを読み取ることができるため、ビッグ データのシナリオに適しています。同期トレーニングと非同期トレーニングの違いは、同期トレーニングでは勾配が更新されるたびにブロックしてすべてのワーカーの結果を待つ必要があるのに対し、非同期トレーニングではブロックされず、トレーニング効率が高くなることです。非同期トレーニングは、通常、ビッグデータや分散シナリオで使用されます。 8. クラウド機械学習 TensorFlow に関するすべてのコンテンツを紹介しました。注意深いネットユーザーなら、TensorFlow は強力ですが、本質的にはライブラリであることに気付いたかもしれません。TensorFlow アプリケーション コードを記述するだけでなく、ユーザーは物理マシンでサービスを起動し、トレーニング データとモデル ファイルのディレクトリを手動で指定する必要があります。メンテナンス コストは比較的高く、マシン間で共有することはできません。 ビッグデータ処理とリソース スケジューリング業界を見ると、Hadoop エコシステムが業界標準になっています。データは MapReduce または Spark インターフェイスを介して処理されます。ユーザーは API を介してタスクを送信し、Yarn が統合されたリソース割り当てとスケジューリングを実行します。これにより、分散コンピューティングが可能になるだけでなく、リソース共有と統合されたスケジューリング プラットフォームを通じてサーバーの使用率が大幅に向上します。残念ながら、TensorFlow はディープラーニングフレームワークとして定義されており、クラスターリソース管理などの機能は含まれていません。しかし、TensorFlow をオープンソース化した後、Google はすぐに Google Cloud ML サービスを発表しました。私たちはアルファ版から Cloud ML の早期ユーザーであり、クラウドでのディープラーニングトレーニングの利便性を深く体験してきました。 Google Cloud ML サービスを使用すると、TensorFlow アプリケーション コードを直接クラウドに送信して実行したり、トレーニング済みのモデルをクラウドに直接デプロイして API から直接アクセスしたりすることができます。TensorFlow の優れた設計のおかげで、Kubernetes と TensorFlow サービングに基づく Cloud Machine Learning サービスを実装できました。アーキテクチャ設計と使用インターフェースは、Google Cloud ML のものと似ています。 TensorFlow は優れたディープラーニング フレームワークです。個人の開発者、研究者、企業にとって投資する価値のあるテクノロジーです。クラウド マシン ラーニングは、環境の初期化、トレーニング タスクの管理、ニューラル ネットワーク モデルのオンライン サービスにおけるユーザーの管理とスケジュールの問題を解決できます。現在、Google Cloud ML はハイパーパラメータの自動チューニングをすでにサポートしており、パラメータチューニングは将来的には技術的な問題ではなく計算上の問題になるでしょう。一部の開発者が TensorFlow の代わりに MXNet などを使用するとしても、私たちはより多くのディープラーニング ユーザーやプラットフォーム開発者とコミュニケーションをとり、コミュニティの発展を促進していきたいと考えています。 要約する まとめると、この記事では主に TensorFlow ディープラーニング フレームワークの学習と応用について紹介しました。deep_recommend_system プロジェクトを通じて、TensorFlow を使用する際の次の 8 つのコア ポイントを紹介します。試用とフィードバックのためにソース コードをダウンロードすることもできます。 |
<<: 分散機械学習プラットフォームの比較: Spark、PMLS、TensorFlow、MXNet
>>: ディープラーニングを自分の仕事にどのように統合しますか?
リカレント ニューラル ネットワーク (RNN)、長期短期記憶 (LSTM)、これらの人気のニューラ...
IT Homeは12月12日、マイクロソフトが2021年にウェブベースの使いやすいビデオ編集ツール...
ジェネレーティブ AI スタートアップの需要は高く、テクノロジー大手からの CVC および VC 資...
海外メディアの報道によると、マイクロソフトは水曜日、小規模な人工知能スタートアップ企業であるボンサイ...
著者: 張傑【51CTO.comオリジナル記事】 AI チップはなぜ必要なのでしょうか?新しいイ...
大規模なモデルの「ブラックボックス」を解体するために、人類解釈可能性チームは、新しいモデルをトレーニ...
この記事はAI新メディアQuantum Bit(公開アカウントID:QbitAI)より許可を得て転載...
AR、VR、3Dプリント、シーン構築、映画制作など多くの分野において、衣服を着た人体の高品質な3Dモ...
昨年のAI開発者会議で、百度は「人工知能時代のAndroidシステム」であると主張する会話型人工知能...
AIの兵器化?大規模言語モデルの誕生以来、人々はその潜在的な影響について議論し続けています。しかし...
2019 年 10 月末時点で、HanLP と呼ばれる自然言語処理ライブラリが GitHub で ...