arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.01368cs.NIcs.CRcs.LG
收藏

用于IoT设备分类的成本感知特征选择

Cost-aware Feature Selection for IoT Device Classification

Georgia Institute of Technology(佐治亚理工学院);Trustwave (Singapore)(Trustwave(新加坡));TUMCREATE (Singapore)(TUMCREATE(新加坡));Heriot-Watt University(赫瑞-瓦特大学);National University of Singapore(新加坡国立大学)

Biswadeep Chakraborty, Dinil Mon Divakaran, Ido Nevat, Gareth W. Peters, Mohan Gurusamy

摘要:

从多个角度来看,将物联网设备分类为不同类型至关重要,包括安全和隐私方面。最近的工作探索了用于对物联网设备进行指纹识别(或分类)的机器学习技术,并取得了可喜的成果。然而,现有的工作已经假设用于构建机器学习模型的特征是容易获得的或者可以容易地从网络流量中提取;换句话说,它们没有考虑与特征提取相关联的成本。在这项工作中,我们采取了一个更现实的方法,并认为特征提取有成本,成本是不同的功能。我们还从当前将误分类损失视为二进制值的做法向前迈出了一步,并根据误分类性能对不同的损失进行论证。因此,更重要的是,我们引入了物联网设备分类风险的概念。我们定义并制定了成本感知物联网设备分类的问题。这是一个组合优化问题,我们开发了一种新的算法来解决它在一个快速和有效的方式使用交叉熵(CE)为基础的随机优化技术。使用真实设备的流量,我们证明了基于CE的算法在选择具有最小误分类风险的特征的同时保持特征提取的成本在指定限制内的能力。

英文摘要:

Classification of IoT devices into different types is of paramount importance, from multiple perspectives, including security and privacy aspects. Recent works have explored machine learning techniques for fingerprinting (or classifying) IoT devices, with promising results. However, existing works have assumed that the features used for building the machine learning models are readily available or can be easily extracted from the network traffic; in other words, they do not consider the costs associated with feature extraction. In this work, we take a more realistic approach, and argue that feature extraction has a cost, and the costs are different for different features. We also take a step forward from the current practice of considering the misclassification loss as a binary value, and make a case for different losses based on the misclassification performance. Thereby, and more importantly, we introduce the notion of risk for IoT device classification. We define and formulate the problem of cost-aware IoT device classification. This being a combinatorial optimization problem, we develop a novel algorithm to solve it in a fast and effective way using the Cross-Entropy (CE) based stochastic optimization technique. Using traffic of real devices, we demonstrate the capability of the CE based algorithm in selecting features with minimal risk of misclassification while keeping the cost for feature extraction within a specified limit.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑