On the Reliability and Stability of Selective Methods in Malware Classification Tasks
Organizations: University College London and core64 London, United Kingdom · University College London London, United Kingdom
Abstract
The performance figures of modern drift-adaptive malware classifiers appear promising, but does this translate to genuine operational reliability? The standard evaluation paradigm primarily focuses on baseline performance metrics, neglecting confidence-error alignment and operational stability. While prior works established the importance of temporal evaluation and introduced selective classification in malware classification tasks, we take a complementary direction by investigating whether malware classifiers maintain reliable and stable confidence estimates under distribution shifts and exploring the tensions between scientific advancement and practical impacts when they do not. We propose Aurora, a framework to evaluate malware classifiers based on their confidence quality and operational resilience. Aurora subjects the confidence profile of a given model to verification to assess the reliability of its estimates. Unreliable confidence estimates erode operational trust, waste valuable annotation budgets on non-informative samples for active learning, and leave error-prone instances undetected in selective classification. Aurora is further complemented by a set of metrics designed to go beyond point-in-time performance, striving towards a more holistic assessment of operational stability throughout temporal evaluation periods. The fragility we observe in SOTA frameworks across datasets of varying drift severity suggests it may be time to revisit the underlying assumptions.
Figures & tables
| Baseline | Reliability | Stability | |||||||||
| Method | F1 | E-AURC | Cap 400 | F1 ∗ | Sen[F1] | Sen[AUROC f ] | Sen[ ] | ||||
| % | 100 | % of errors | % | F1 ∗ (pp) | pp/month | /month | first last | pp/month | (%) | ||
| 400 | DeepDrebin | 38.5 | 4.53 | 30.3 | 31.1 | 7.4 | 1.23 | 3.6 | 0.72 0.65 | 0.01 | 9.9 |
| CADE-OOD | 38.3 | 8.20 | 18.0 | 42.5 | 4.2 | 0.81 | 1.4 | 0.56 0.51 | 0.24 | 12.7 | |
| CADE-MSP | 38.3 | 7.04 | 21.4 | 2.7 | 35.5 | 0.81 | 1.8 | 0.54 0.59 | 0.48 | 23.6 | |
| HCC-PL | 39.1 | 7.12 | 17.2 | 43.2 | 4.1 | 1.04 | 5.8 | 0.62 0.48 | 0.23 | 16.2 | |
| Baseline | Reliability | Stability | |||
| F1 | E-AURC | Cap | F1 | Sen[F1] | |
| highest, % | lowest, 100 | highest, % | rows | range, pp/month | |
| 50 | DRMD 33.1 | CCE (DD) 4.95 | RAMDA 7.2 | 10/16 | to |
| 100 | DRMD 36.0 | DD, CCE (DD) 4.85 | RAMDA 12.0 | 10/16 | to |
| 200 | DRMD 40.2 | DD 4.69 | RAMDA 19.2 | 10/16 | to |
| 400 | DRMD 43.5 | DD 4.53 | DRMD 30.6 | 9/16 | to |
| Baseline | Reliability | Stability | ||
| Dataset | F1 | E-AURC | Sen[F1] | |
| highest, % | lowest, 100 | range, pp/month | ||
| AndroZoo | 50 | HCC 75.1 | CCE (Drebin) 1.24 | to |
| 100 | HCC 76.9 | CCE (Drebin) 0.93 | to | |
| 200 | HCC 78.3 | CCE (Drebin) 0.82 | to | |
| 400 | HCC 84.5 | CCE (Drebin) 0.63 | to |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Baseline | Reliability | Stability | |||||||||
| Method | F1 | E-AURC | Cap | F1 ∗ | Sen[F1] | Sen[AUROC f ] | Sen[ ] | ||||
| % | 100 | % of errors | % | F1 ∗ (pp) | pp/month | /month | first last | pp/month | (%) | ||
| 50 | DeepDrebin | 29.9 | 5.07 | 6.5 | 28.3 | 1.5 | 1.38 | 4.7 | 0.73 0.62 | 0.01 | 1.1 |
| CADE-OOD c | 26.9 | 8.27 | 4.5 | 28.2 | 1.3 | 1.28 | 0.6 | 0.55 0.52 | 0.04 | 4.0 | |
| CADE-MSP c | 26.9 | 7.84 | 6.0 | 18.9 | 8.2 | 1.28 | 0.4 | 0.53 0.54 | 0.01 | 2.3 | |
| HCC-PL a | 27.9 | 6.56 | 4.3 | 28.1 | 0.2 | 1.33 | 4.7 | 0.64 0.53 | 0.01 | 1.2 | |
| Baseline | Reliability | Stability | |||||||||
| Method | F1 | E-AURC | Cap | F1 ∗ | Sen[F1] | Sen[AUROC f ] | Sen[ ] | ||||
| % | 100 | % of errors | % | F1 ∗ (pp) | pp/month | /month | first last | pp/month | (%) | ||
| 50 | DeepDrebin | 54.2 | 5.34 | 14.3 | 55.2 | 1.0 | 1.81 | 24.5 | 0.76 0.46 | 0.11 | 3.0 |
| CADE-OOD c | 62.4 | 5.95 | 4.1 | 62.4 † | 0.0 † | 0.66 | 4.0 | 0.47 0.42 | 0.15 † | 1.1 † | |
| CADE-MSP c | 62.4 | 5.43 | 11.1 | 60.7 | 1.7 | 0.66 | 2.1 | 0.55 0.50 | 0.07 | 4.6 | |
| HCC-PL a | 75.1 | 1.27 | 26.8 | 76.5 | 1.3 | 0.79 | 7.9 | 0.75 0.85 | 0.31 | 2.1 | |
| Baseline | Reliability | Stability | |||||||||
| Method | F1 | E-AURC | Cap | F1 ∗ | Sen[F1] | Sen[AUROC f ] | Sen[ ] | ||||
| % | 100 | % of errors | % | F1 ∗ (pp) | pp/month | /month | first last | pp/month | (%) | ||
| 50 | DeepDrebin | 89.7 | 0.61 | 28.3 | 91.6 | 1.8 | 0.12 | 1.4 | 0.85 0.85 | 0.01 | 1.4 |
| CADE-OOD c | 85.8 | 1.94 | 4.3 | 84.8 | 1.0 | 0.03 | 1.3 | 0.69 0.55 | 0.01 | 1.1 | |
| CADE-MSP c | 85.8 | 4.15 | 5.4 | 86.4 | 0.6 | 0.03 | 0.1 | 0.38 0.43 | 0.01 | 1.9 | |
| HCC-PL a | 88.6 | 1.78 | 19.6 | 90.0 | 1.4 | 0.06 | 1.9 | 0.78 0.63 | 0.00 | 1.4 | |
| Baseline | Reliability | Stability | |||||||||
| Method | F1 | E-AURC | Cap | F1 ∗ | Sen[F1] | Sen[AUROC f ] | Sen[ ] | ||||
| % | 100 | % of errors | % | F1 ∗ (pp) | pp/month | /month | first last | pp/month | (%) | ||
| 50 | DeepDrebin | 82.5 | 1.42 | 18.7 | 84.4 | 1.9 | 0.02 | 3.2 | 0.83 0.74 | 0.00 | 1.3 |
| CADE-OOD c | 68.7 | 4.65 | 5.8 | 66.0 | 2.8 | 0.46 | 0.9 | 0.55 0.52 | 0.00 | 1.2 | |
| CADE-MSP c | 68.7 | 6.11 | 5.2 | 65.4 | 3.3 | 0.46 | 1.2 | 0.48 0.52 | 0.00 | 2.7 | |
| HCC-PL a | 82.7 | 4.47 | 15.6 | 83.8 | 1.1 | 0.05 | 3.0 | 0.68 0.61 | 0.01 | 1.4 | |
| Metric | Definition | Literature | Exclusion rationale | VTT2 evidence |
|---|---|---|---|---|
| Reliability | ||||
| Mean failure AUROC (AUROC f ) | Probability that an error receives higher uncertainty than a correct prediction; threshold-free ranking. | Error-detection baseline [ 33 ] ; AUROC [ 68 ] ; ranking quality [ 37 ] . | Omits error frequency and high-confidence error concentration [ 69 , 55 ] ; order-preserving score transforms tie. Retained for Sen[AUROC f ] and its first last companion. | with E-AURC. |
| AUGRC | Area under generalized risk (probability of an accepted error) across coverage. | Traub et al. [ 53 ] . | Weights selective risk by coverage, downweighting the high-confidence region targeted by Aurora . | with E-AURC; DRMD–DeepDrebin gap: 0.87 E-AURC points versus 0.26 AUGRC points. |
| NAURC | Excess area normalized by its ceiling: 0 for oracle ranking, 1 for an uninformative score; removes error-rate dependence. | Proposed by Cattelan and Silva [ 52 ] ; used by Heng and Soh [ 57 ] . | Closely tracks failure AUROC on this cohort; E-AURC retains error weighting and is interpreted alongside F1. | with mean failure AUROC, with E-AURC. |
| SR 10 | Error rate among the 10% most confident monthly predictions. | Fixed-coverage risk [ 39 , 70 ] ; partial AURC [ 56 ] . | Covers the region E-AURC weights most; reported with E-AURC in the results text. | with E-AURC (13 rows); range 3.4–7.0% versus about 8% overall error. |
| AURC[F1] ∗ | Area under accepted-set F1 over target coverages 5–100%; thresholds fixed after calibration on the preceding window. | No prior precedent; Transcend reports accepted-set F1 at one rate [ 45 , 5 ] . | F1 is non-additive over coverage; drift separates target from realized coverage. Excluding months with fewer than 20 accepted samples gives methods different support. Replaced by F1 ∗ at the single calibrated operating point. | Largest with a headline metric: 0.68 ( F1 ∗ , per-seed values); distinct but not interpretable. |
Malware Classification
Malware classification uses features of files, apps, and behavior to identify malicious code and assign it to families or types.
See all 44 papers