Memory-Based Malware Detection under Limited Data Conditions: A Comparative Evaluation of TabPFN and Ensemble Models
By: Valentin Leroy, Shuvalaxmi Dass, Sharif Ullah
Potential Business Impact:
Finds computer viruses with little data.
Artificial intelligence and machine learning have significantly advanced malware research by enabling automated threat detection and behavior analysis. However, the availability of exploitable data is limited, due to the absence of large datasets with real-world data. Despite the progress of AI in cybersecurity, malware analysis still suffers from this data scarcity, which limits model generalization. In order to tackle this difficulty, this workinvestigates TabPFN, a learning-free model designed for low-data regimes. We evaluate its performance against established baselines such as Random Forest, LightGBM and XGBoost, across multiple class configurations. Our experimental results indicate that TabPFN surpasses all other models in low-data regimes, with a 2% to 6% improvement observed across multiple performance metrics. However, this increase in performance has an impact on its computation time in a particular case. These findings highlight both the promise and the practical limitations of integrating TabPFN into cybersecurity workflows.
Similar Papers
TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Models
Machine Learning (CS)
Makes computers learn from bigger, more complex data.
nanoTabPFN: A Lightweight and Educational Reimplementation of TabPFN
Machine Learning (CS)
Makes smart computer models easy to learn.
Real-TabPFN: Improving Tabular Foundation Models via Continued Pre-training With Real-World Data
Machine Learning (CS)
Makes computers learn from fake data better.