Skip to content

Classifier-Dependent Benefits of Pseudo-Labeling for Semi-Supervised Android Malware Attribution

Aug 2026 · 0 citations · 13 references
Computer Science

TL;DR

A systematic evaluation of pseudo-labeling across six classifiers on the CICMalDroid 2020 dataset, using five-fold stratified cross-validation and paired t-tests across five labeled ratios, finds that SSL benefit is strongly classifier-dependent.

Abstract

Detecting and classifying Android malware families remains challenging due to high feature dimensionality, class imbalance, and the high cost of expert-labeled data. Semi-supervised learning (SSL) offers a way to leverage unlabeled samples, but prior works rarely test whether SSL benefits generalize across classifier types or report statistical significance. We present a systematic evaluation of pseudo-labeling across six classifiers (LightGBM, XGBoost, Random Forest, Logistic Regression, MLP, and SVM) on the CICMalDroid 2020 dataset, using five-fold stratified cross-validation and paired t-tests across five labeled ratios (1-20%). We find that SSL benefit is strongly classifier-dependent: SVM shows the largest significant gain (+4.4% accuracy at 5% labels, p = 0.0028), LightGBM improves modestly (+0.8 to +1.3% at 2-5% labels), while Random Forest is significantly harmed at low label ratios (-3.1% at 1% labels). Per-class analysis reveals SSL disproportionately benefits the hardest-to-classify families, with Adware F1 improving by +13.8 percentage points versus only +0.8 for the already well-classified Benign class. We further show that approximately 800 labeled samples (10% of the dataset) yield near-optimal performance across all classifiers. These findings offer practical guidance on when and with which classifier pseudo-labeling is worthwhile for Android malware classification.

View source

Similar papers

Open access Aug 2026

Optimizing Multiclass Android Malware Family Classification Using SMOTE-Tomek Links and XGBoost

The results confirm that hybrid resampling combined with optimized gradient boosting improves classification reliability, especially in addressing severe class imbalance and enhancing recognition capability across diverse Android malware families.

Ali Nur Ikhsan, Adam Prayogo Kuncoro, Debby Ummul Hidayah et al. · 0 citations
Preprint Sep 2026

When Label Noise Meets Class Imbalance: A Robust Framework for Android Malware Family Classification

Machine learning methods for Android malware family classification have achieved high accuracy, but their application is hindered by two major challenges. First, the widely used code obfuscation severely disrupts the automated labeling process and introduces substantial label noise into training datasets. Second, train...

Hao-Lan Zhang, Cui-Ying Gao, Fu-Lin Zhao et al. · 0 citations
Conference Aug 2026

Hybrid Representation Learning for Robust Multiclass Malware Classification Under Class Imbalance

The increasing sophistication of modern malware, particularly in the form of polymorphic and packed variants, poses significant challenges to traditional detection systems. While machine learning-based approaches have shown promise, many existing methods rely on single feature representations and are evaluated using me...

Akshaya Mogili, G. Lishita, R. N · 0 citations
Review Open access Oct 2026

The Illusion of Success: Learning-Based Android Malware Detectors (Replicability Study)

A deliberately naïve and unreliable classifier, which uses application package names as features for classification, performs comparably to and sometimes even better than the state-of-the-art tools when compared under the same setup when compared under the same setup.

Michael Tegegn, Julia Rubin · 0 citations
Open access 2023

COMPARATIVE EVALUATION OF CLASS-IMBALANCE CORRECTION TECHNIQUES FOR SOFTWARE DEFECT PREDICTION

This study compared no correction, random oversampling, random undersampling, SMOTE, ADASYN, and class-weighted learning across logistic regression, decision tree, random forest, support vector machine, and neural network classifiers to find accuracy alone is unsuitable for selecting defect predictors.

L. Akpan · 0 citations

Related blog posts

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.