Static Program Slicing Using Language Models With Dataflow-Aware Pretraining and Constrained Decoding

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Pengfei, Wang, Shaowei, Chen, Tse-Hsun, Asaduzzaman, Muhammad
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913109313060864
author He, Pengfei
Wang, Shaowei
Chen, Tse-Hsun
Asaduzzaman, Muhammad
author_facet He, Pengfei
Wang, Shaowei
Chen, Tse-Hsun
Asaduzzaman, Muhammad
contents Static program slicing is a fundamental software engineering technique for isolating code relevant to specific variables. While recent learning-based approaches using language models (LMs) show promise in automating slice prediction, they suffer from inaccurate dependency modeling and unconstrained generation, where LMs fail to capture precise data flow relations and produce slices containing hallucinated tokens and statements. To address these challenges, we propose Sliceformer, a novel approach that reformulates static program slicing as a sequence-to-sequence task using small language models such as CodeT5+. Sliceformer introduces two key innovations that directly target the identified limitations. First, to improve dependency modeling, we design dataflow-aware pretraining objectives that leverage data flow graphs (DFG) to teach models data dependencies through dataflow-preserving statement permutation and dataflow-aware span corruption. Second, to eliminate hallucination, we develop a constrained decoding mechanism that enforces both lexical and syntactic constraints. We evaluate Sliceformer on Java and Python program slicing benchmarks, demonstrating consistent improvements over state-of-the-art baselines with up to 22% gain in ExactMatch.
format Preprint
id arxiv_https___arxiv_org_abs_2604_26961
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Static Program Slicing Using Language Models With Dataflow-Aware Pretraining and Constrained Decoding
He, Pengfei
Wang, Shaowei
Chen, Tse-Hsun
Asaduzzaman, Muhammad
Software Engineering
Artificial Intelligence
Programming Languages
Static program slicing is a fundamental software engineering technique for isolating code relevant to specific variables. While recent learning-based approaches using language models (LMs) show promise in automating slice prediction, they suffer from inaccurate dependency modeling and unconstrained generation, where LMs fail to capture precise data flow relations and produce slices containing hallucinated tokens and statements. To address these challenges, we propose Sliceformer, a novel approach that reformulates static program slicing as a sequence-to-sequence task using small language models such as CodeT5+. Sliceformer introduces two key innovations that directly target the identified limitations. First, to improve dependency modeling, we design dataflow-aware pretraining objectives that leverage data flow graphs (DFG) to teach models data dependencies through dataflow-preserving statement permutation and dataflow-aware span corruption. Second, to eliminate hallucination, we develop a constrained decoding mechanism that enforces both lexical and syntactic constraints. We evaluate Sliceformer on Java and Python program slicing benchmarks, demonstrating consistent improvements over state-of-the-art baselines with up to 22% gain in ExactMatch.
title Static Program Slicing Using Language Models With Dataflow-Aware Pretraining and Constrained Decoding
topic Software Engineering
Artificial Intelligence
Programming Languages
url https://arxiv.org/abs/2604.26961