mlsanityData to 14 Oct 2026

Teacher-free Latent Self-distillation and Class-separable Representations for Lightweight IoT Attack Detection

arXiv 2403.15509 ↗First posted 22 Mar 2024cs.CR cs.AI cs.LGcited by 0

On the map: Knowledge distillation → Federated, robust and private distillation

Abstract (arXiv)

Knowledge distillation (KD) has been widely used to improve lightweight AI models by transferring soft-label knowledge from a large teacher model to a student model. However, existing KD methods are primarily designed for the image domain rather than lightweight IoT devices, and they often struggle to maintain well-separated feature representations for different attack types, especially as the number of classes increases and attack behaviors become more diverse. This paper proposes a novel textitteacher-free latent self-distillation framework based on a Twin Autoencoder (TAE). Instead of relying on an external teacher, TAE self-learns intrinsic class-wise latent representations, which act as soft labels, similar to KD, but without requiring a teacher model. The decoder then projects the input back into soft labels, enforcing class separation in the decoder output. The resulting decoder representations are used for classification, improving the discrimination between benign and malicious traffic while maintaining a lightweight design suitable for IoT deployment. We theoretically derive conditions for perfect class separation and show that lower empirical risk yields better representations, identifying regimes where TAE achieves strictly lower empirical risk than models using fixed class centers. Extensive experiments on 13 cybersecurity datasets, covering IoT botnets, network intrusion detection, malware, cloud DDoS, and synthetic multi-class data, show that TAE achieves up to 96.1% average accuracy for IoT attack detection and 98.7% for cloud intrusion detection. With a compact model size (1 MB) and ultra-fast inference (0.26 μs per sample), TAE offers a practical and scalable solution for real-world cybersecurity and IoT systems.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Federated, robust and private distillation, one of 12 directions in Knowledge distillation. Builds on 13 of its 44 references that are in the map. First posted on arXiv 22 Mar 2024.

Builds on

13 of 44 references.

References · 13

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  4. 2016Understanding deep learning requires rethinking generalizationcited by 1,674
  5. 2023When Do Neural Nets Outperform Boosted Trees on Tabular Data?cited by 223
  6. 2019Preventing Posterior Collapse with delta-VAEscited by 86
  7. 2018N-BaIoT: Network-based Detection of IoT Botnet Attacks Using Deep Autoencoderscited by 67
  8. 2021Preventing Oversmoothing in VAE via Generalized Variance Parameterizationcited by 15
  9. 2021Self-Distilled Self-Supervised Representation Learningcited by 14
  10. 2022IGRF-RFE: A Hybrid Feature Selection Method for MLP-based Network Intrusion Detection on UNSW-NB15 Datasetcited by 7
  11. 2023Constrained Twin Variational Auto-Encoder for Intrusion Detection in IoT Systemscited by 5
  12. 2023The Role of Codeword-to-Class Assignments in Error-Correcting Codes: An Empirical Studycited by 2
1 more
  1. 2023Online Self-Supervised Deep Learning for Intrusion Detection Systemscited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Lightweight Intrusion Detection in IoT via SHAP-Guided Feature Pruning and Knowledge-Distilled Kronecker Networks
  2. 2025LENS-XAI: Redefining Lightweight and Explainable Network Security through Knowledge Distillation and Variational Autoencoders for Scalable Intrusion Detection in Cybersecurity
  3. 2026Unlearning to Protect: A Distilled Reinforcement Learning Framework with Privacy-Preserving Feature Unlearning and XAI for IoT Security
  4. 2026Semantic Multi-Agent Intrusion Detection for IoT:Zero-Day and Adversarial Threats with Risk-Aware Reasoning
  5. 2026LUT-Compiled Kolmogorov-Arnold Networks for Lightweight DoS Detection on IoT Edge Devices
  6. 2026Lightweight LLMs for Network Attack Detection in IoT Networks ↗
  7. 2026Mean Masked Autoencoder with Flow-Mixing for Encrypted Traffic Classification
  8. 2026Latent Diffusion for Internet of Things Attack Data Generation in Intrusion Detection