mlsanityData to 7 Oct 2026

Automatic Identification of Parallelizable Loops Using Transformer-Based Source Code Representations

arXiv 2603.30040 ↗First posted 31 Mar 2026cs.SE cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Automatic parallelization remains a challenging problem in software engineering, particularly in identifying code regions where loops can be safely executed in parallel on modern multi-core architectures. Traditional static analysis techniques, such as dependence analysis and polyhedral models, often struggle with irregular or dynamically structured code. In this work, we propose a Transformer-based approach to classify the parallelization potential of source code, focusing on distinguishing independent (parallelizable) loops from undefined ones. We adopt DistilBERT to process source code sequences using subword tokenization, enabling the model to capture contextual syntactic and semantic patterns without handcrafted features. The approach is evaluated on a balanced dataset combining synthetically generated loops and manually annotated real-world code, using 10-fold cross-validation and multiple performance metrics. Results show consistently high performance, with mean accuracy above 99% and low false positive rates, demonstrating robustness and reliability. Compared to prior token-based methods, the proposed approach simplifies preprocessing while improving generalization and maintaining computational efficiency. These findings highlight the potential of lightweight Transformer models for practical identification of parallelization opportunities at the loop level.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 7 of its 20 references that are in the map. First posted on arXiv 31 Mar 2026.

Builds on

7 of 20 references.

References · 7

  1. 2017Attention Is All You Needcited by 59,932
  2. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  3. 2020CodeBERT: A Pre-Trained Model for Programming and Natural Languagescited by 1,286
  4. 2020GraphCodeBERT: Pre-training Code Representations with Data Flowcited by 600
  5. 2017A Survey of Machine Learning for Big Code and Naturalnesscited by 220
  6. 2016A Convolutional Attention Network for Extreme Summarization of Source Codecited by 90
  7. 2025Discovering Software Parallelization Points Using Deep Neural Networkscited by 1

Built on by

Not cited yet.