Skip to main content
QUICK REVIEW

[論文レビュー] Extending Isolation Forest for Anomaly Detection in Big Data via K-Means

Tahmid Rahman Laskar, Jimmy Xiangji Huang|arXiv (Cornell University)|Apr 27, 2021
Network Security and Intrusion Detection参考文献 89被引用数 4
ひとこと要約

本論文では、大規模な産業用ネットワークトラフィックデータにおける効果的でリアルタイムな異常検出を可能にする、孤立木(Isolation Forest)とK-Meansクラスタリングを組み合わせた新しい教師なし異常検出フレームワーク、IForest-KMeansを提案する。Apache Sparkを用いた分散処理により、大規模データにおける孤立木の学習不安定性を解消し、少ないハイパーパrameterで最先端のモデルと同等の性能を達成するとともに、ライブストリーミングデータへの展開を可能にした。

ABSTRACT

Industrial Information Technology (IT) infrastructures are often vulnerable to cyberattacks. To ensure security to the computer systems in an industrial environment, it is required to build effective intrusion detection systems to monitor the cyber-physical systems (e.g., computer networks) in the industry for malicious activities. This paper aims to build such intrusion detection systems to protect the computer networks from cyberattacks. More specifically, we propose a novel unsupervised machine learning approach that combines the K-Means algorithm with the Isolation Forest for anomaly detection in industrial big data scenarios. Since our objective is to build the intrusion detection system for the big data scenario in the industrial domain, we utilize the Apache Spark framework to implement our proposed model which was trained in large network traffic data (about 123 million instances of network traffic) stored in Elasticsearch. Moreover, we evaluate our proposed model on the live streaming data and find that our proposed system can be used for real-time anomaly detection in the industrial setup. In addition, we address different challenges that we face while training our model on large datasets and explicitly describe how these issues were resolved. Based on our empirical evaluation in different use-cases for anomaly detection in real-world network traffic data, we observe that our proposed system is effective to detect anomalies in big data scenarios. Finally, we evaluate our proposed model on several academic datasets to compare with other models and find that it provides comparable performance with other state-of-the-art approaches.

研究の動機と目的

  • ラベル付きデータが入手不可能な大規模かつラベルなし産業用ネットワークトラフィックデータに対して、孤立木の学習を実行する課題に対処すること。
  • K-Meansクラスタリングを用いてデータを分割・前処理することで、大規模データに適用した場合の孤立木における不安定性とスケーラビリティの問題を克服すること。
  • 産業用サイバーフィジカルシステムにおけるライブストリーミングネットワークデータに対してリアルタイムな異常検出を可能にすること。
  • 12300万件の実世界の大規模データセットと学術的ベンチマークの両方で、提案手法の有効性を実証し、最先端のモデルと同等の性能を達成すること。
  • 孤立木における主な学習課題を解決することで、大規模データ環境における異常検出の再現可能フレームワークを提供すること。

提案手法

  • 大規模なネットワークトラフィックデータをK-Meansクラスタリングで小さな、より管理しやすいクラスタに分割し、その後それらを孤立木に供給する。
  • Apache Sparkを用いて学習および推論パイプラインを分散処理し、12300万件のネットワークトラフィックインスタンスのスケーラブルな処理を実現する。
  • 孤立木モデルにK-Meansの重心(Centroids)を追加特徴として統合し、局所的なデータ構造を捉えることで異常検出性能を向上させる。
  • CR(混入率)およびAQRE(異常分位数比推定値)などのハイパーパrameterを最適化し、検出性能とモデル安定性のバランスを取る。
  • 教師なし性を活かしてラベルなしデータでモデルを学習し、K-Meansによるデータの複雑さ低減と収束性の向上を実現する。
  • ライブストリーミングデータに対してリアルタイム推論を実行し、産業環境での展開状況での性能を検証する。
Figure 1: Normal network traffic pattern at iSecurity. Y-axis represents the total number of events occurred on each timestamp represented by the X-axis.
Figure 1: Normal network traffic pattern at iSecurity. Y-axis represents the total number of events occurred on each timestamp represented by the X-axis.

実験結果

リサーチクエスチョン

  • RQ1K-Meansクラスタリングと孤立木を組み合わせることで、大規模産業用ネットワークトラフィックデータにおける異常検出のスケーラビリティと安定性が向上するか?
  • RQ2提案されたIForest-KMeansフレームワークは、従来の孤立木および他の最先端モデルと比較して、実世界の大規模データ上でどの程度の性能を示すか?
  • RQ3K-Meansの統合によって、孤立木における重要なハイパーパrameter(CRおよびAQRE)への依存度はどの程度低下するか?
  • RQ4提案されたシステムは、産業環境におけるライブストリーミングネットワークデータに対してリアルタイムな異常検出を達成できるか?
  • RQ5大規模データセットで孤立木を学習する際の主な課題は何か。また、クラスタリングベースの前処理によってこれらは効果的に緩和できるか?

主な発見

  • CRを0.1から0.4に引き上げた際、Breastwデータセットで48%の性能向上を達成した。これはハイパーパrameterチューニングへの感受性を示している。
  • AQREを0.1から0.5に引き上げた場合、標準的な孤立木の性能は著しく低下し、統計的に有意なp値(≤ 0.05)を示した。これは大規模データにおける不安定性を強調している。
  • 少ないハイパーパrameterを用いながらも、12個の学術的データセットにおいて孤立木と同等の性能を達成した。これは、ロバストネスと一般化能力の高さを示している。
  • iSecurityでのライブストリーミングデータに対して、リアルタイムで異常を検出できた。これは産業環境への展開可能性を裏付けている。
  • Elasticsearchに保存された12300万件のネットワークトラフィックインスタンスを用いてモデルを学習した。これは、実世界の大規模データ環境におけるスケーラビリティと有効性を確認している。
  • 対応t検定により、デフォルト設定と最適設定の間で性能差が統計的に有意(p ≤ 0.05)であることが確認された。これはチューニングの重要性を裏付けている。
Figure 2: Network traffic pattern during a cyberattack targeting the port 3389 using the TCP protocol on January 16th: the sudden spike in the number of events at 1 to 2 am and 5 to 6 am are flagged as cyberattacks by the network expert at iSecurity. At any other time interval, there are no such spi
Figure 2: Network traffic pattern during a cyberattack targeting the port 3389 using the TCP protocol on January 16th: the sudden spike in the number of events at 1 to 2 am and 5 to 6 am are flagged as cyberattacks by the network expert at iSecurity. At any other time interval, there are no such spi

より良い研究を、今すぐ始めましょう

論文の読解から最終レビューまで、研究時間を劇的に削減しましょう。

クレジットカード登録不要

このレビューはAIが作成し、人間の編集者が確認しました。