NSF CAREER: Contextual Robustness for ML-powered Network-based Functions

Overview

This project focuses on making Machine Learning (ML) more reliable for important communication networking functions, such as managing the resources of a computer network or fixing network problems. While ML can be powerful, it sometimes fails in unexpected ways and does not provide any guarantees, making it risky for critical network operations, especially due to temporal (time-sensitive) contexts. This project will investigate ML-based approaches to networking functions, identify realistic failure cases, and develop methods to make ML systems more dependable for networking. The goal is to develop NETFORTIFY, an open-source framework that helps researchers and engineers test and improve ML-powered approaches to ensure they work well in real-world conditions.

The proposed research advances ML-based networking through three key thrusts: (1) Contextual Robustness Definition: this includes formal semantics to define robustness in ML-powered approaches to certain networking functions, ensuring they meet required properties under operational constraints, alongside a catalog of transformations for specifying realistic conditions; (2) Robustness Assessment: this task integrates formal methods with adversarial ML to generate failure-directed and realistic scenarios for diverse implementations; and (3) Robustness Enhancement: Leveraging failure-directed scenarios and domain knowledge, this thrust enables logic and adversarial retraining, and robustness certification.

This project will enhance the reliability of ML-powered networking allowing for further automation. By developing NETFORTIFY, an open-source framework for testing and strengthening ML-powered networking, it will set new testing standards for robustness, benefiting researchers, industry, and network operators. Additionally, the project integrates research with education by introducing hands-on learning experiences, such as an interactive NETFORTIFY game, to teach students about ML in networking, and seminars for high school teachers on how to integrate networking technology and ML into STEM education.

Publications

Robustifying ML-powered Network Classifiers with PANTS

Minhao Jin, Maria Apostolaki
USENIX Security 2025
Paper