RobustEncoder: Leveraging K-Means clustering technique to defend NLP models against backdoor attacks
As machine learning (ML) systems become increasingly integrated into real-world applications for sensitive tasks, ensuring the security and privacy of these models becomes paramount. Deep Neural Networks (DNNs), in particular, are susceptible to backdoor attacks, where adversaries manipulate training data by inserting specially crafted samples. While the NLP community has extensively studied backdoor attacks, there remains a gap in effective defense mechanisms. To address this, we propose RobustEncoder, a novel approach leveraging K-Means clustering to detect and mitigate backdoor attacks in text-based models. Our method demonstrates significant efficacy in identifying and neutralizing backdoor triggers, as evidenced by extensive empirical evaluations. Additionally, we propose potential applications of blockchain technology to further enhance the security and integrity of the defense mechanisms in future implementations.
Authors
- Hewa Majeed Zangana (ORCID: https://orcid.org/0000-0001-7909-254X)
- Marwan Omar (ORCID: https://orcid.org/0000-0002-3392-0052)
- Jamal N. Al-Karaki (ORCID: https://orcid.org/0009-0000-7833-3970)
- Luay Albtosh (ORCID: https://orcid.org/0009-0009-0338-9123)
- Derek Mohammed
Institutions
- Saint Leo University (US)
- Illinois Institute of Technology (US)
- Duhok Polytechnic University (IQ)
- Zayed University (AE)
- Capitol Technology University (US)
Publication Details
- Published
- 2024-11-26
- DOI
- https://doi.org/10.1109/bcca62388.2024.10844395
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- article
- Field-Weighted Citation Impact
- 0.00