二値分類モデルとは、データが二つのカテゴリーのどちらに属するかを予測するAIモデルです。スパム判定や購買予測などの用途、ロジスティック回帰やSVMなどの代表的手法、正解率・適合率・再現率・F1スコアによる評価を解説します。基本ながら幅広い分野で使われる重要なモデルです。
二値分類モデル(Binary Classification)とは、与えられたデータが二つのカテゴリーのうちどちらに属するかを予測するAIモデルのことです。本記事では、基本の仕組み、代表的な手法、評価指標、活用分野を整理します。
たとえば、メールが「スパム」か「非スパム」か、顧客が「購入する」か「購入しない」か、といった予測を行います。このモデルは、AI技術のなかでも基本的ながら非常に重要な位置を占めており、さまざまな分野で活用されています。
二値分類モデルは、入力されたデータの特徴量をもとに、そのデータがどちらのカテゴリーに属するかの確率を計算します。この確率が一定の閾値(たとえば0.5)を超えれば一方のカテゴリー、下回ればもう一方のカテゴリーに分類されます。
二値分類モデルにはさまざまな種類があり、データの特性や目的に応じて使い分けられます。
| 手法 | 概要 |
|---|---|
| ロジスティック回帰 | 確率を出力する基本的な分類手法 |
| サポートベクターマシン(SVM) | クラスの境界を求めて分類する手法 |
| 決定木・ランダムフォレスト | 条件分岐を組み合わせて分類する手法 |
| ニューラルネットワーク | 多層のネットワークで複雑な境界を学習する手法 |
ロジスティック回帰、サポートベクターマシン(SVM)、決定木、ランダムフォレスト、ニューラルネットワークなどが代表的です。
二値分類モデルを構築する際には、まず適切な学習データを用意する必要があります。学習データは、各データがどのカテゴリーに属するかの正解ラベルを含んでいる必要があります。次に、用意した学習データを用いてモデルを学習させ、学習が終わったら、モデルの性能を評価するためにテストデータを使用します。このように正解ラベル付きのデータで学ぶ形式は教師あり学習に分類されます。
モデルの評価には、さまざまな指標が用いられます。
| 指標 | 意味 |
|---|---|
| 正解率(Accuracy) | 全体のデータのうち正しく分類できた割合 |
| 適合率(Precision) | positive と予測したうち、実際に positive だった割合 |
| 再現率(Recall) | 実際に positive なうち、positive と予測できた割合 |
| F1スコア | 適合率と再現率の調和平均 |
正解率は、全体のデータのなかで正しく分類できたデータの割合を示します。適合率は、モデルが positive と予測したデータのうち、実際に positive であったデータの割合を示します。再現率は、実際に positive なデータのうち、モデルが positive と予測できたデータの割合を示します。F1スコアは、適合率と再現率の調和平均で、モデルの総合的な性能を評価するために使用されます。
適合率の考え方は、適合率の項目で詳しく解説しています。全体の正解割合である正答率とあわせて確認すると理解が深まります。
二値分類モデルは、さまざまな分野で活用されています。医療分野では、患者のデータから病気の有無を予測したり、金融分野では、顧客の信用情報からローンの返済能力を予測したり、マーケティング分野では、顧客の購買履歴から商品を購入するかどうかを予測したりします。また、製造業では、製品の品質データから不良品を検出したり、セキュリティ分野では、ネットワークのトラフィックデータから不正アクセスを検知したりします。これらの事例からもわかるように、二値分類モデルは、さまざまな問題解決に役立つ強力なツールです。
実際は:偏ったデータでは、正解率が高くても役に立たない場合があります。適合率や再現率もあわせて確認しましょう。
実際は:二値分類は二つのカテゴリーを扱う特別な場合です。三つ以上のカテゴリーを扱うのは多クラス分類です。
実際は:閾値は目的に応じて調整できます。見落としを減らしたいか、誤検出を減らしたいかで適切な値は変わります。
A. データの量や性質、求める解釈性によって異なります。まずは基本的な手法で試し、必要に応じて比較検討するのが現実的です。
A. 目的によります。誤検出を避けたいなら適合率、見落としを避けたいなら再現率、両者のバランスならF1スコアが目安になります。
用語の一覧はAI用語辞典トップからたどれます。実際のツール比較はAIツール一覧をご覧ください。
本記事にはアフィリエイトリンクを含む場合があります。経由での申し込み時に当サイトが報酬を受け取ることがありますが、掲載内容・評価は報酬の有無に左右されません。記載情報は公開時点のものであり、最新情報は各公式サイトをご確認ください。
編集部の評価方法論を見る →