Saved in:
Bibliographic Details
Main Authors: Mo, Mingqiao, Tan, Yunlong, Zhang, Hao, Zhang, Heng, He, Yangfan
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2601.20679
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908793950961664
author Mo, Mingqiao
Tan, Yunlong
Zhang, Hao
Zhang, Heng
He, Yangfan
author_facet Mo, Mingqiao
Tan, Yunlong
Zhang, Hao
Zhang, Heng
He, Yangfan
contents Large language models (LLMs) have achieved remarkable progress in code generation, yet their potential for software protection remains largely untapped. Reverse engineering continues to threaten software security, while traditional virtual machine protection (VMP) relies on rigid, rule-based transformations that are costly to design and vulnerable to automated analysis. In this work, we present the first protection-aware framework that learns robust representations of VMP-protected code. Our approach builds large-scale paired datasets of source code and normalized VM implementations, and introduces hierarchical dependency modeling at intra-, preceding-, and inter-instruction levels. We jointly optimize language modeling with functionality-aware and protection-aware contrastive objectives to capture both semantic equivalence and protection strength. To further assess resilience, we propose a protection effectiveness optimization task that quantifies and ranks different VM variants derived from the same source. Coupled with a two-stage continual pre-training and fine-tuning pipeline, our method enables models to generate, compare, and reason over protected code. Extensive experiments show that our framework significantly improves robustness across diverse protection levels, opening a new research direction for learning-based software defense. In this work, we present ShieldedCode, the first protection-aware framework that learns robust representations of VMP-protected code. Our method achieves 26.95% Pass@1 on L0 VM code generation compared to 22.58% for GPT-4o., and improves binary similarity detection Recall@1 by 10% over state of art methods like jTrans.
format Preprint
id arxiv_https___arxiv_org_abs_2601_20679
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ShieldedCode: Learning Robust Representations for Virtual Machine Protected Code
Mo, Mingqiao
Tan, Yunlong
Zhang, Hao
Zhang, Heng
He, Yangfan
Computation and Language
Large language models (LLMs) have achieved remarkable progress in code generation, yet their potential for software protection remains largely untapped. Reverse engineering continues to threaten software security, while traditional virtual machine protection (VMP) relies on rigid, rule-based transformations that are costly to design and vulnerable to automated analysis. In this work, we present the first protection-aware framework that learns robust representations of VMP-protected code. Our approach builds large-scale paired datasets of source code and normalized VM implementations, and introduces hierarchical dependency modeling at intra-, preceding-, and inter-instruction levels. We jointly optimize language modeling with functionality-aware and protection-aware contrastive objectives to capture both semantic equivalence and protection strength. To further assess resilience, we propose a protection effectiveness optimization task that quantifies and ranks different VM variants derived from the same source. Coupled with a two-stage continual pre-training and fine-tuning pipeline, our method enables models to generate, compare, and reason over protected code. Extensive experiments show that our framework significantly improves robustness across diverse protection levels, opening a new research direction for learning-based software defense. In this work, we present ShieldedCode, the first protection-aware framework that learns robust representations of VMP-protected code. Our method achieves 26.95% Pass@1 on L0 VM code generation compared to 22.58% for GPT-4o., and improves binary similarity detection Recall@1 by 10% over state of art methods like jTrans.
title ShieldedCode: Learning Robust Representations for Virtual Machine Protected Code
topic Computation and Language
url https://arxiv.org/abs/2601.20679