Performance and Energy Consumption of Parallel Machine Learning Algorithms
Wu, Xidong, Brazzle, Preston, Cahoon, Stephen
–arXiv.org Artificial Intelligence
Machine learning models have achieved remarkable success in various real-world applications such as data science, computer vision, and natural language processing. However, model training in machine learning requires large-scale data sets and multiple iterations before it can work properly. Parallelization of training algorithms is a common strategy to speed up the process of training. Power consumption is also an important metric for any type of computation, especially high-performance applications. Machine learning algorithms that can be used on low-power platforms such as sensors and mobile devices have been researched, but less power optimization is done for algorithms designed for high-performance computing. In this paper, we present a C++ implementation of logistic regression and the genetic algorithm, and a Python implementation of neural networks with stochastic gradient descent (SGD) algorithm on classification tasks. We will show the impact that the complexity of the model and the size of the training data have on the parallel efficiency of the algorithm in terms of both power and performance. We also tested these implementations using shard-memory parallelism, distributed memory parallelism, and GPU acceleration to speed up machine learning model training. Machine learning is a class of data-driven algorithms and models where models progressively improve as they gain experience. It has many applications from image classification to robot control [1]. By providing a set of training data, models can train themselves to accurately process new data outside of the training set.
arXiv.org Artificial Intelligence
May-1-2023
- Country:
- North America > United States
- New York > New York County
- New York City (0.04)
- California > Alameda County
- Berkeley (0.04)
- New York > New York County
- Asia > Middle East
- Jordan (0.04)
- North America > United States
- Genre:
- Research Report > New Finding (0.37)
- Industry:
- Information Technology (0.46)
- Energy (0.41)
- Technology: