Structure‐Aware Representation Learning for Effective Performance Prediction
ABSTRACT Application performance is a function of several unknowns stemming from the interactions between the application, runtime, OS, and underlying hardware, making it challenging to model performance using deep learning techniques, especially without a large labeled dataset. Collecting such labeled longitudinal datasets can take weeks. Intuitively, developers could save analysis time during code development by taking a comparative approach between multiple applications. However, the unknown dynamic interactions between applications and execution environments make it difficult for deep learning‐based models to predict the performance of new applications. In this paper, we address these problems by presenting a labeled dataset for the community and taking a comparative analysis approach to explore the source code differences between different correct implementations of the same problem. This paper assesses the feasibility of using purely static information, for example, Abstract Syntax Tree (AST), of applications to predict performance change based on code structure. We evaluate several deep learning‐based representation learning techniques for source code and propose an architecture for the tree‐based Long Short‐Term Memory (LSTM) models to discover latent representations for a source code's hierarchical structure. We demonstrate that our proposed architecture enables feed‐forward predictive models to predict change in performance using source code with up to 84% accuracy.
Authors
- Tanzima Islam (ORCID: https://orcid.org/0000-0003-2877-5871)
- Chase Phelps (ORCID: https://orcid.org/0000-0003-1480-0917)
- Nathan Pinnow
- Jayaraman J. Thiagarajan (ORCID: https://orcid.org/0000-0002-8517-5816)
- Tarek Ramadan (ORCID: https://orcid.org/0000-0002-6489-7668)
Institutions
- Lawrence Livermore National Laboratory (US)
- Texas State University (US)
Publication Details
- Journal
- Concurrency and Computation Practice and Experience
- Published
- 2025-04-08
- DOI
- https://doi.org/10.1002/cpe.70046
- Primary Topic
- Software System Performance and Reliability
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- U.S. Department of Energy
- Advanced Micro Devices
- University of Texas at Austin
- Texas State University
- Office of Science
- Department of Mechanical Engineering, University of Texas at Austin
- Advanced Scientific Computing Research