用語集

微調整

Ultralytics YOLO のような機械学習モデルを特定のタスクのために微調整します。方法、アプリケーション、ベストプラクティスをここで学びましょう!

Ultralytics HUB で
を使ってYOLO モデルをシンプルにトレーニングする。

さらに詳しく

ファインチューニングは、機械学習における重要なテクニックであり、事前に訓練されたモデルを新しい特定のデータセット上で改良し、特定のタスクに対するパフォーマンスを向上させる。ファインチューニングは、時間がかかり膨大なデータを必要とするモデルをゼロから学習する代わりに、大規模で一般的なデータセット上でモデルが既に学習した知識を活用する。このアプローチは、新しいタスクが、モデルが元々トレーニングされたタスクに関連している場合に特に効果的であり、より少ないデータで、より速い収束、より優れた汎化、精度の向上を可能にする。

微調整とは何か?

ファインチューニングでは、「事前学習済みモデル」と呼ばれる、広範なデータセットですでに学習されたモデルを、タスクに特化した小規模なデータセットでさらに学習させる。このプロセスでは通常、事前学習済みモデルの一部またはすべてのレイヤーを凍結解除し、学習プロセスを継続する。その目的は、事前に訓練されたモデルの学習特徴を新しいデータセットのニュアンスに適応させることであり、それによって目的のタスクに対する性能を向上させることである。これは、コンピュータ・ビジョンや 自然言語処理(NLP)のような分野で特に有益であり、事前に訓練されたモデルは、より具体的なアプリケーションに効果的に移行できる一般的な特徴を捉えている。

なぜ微調整が重要なのか?

微調整は、機械学習プロジェクトにおいていくつかの重要な利点をもたらす。第一に、学習時間と計算資源を大幅に削減できる。モデルはすでに学習された特徴からスタートするため、ゼロから学習するよりもはるかに速く収束する。これは、畳み込みニューラルネットワーク(CNN)やトランスフォーマーのような複雑なモデルを扱う場合に特に重要である。第二に、微調整はしばしばモデルの精度を高めることにつながる。事前に訓練されたモデルは、大規模なデータセットからロバストで汎化可能な特徴を学習しており、特定のタスクに適応させると、タスク固有のデータのみで訓練されたモデルよりも優れた性能を発揮する。第三に、膨大なデータセットの必要性が軽減される。ファインチューニングにより、限られたデータでも効果的な学習が可能になる。なぜなら、事前に学習されたモデルは、より広範なデータ分布から本質的なパターンをすでに捉えているからである。これは、医療画像解析のように、データ収集が困難であったり高価であったりする領域で特に有用である。

微調整の仕組み

微調整のプロセスは通常、適切な事前訓練済みモデルを選択することから始まる。例えば、物体検出のようなコンピュータビジョンのタスクでは、ImageNetで事前に訓練されたモデルが一般的に使用される。次のステップでは、新しいタスクのクラス数に合わせて、事前に訓練されたモデルの最終分類レイヤーを置き換える。その後、新しいデータセットでモデルを訓練する。多くの場合、実務家は、一般的な特徴を捉える事前学習済みモデルの初期レイヤーを凍結し、タスク固有の特徴を学習する後期レイヤーのみを訓練するなどのテクニックを採用する。あるいは、すべての層の凍結を解除し、すべての重みを微妙に調整するために非常に小さな学習率で訓練することもできる。レイヤーの凍結やハイパーパラメータのチューニングなど、さまざまな戦略を試すことは、手元の特定のタスクに最適なパフォーマンスを達成するために不可欠である。Ultralytics HUBのようなプラットフォームは、データセットを管理し、Ultralytics YOLO モデルを効率的に訓練するためのツールを提供し、このプロセスを簡素化する。

微調整の応用

ファインチューニングは、さまざまなAI領域で広く応用されている。いくつか例を挙げよう:

  • 医療画像: ヘルスケアにおけるAIでは、X線やMRIスキャンなどの医療画像から特定の疾患を検出するために、事前に訓練された画像分類モデルを微調整することができる。例えば、一般的な画像で事前に訓練されたモデルを、脳のMRIスキャンで腫瘍を正確に識別するために適応させることができ、診断や治療計画を大幅に支援することができる。このアプリケーションは、一般的な画像の特徴を医療画像解析の特定のコンテキストに転送するための微調整の能力を活用しています。
  • カスタム・オブジェクト検出: 農業AIのような用途では、Ultralytics YOLO 、大規模なデータセットで事前に訓練されたモデルを微調整して、農地内のさまざまな種類の作物、害虫、病気などの特定のオブジェクトを検出することができる。これにより、作物の健康状態や収穫量を正確かつ自動的に監視することが可能になり、農業の効率が向上する。例えば Ultralytics YOLOv8例えば、農家は、特定の農業の課題を検出し、管理するために、特定の農場の画像でカスタムモデルを訓練することができます。

微調整 vs トランスファー学習

しばしば同じ意味で使われるが、微調整は、より広い概念である転移学習の 中の特定のテクニックである。転移学習は、ある問題を解決することで得た知識を活用し、異なるが関連する問題に適用するという考え方を包含している。ファインチューニングは、事前に訓練されたモデルを、新しいデータセット上でその重みを調整する転移学習の手法である。他の転移学習手法では、新しいモデル・アーキテクチャで事前に訓練された特徴を使用したり、重みをさらに訓練することなく、事前に訓練されたモデルを特徴抽出器として使用したりすることがある。要するに、ファインチューニングは、転移学習に対するより実践的なアプローチであり、既存のモデルの継続的なトレーニングに重点を置いている。

微調整のためのツールとプラットフォーム

微調整プロセスを容易にするツールやプラットフォームがいくつかある。ディープラーニングフレームワーク PyTorchTensorFlowなどのディープ・ラーニング・フレームワークは、事前に訓練されたモデルをロードし、レイヤーを修正し、微調整を行うための広範なライブラリーと機能を提供している。Ultralytics HUBは、Ultralytics YOLO モデルのトレーニングと微調整のために特別に設計された合理化された環境を提供します。データセットの管理、モデルのトレーニング、デプロイメントのための機能でワークフローを簡素化し、コーディングの専門知識が乏しいユーザーでも微調整にアクセスできるようにしている。Google Colabや Amazon SageMakerのようなクラウドプラットフォームは、特に大規模なモデルやデータセットに対して、ファインチューニングタスクを効率的に実行するために必要なGPUなどの計算リソースを提供する。

結論として、ファインチューニングは現代の機械学習において不可欠な手法であり、強力な事前学習済みモデルを特定のタスクに効率的に適応させることができる。学習した特徴を活用し、新しいデータセットにモデルを調整することで、ファインチューニングは時間、リソース、データを節約すると同時に、しばしば優れたモデル性能を達成する。その幅広い適用可能性により、実用的なAI開発の基礎となっている。

すべて読む