Toward More Generalized Malicious URL Detection Models
Not on the map. Why?
Abstract (arXiv)
This paper reveals a data bias issue that can severely affect the performance while conducting a machine learning model for malicious URL detection. We describe how such bias can be identified using interpretable machine learning techniques, and further argue that such biases naturally exist in the real world security data for training a classification model. We then propose a debiased training strategy that can be applied to most deep-learning based models to alleviate the negative effects from the biased features. The solution is based on the technique of self-supervised adversarial training to train deep neural networks learning invariant embedding from biased data. We conduct a wide range of experiments to demonstrate that the proposed strategy can lead to significantly better generalization capability for both CNN-based and RNN-based detection models.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 13 of its 23 references that are in the map. First posted on arXiv 21 Feb 2022.
Builds on
13 of 23 references.
References · 13
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2017SmoothGrad: removing noise by adding noisecited by 1,097
- 2018Learning Not to Learn: Training Deep Neural Networks with Biased Datacited by 255
- 2015The Variational Fair Autoencodercited by 223
- 2015Censoring Representations with an Adversarycited by 191
- 2017Data Decisions and Theoretical Implications when Adversarially Learning Fair Representationscited by 165
- 2017Controllable Invariance through Adversarial Feature Learningcited by 127
- 2017Malware Detection by Eating a Whole EXEcited by 114
- 2018Turning a Blind Eye: Explicit Removal of Biases and Variation from Deep Neural Network Embeddingscited by 114
- 2019Adversarial Invariant Feature Learning with Accuracy Constraint for Domain Generalizationcited by 43
- 2018URLNet: Learning a URL Representation with Deep Learning for Malicious URL Detectioncited by 34
1 more
- 2016SoK: Applying Machine Learning in Security - A Surveycited by 1
Built on by
Citing papers per year: 2024: 5; 2025: 4.
9 papers.
- 2024Code Less, Align More: Efficient LLM Fine-tuning for Code Generation with Data Pruningcited by 19
- 2024Enhance Modality Robustness in Text-Centric Multimodal Alignment with Adversarial Promptingcited by 5
- 2024Enhance the Robustness of Text-Centric Multimodal Alignmentscited by 4
- 2025Uncertainty Profiles for LLMs: Uncertainty Source Decomposition and Adaptive Model-Metric Selectioncited by 4
- 2024Continuous Multi-Task Pre-training for Malicious URL Detection and Webpage Classificationcited by 3
- 2024Benchmarking Large Language Model Uncertainty for Prompt Optimizationcited by 2
- 2025From Past to Present: A Survey of Malicious URL Detection Techniques, Datasets and Code Repositoriescited by 1
- 2025WebGuard++:Interpretable Malicious URL Detection via Bidirectional Fusion of HTML Subgraphs and Multi-Scale Convolutional BERTcited by 1
- 2025Multimodal Chip Physical Design Engineer Assistantcited by 1
- 2025Multimodal Chip Physical Design Engineer Assistant2 Jul 2025
- 2025WebGuard++:Interpretable Malicious URL Detection via Bidirectional Fusion of HTML Subgraphs and Multi-Scale Convolutional BERT24 Jun 2025
- 2025Uncertainty Profiles for LLMs: Uncertainty Source Decomposition and Adaptive Model-Metric Selection12 May 2025
- 2025From Past to Present: A Survey of Malicious URL Detection Techniques, Datasets and Code Repositories23 Apr 2025
- 2024Benchmarking Large Language Model Uncertainty for Prompt Optimization16 Sep 2024
- 2024Enhance Modality Robustness in Text-Centric Multimodal Alignment with Adversarial Prompting19 Aug 2024
- 2024Code Less, Align More: Efficient LLM Fine-tuning for Code Generation with Data Pruning6 Jul 2024
- 2024Enhance the Robustness of Text-Centric Multimodal Alignments6 Jul 2024
- 2024Continuous Multi-Task Pre-training for Malicious URL Detection and Webpage Classification18 Feb 2024