Abstract
Encrypted traffic classification is an important technique for network management and network security assurance. Current mainstream methods utilize machine learning and deep learning techniques for classification, and train models with complete data pre-collected in experimental environments to achieve excellent classification performance. However, real network environments are widely affected by packet loss, which causes traffic data to exhibit a different distribution compared to the training and negatively impacts existing methods. To address this problem, in this paper, we propose APMA, an Anti-Packet-loss encrypted traffic classification method based on Masked Autoencoder, which utilizes two traffic-specific mask pre-training tasks to extract generalized knowledge from large-scale unlabeled traffic, and simulates packet loss to learn robust traffic representations that are applicable to different network environments. Extensive evaluations show that APMA can be applied to a variety of classification scenarios and network environments, achieving over 90% classification accuracy even at 15% packet loss.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Bhadra, D.R., Joshi, C.A., Soni, P.R., Vyas, N.P., Jhaveri, R.H.: Packet loss probability in wireless networks: a survey. In: 2015 International Conference on Communications and Signal Processing (ICCSP), pp. 1348–1354. IEEE (2015)
Cavitt, J., Wallace, S., Zhao, X.: Detecting cyber attacks with packet loss resilience for power systems. Sustain. Comput. Inform. Syst. 34, 100629 (2022)
Dadkhah, S., Mahdikhani, H., Danso, P.K., Zohourian, A., Truong, K.A., Ghorbani, A.A.: Towards the development of a realistic multidimensional IoT profiling dataset. In: 2022 19th Annual International Conference on Privacy, Security & Trust (PST), pp. 1–11. IEEE (2022)
Draper-Gil, G., Lashkari, A.H., Mamun, M.S.I., Ghorbani, A.A.: Characterization of encrypted and VPN traffic using time-related. In: Proceedings of the 2nd International Conference on Information Systems Security and Privacy (ICISSP), pp. 407–414 (2016)
Hayes, J., et al.: k-fingerprinting: a robust scalable website fingerprinting technique. In: USENIX security symposium, pp. 1187–1203 (2016)
He, H.Y., Yang, Z.G., Chen, X.N.: PERT: payload encoding representation from transformer for encrypted traffic classification. In: 2020 ITU Kaleidoscope: Industry-Driven Digital Transformation (ITU K), pp. 1–8. IEEE (2020)
He, K., Chen, X., Xie, S., Li, Y., Dollar, P., Girshick, R.: Masked autoencoders are scalable vision learners. In: 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 15979–15988. IEEE (2022)
Lamberti, A.: (2023). https://obkio.com/blog/acceptable-packet-loss/
Lashkari, A.H., Gil, G.D., Mamun, M.S.I., Ghorbani, A.A.: Characterization of tor traffic using time based features. In: International Conference on Information Systems Security and Privacy, vol. 2, pp. 253–262. SciTePress (2017)
Lin, X., Xiong, G., Gou, G., Li, Z., Shi, J., Yu, J.: Et-bert: a contextualized datagram representation with pre-training transformers for encrypted traffic classification. In: Proceedings of the ACM Web Conference 2022, pp. 633–642 (2022)
Liu, C., He, L., Xiong, G., Cao, Z., Li, Z.: Fs-net: a flow sequence network for encrypted traffic classification. In: IEEE INFOCOM 2019-IEEE Conference On Computer Communications, pp. 1171–1179. IEEE (2019)
Mansfield, K.C., Antonakos, J.L.: Computer networking for LANS to WANS: hardware, software and security. Delmar Learning (2009)
Moore, A., Zuev, D., Crogan, M.: Discriminators for use in flow-based classification. Queen Mary, University of London, Department of Computer Science, Tech. rep. (2013)
Shapira, T., Shavitt, Y.: Flowpic: encrypted internet traffic classification is as easy as image recognition. In: IEEE INFOCOM 2019-IEEE Conference on Computer Communications Workshops (INFOCOM WKSHPS), pp. 680–687. IEEE (2019)
Sharafaldin, I., Lashkari, A.H., Ghorbani, A.A.: Toward generating a new intrusion detection dataset and intrusion traffic characterization. ICISSp 1, 108–116 (2018)
Shen, M., Zhang, J., Zhu, L., Xu, K., Du, X., Liu, Y.: Encrypted traffic classification of decentralized applications on ethereum using feature fusion. In: Proceedings of the International Symposium on Quality of Service, pp. 1–10 (2019)
Shi, J., Yang, C., Yang, Q., Zhang, J., Cui, M., Xiong, G.: Spbiseq: an early-stage fingerprint generation method with high robustness for encrypted mobile application traffic. In: 2021 IEEE 4th International Conference on Computer and Communication Engineering Technology (CCET), pp. 185–193. IEEE (2021)
Sirinam, P., Imani, M., Juarez, M., Wright, M.: Deep fingerprinting: undermining website fingerprinting defenses with deep learning. In: Proceedings of the 2018 ACM SIGSAC Conference on Computer and Communications Security, pp. 1928–1943 (2018)
Taylor, V.F., Spolaor, R., Conti, M., Martinovic, I.: Robust smartphone app identification via encrypted network traffic analysis. IEEE Trans. Inf. Forensics Secur. 13(1), 63–78 (2018)
Vaswani, A., et al.: Attention is all you need. Adv. Neural Inf. Process. Syst. 30 (2017)
Wang, W., Zhu, M., Wang, J., Zeng, X., Yang, Z.: End-to-end encrypted traffic classification with one-dimensional convolution neural networks. In: 2017 IEEE International Conference on Intelligence and Security Informatics (ISI), pp. 43–48. IEEE (2017)
Wang, W., Zhu, M., Zeng, X., Ye, X., Sheng, Y.: Malware traffic classification using convolutional neural network for representation learning. In: 2017 International Conference on Information Networking (ICOIN), pp. 712–717. IEEE (2017)
Wang, X., Chen, S., Su, J.: App-net: a hybrid neural network for encrypted mobile traffic classification. In: IEEE INFOCOM 2020-IEEE Conference on Computer Communications Workshops (INFOCOM WKSHPS), pp. 424–429. IEEE (2020)
Zhang, H., et al.: Tfe-gnn: A temporal fusion encoder using graph neural networks for fine-grained encrypted traffic classification. In: Proceedings of the ACM Web Conference 2023, pp. 2066–2075 (2023)
Zhao, R., et al.: Yet another traffic classifier: a masked autoencoder based traffic transformer with multi-level flow representation. In: Proceedings of the AAAI Conference on Artificial Intelligence, pp. 5420–5427 (2023)
Acknowledgements
This study was funded by The National Key Research and Development Program of China (No. 2023YFB3106700).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Ethics declarations
Disclosure of Interests
The authors have no competing interests to declare that are relevant to the content of this article.
Rights and permissions
Copyright information
© 2025 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Yang, C. et al. (2025). Anti-Packet-Loss Encrypted Traffic Classification via Masked Autoencoder. In: Cai, Z., Takabi, D., Guo, S., Zou, Y. (eds) Wireless Artificial Intelligent Computing Systems and Applications. WASA 2024. Lecture Notes in Computer Science, vol 14997. Springer, Cham. https://doi.org/10.1007/978-3-031-71464-1_7
Download citation
DOI: https://doi.org/10.1007/978-3-031-71464-1_7
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-71463-4
Online ISBN: 978-3-031-71464-1
eBook Packages: Computer ScienceComputer Science (R0)