Machine Unlearning for the XGBoost Model with Network Intrusion Datasets
About
Machine Unlearning (MU) has emerged as an important technique for removing specific data points from trained models without requiring full retraining. However, most existing MU research focuses on deep learning and image data, leaving a gap in the domain of network intrusion detection, which relies heavily on tabular data. This work introduces XGBoost-Forget, an unlearning approach for the XGBoost model, to address this gap. The approach is evaluated on two tabular Network Intrusion (NI) datasets, IoT-23 and GeNIS, using multiple metrics to assess model performance, unlearning efficiency, and forgetting quality. The results show that XGBoost-Forget maintains predictive performance close to the original model while providing significantly faster unlearning, demonstrating its potential for MU in tabular NI settings.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Classification | IoT-23 (test) | Accuracy98.2863 | 6 | |
| Attack Success Rate | GeNIS (All shards) | ASR1.5797 | 6 | |
| Tabular Classification | GeNIS | Accuracy99.9781 | 6 | |
| Attack Success Rate | IoT-23 All shards | ASR6.1692 | 4 | |
| Attack Success Rate | IoT-23 Infected shard only | ASR10.4589 | 4 | |
| Attack Success Rate | GeNIS (Infected shard only) | ASR1.584 | 4 | |
| Attack Success Rate | IoT-23 Retraining scope | -- | 2 |