Skip to content

Attack and Anomaly Detection on IoT devices with the use of Kolmogorov-Arnold-Network(KAN).

Notifications You must be signed in to change notification settings

Shubhm1shra/Attack-Detection-on-IoT-Devices-with-KAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

Attack-Detection-on-IoT-Devices-with-KAN

DataSet : https://www.kaggle.com/datasets/francoisxa/ds2ostraffictraces

Faster-KAN : https://github.com/AthanasiosDelis/faster-kan

Kolmogorov-Arnold-Network (KAN) :-

Kolmogorov-Arnold-Network is a approach to neural network inspired by Kolmogorov-Arnold representation theorem. Unlike MLP, KAN utlizes the concept of updatable activation function rather than fixed activation function which allows them to fulfil the functionality of both weights and activation functions. These activations functions are present at edges rather than nodes. All in all, KANs offer a better accuracy and performance along with better iterpretability but lacks on the side of computational efficiency.

$$MLP : f(x) = \sum_{i=1}^{N(e)}a_{i}\sigma(w_{i}.x + b_{i})$$

$$KAN : f(x) = f(x_{1},...,x_{n}) = ∑_{q=1}^{2n + 1}𝚽_{q}(∑_{p=1}^{n}Φ_{q, p}(x_{p}))$$

For better computational efficiency, I utilized faster-kan by Athanasios Delis which makes use Radial Basis Function, and functions capable of reflectional symmetry.

Guassian-Radial-Basis-Function (RBF):-

A Radial Basis Function is a real-valued function, the value of which depends only on the distance from the origin. Among various types of Radial Basis Function, Guassian Radial Basis Function is the most common.

  • $b_{i}(u) = e^{(\frac{-(u - u_{i})^{2}}{h})}$

Reflextional-Switch-Activation Function (RSWAF):-

Uses function which have reflextionary symmetry, allows us to retain performance while reducing computation time.

  • $b_{i}(u) = 1 - (tanh(\frac{u - u_{i}}{h}))^{2}$

About the model :-

Trained 2 Faster-KAN models with hidden layer sizes as mentioned below, both the models acheived approaximately the same accuracy and final loss, with smaller model being better in the terms of speed capable of executing 45 (approx) iterations/sec. and larger model executing 13 iterations/sec.

  • Input Dimension : 11
  • Hidden Layer Size (Original) : 100
  • Hidden Layer Size (Small) : 21
  • Training Accuracy : 100%
  • Validation Accuracy : 99.3%

About

Attack and Anomaly Detection on IoT devices with the use of Kolmogorov-Arnold-Network(KAN).

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published