Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Retrofit: Continual Learning with Controlled Forgetting for Binary Security Detection and Analysis

About

Binary security has increasingly relied on deep learning to reason about malware behavior and program semantics. However, the performance often degrades as threat landscapes evolve and code representations shift. While continual learning (CL) offers a natural solution through sequential updates, most existing approaches rely on data replay or unconstrained updates, limiting their applicability and effectiveness in data-sensitive security environments. We propose RETROFIT, which regulates knowledge retention and adaptation with controlled forgetting at each update, without requiring historical data. Our key idea is to consolidate previously trained and newly fine-tuned models, serving as teachers of legacy and emergent knowledge, through retrospective-free parameter merging. Forgetting control is achieved by 1) constraining parameter changes to low-rank and sparse subspaces for approximate orthogonality, and 2) employing a confidence-guided arbitration mechanism to dynamically aggregate knowledge from both teachers. Our evaluation on two representative applications demonstrates that RETROFIT consistently mitigates forgetting while maintaining adaptability. In malware detection under temporal drift, it substantially improves the retention score, from 20.2% to 38.6% over CL baselines, and exceeds the oracle upper bound on new data. In binary summarization across decompilation levels, where analyzing stripped binaries is especially challenging, RETROFIT achieves over 2x the BLEU score of transfer learning used in prior work and surpasses all baselines in cross-representation generalization.

Yiling He, Junchi Lei, Hongyu She, Shuo Shao, Xinran Zheng, Yiping Liu, Zhan Qin, Lorenzo Cavallaro• 2025

Related benchmarks

TaskDatasetResultRank
Binary AnalysisCAPYBARA Dec-Strip
ROUGE-L19.8
13
Binary AnalysisCAPYBARA Dec-Strip
EM Score5.62
13
Binary AnalysisCAPYBARA Dec-Strip
METEOR Score18.2
13
Binary AnalysisCAPYBARA Dec-Anon
ROUGE-L31.1
13
Binary AnalysisCAPYBARA XRep
ROUGE-L32.4
13
Binary AnalysisCAPYBARA Dec-Anon
Exact Match (EM)15.77
13
Binary AnalysisCAPYBARA XRep
EM14.82
13
Binary AnalysisCAPYBARA Dec-Anon
METEOR27.7
13
Binary AnalysisCAPYBARA XRep
METEOR0.29
13
Binary AnalysisCAPYBARA Dec-Full
ROUGE-L46.3
13
Showing 10 of 22 rows

Other info

Follow for update