• Home
  • Search
  • Using Ensemble Learning to Improve Automatic Vectorization of Tensor Contraction Program
  • Cite Icon5
  • https://doi.org/10.1109/access.2018.2867151Copy DOI Icon

Using Ensemble Learning to Improve Automatic Vectorization of Tensor Contraction Program

Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Automatic vectorization is crucial for improving the performance of computationally intensive programs. Existing compilers use conservative optimization strategies for automatic vectorization, which, in many cases, lead to the loss of vectorization opportunity. Studies have shown that the use of machine learning algorithms to build a performance prediction model is beneficial to improve the program performance. The model input is program features, and the output is the predicted optimization strategies or the program performance related to the optimization. In this paper, we focus on a computational intensive loop structure-tensor contraction, which is common in quantum chemical simulations. Most existing machine learning methods rely on control and data flow graphs as features to represent programs, but different tensor contraction kernels have the same control and data flow graphs. In addition, the existing methods often use the same kind of learning algorithm to construct a learning model, which is prone to overfitting and low-precision problems. In this paper, we propose an automatic vectorization performance enhancement method based on ensemble learning. We construct an ensemble learning model to predict the performance of tensor contraction kernels with different vectorization strategies and select the best vectorization strategy for the kernels. According to the storage access patterns of the tensor contraction kernels, we propose a static method for features representation. Based on the multi-algorithm ensemble learning strategy, we obtain better learning results than the single learning algorithm. The experimental results show that the prediction model achieves 88% and 87% prediction efficiency on two different platforms with different instruction sets, data types, and compilers. Compared with the existing methods, the prediction efficiency is greatly improved. In addition, the average peak performance is $2.96 \times $ of Intel ICC 12.0 and $2.98 \times $ of GCC 4.6 compiler, respectively.

Similar Papers
  • Research Article
  • Citations1

Rethinking Control Flow in Spatial Architectures: Insights Into Control Flow Plane Design

  • Jan 01, 2025
  • IEEE Transactions on Computers
  • Jinyi Deng +8
  • Conference Article

From behavioral description to systolic array based architectures

  • Feb 28, 1994
  • A Balboni +3
  • Conference Article
  • Citations16

Detecting Android application malicious behaviors based on the analysis of control flows and data flows

  • Oct 13, 2017
  • Peter Zegzhda +3
  • Research Article

An Ameliorated Methodology for the Abstraction of Object Oriented Features from Software Requirements Specification

  • Jan 01, 2015
  • Procedia Computer Science
  • A.M Shivaram +1
  • Research Article

점진적 실행을 통한 소프트웨어의 구조 그래프 생성

  • Jul 31, 2013
  • Journal of the Korea Society of Computer and Information
  • Hye-Ryun Lee +4
  • Research Article
  • Citations1

A compiling technique for dataflow machines — new algorithm for optimum translation from control flow graph into dataflow graph

  • Jan 01, 1996
  • Systems and Computers in Japan
  • Toshiaki Yasue +1
  • Conference Article

Automated parallel exploration of datapath and Unrolling Factor in High Level Synthesis using hyper-dimensional particle swarm encoding

  • May 01, 2014
  • Anirban Sengupta +1
  • Conference Article
  • Citations7

Improved interconnect sharing by identity operation insertion

  • Jan 20, 2003
  • D Herrmann +1
  • Conference Article
  • Citations2

BCFA

  • Jun 27, 2020
  • Ramanathan Ramu +3
  • Book Chapter
  • Citations1

Are Textual Prompts in Large Language Models Sufficient for Vulnerability Detection?

  • Jun 27, 2025
  • Puya Pakshad +1
  • Research Article
  • Citations219

BGNN4VD: Constructing Bidirectional Graph Neural-Network for Vulnerability Detection

  • Mar 20, 2021
  • Information and Software Technology
  • Sicong Cao +4
  • Research Article

An integrated graph neural network model for joint software defect prediction and code quality assessment.

  • Dec 11, 2025
  • Scientific reports
  • Ping Dai +3
  • Research Article

Investigation into the use of Machine Learning Algorithms for Detecting Insider Threats through IP Spoofing in Organizational Networks

  • Dec 11, 2025
  • NEWPORT INTERNATIONAL JOURNAL OF ENGINEERING AND PHYSICAL SCIENCES
  • Akawuku I Godspower +4
  • Conference Article
  • Citations22

Adaptive Call-Site Sensitive Control Flow Integrity

  • Jun 01, 2019
  • Mustakimur Khandaker +5
  • Research Article
  • Citations2

Software Weakness Detection in Solidity Smart Contracts Using Control and Data Flow Analysis: A Novel Approach with Graph Neural Networks

  • Aug 10, 2024
  • Electronics
  • Aria Seo +4
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.