-
From Words to Watts: Benchmarking the Energy Costs of Large Language Model Inference
Authors:
Siddharth Samsi,
Dan Zhao,
Joseph McDonald,
Baolin Li,
Adam Michaleas,
Michael Jones,
William Bergeron,
Jeremy Kepner,
Devesh Tiwari,
Vijay Gadepally
Abstract:
Large language models (LLMs) have exploded in popularity due to their new generative capabilities that go far beyond prior state-of-the-art. These technologies are increasingly being leveraged in various domains such as law, finance, and medicine. However, these models carry significant computational challenges, especially the compute and energy costs required for inference. Inference energy costs…
▽ More
Large language models (LLMs) have exploded in popularity due to their new generative capabilities that go far beyond prior state-of-the-art. These technologies are increasingly being leveraged in various domains such as law, finance, and medicine. However, these models carry significant computational challenges, especially the compute and energy costs required for inference. Inference energy costs already receive less attention than the energy costs of training LLMs -- despite how often these large models are called on to conduct inference in reality (e.g., ChatGPT). As these state-of-the-art LLMs see increasing usage and deployment in various domains, a better understanding of their resource utilization is crucial for cost-savings, scaling performance, efficient hardware usage, and optimal inference strategies.
In this paper, we describe experiments conducted to study the computational and energy utilization of inference with LLMs. We benchmark and conduct a preliminary analysis of the inference performance and inference energy costs of different sizes of LLaMA -- a recent state-of-the-art LLM -- developed by Meta AI on two generations of popular GPUs (NVIDIA V100 \& A100) and two datasets (Alpaca and GSM8K) to reflect the diverse set of tasks/benchmarks for LLMs in research and practice. We present the results of multi-node, multi-GPU inference using model sharding across up to 32 GPUs. To our knowledge, our work is the one of the first to study LLM inference performance from the perspective of computational and energy resources at this scale.
△ Less
Submitted 4 October, 2023;
originally announced October 2023.
-
Hypersparse Network Flow Analysis of Packets with GraphBLAS
Authors:
Tyler Trigg,
Chad Meiners,
Sandeep Pisharody,
Hayden Jananthan,
Michael Jones,
Adam Michaleas,
Timothy Davis,
Erik Welch,
William Arcand,
David Bestor,
William Bergeron,
Chansup Byun,
Vijay Gadepally,
Micheal Houle,
Matthew Hubbell,
Anna Klein,
Peter Michaleas,
Lauren Milechin,
Julie Mullen,
Andrew Prout,
Albert Reuther,
Antonio Rosa,
Siddharth Samsi,
Doug Stetson,
Charles Yee
, et al. (1 additional authors not shown)
Abstract:
Internet analysis is a major challenge due to the volume and rate of network traffic. In lieu of analyzing traffic as raw packets, network analysts often rely on compressed network flows (netflows) that contain the start time, stop time, source, destination, and number of packets in each direction. However, many traffic analyses benefit from temporal aggregation of multiple simultaneous netflows,…
▽ More
Internet analysis is a major challenge due to the volume and rate of network traffic. In lieu of analyzing traffic as raw packets, network analysts often rely on compressed network flows (netflows) that contain the start time, stop time, source, destination, and number of packets in each direction. However, many traffic analyses benefit from temporal aggregation of multiple simultaneous netflows, which can be computationally challenging. To alleviate this concern, a novel netflow compression and resampling method has been developed leveraging GraphBLAS hyperspace traffic matrices that preserve anonymization while enabling subrange analysis. Standard multitemporal spatial analyses are then performed on each subrange to generate detailed statistical aggregates of the source packets, source fan-out, unique links, destination fan-in, and destination packets of each subrange which can then be used for background modeling and anomaly detection. A simple file format based on GraphBLAS sparse matrices is developed for storing these statistical aggregates. This method is scale tested on the MIT SuperCloud using a 50 trillion packet netflow corpus from several hundred sites collected over several months. The resulting compression achieved is significant (<0.1 bit per packet) enabling extremely large netflow analyses to be stored and transported. The single node parallel performance is analyzed in terms of both processors and threads showing that a single node can perform hundreds of simultaneous analyses at over a million packets/sec (roughly equivalent to a 10 Gigabit link).
△ Less
Submitted 13 September, 2022;
originally announced September 2022.
-
The MIT Supercloud Workload Classification Challenge
Authors:
Benny J. Tang,
Qiqi Chen,
Matthew L. Weiss,
Nathan Frey,
Joseph McDonald,
David Bestor,
Charles Yee,
William Arcand,
Chansup Byun,
Daniel Edelman,
Matthew Hubbell,
Michael Jones,
Jeremy Kepner,
Anna Klein,
Adam Michaleas,
Peter Michaleas,
Lauren Milechin,
Julia Mullen,
Andrew Prout,
Albert Reuther,
Antonio Rosa,
Andrew Bowne,
Lindsey McEvoy,
Baolin Li,
Devesh Tiwari
, et al. (2 additional authors not shown)
Abstract:
High-Performance Computing (HPC) centers and cloud providers support an increasingly diverse set of applications on heterogenous hardware. As Artificial Intelligence (AI) and Machine Learning (ML) workloads have become an increasingly larger share of the compute workloads, new approaches to optimized resource usage, allocation, and deployment of new AI frameworks are needed. By identifying compute…
▽ More
High-Performance Computing (HPC) centers and cloud providers support an increasingly diverse set of applications on heterogenous hardware. As Artificial Intelligence (AI) and Machine Learning (ML) workloads have become an increasingly larger share of the compute workloads, new approaches to optimized resource usage, allocation, and deployment of new AI frameworks are needed. By identifying compute workloads and their utilization characteristics, HPC systems may be able to better match available resources with the application demand. By leveraging datacenter instrumentation, it may be possible to develop AI-based approaches that can identify workloads and provide feedback to researchers and datacenter operators for improving operational efficiency. To enable this research, we released the MIT Supercloud Dataset, which provides detailed monitoring logs from the MIT Supercloud cluster. This dataset includes CPU and GPU usage by jobs, memory usage, and file system logs. In this paper, we present a workload classification challenge based on this dataset. We introduce a labelled dataset that can be used to develop new approaches to workload classification and present initial results based on existing approaches. The goal of this challenge is to foster algorithmic innovations in the analysis of compute workloads that can achieve higher accuracy than existing methods. Data and code will be made publicly available via the Datacenter Challenge website : https://dcc.mit.edu.
△ Less
Submitted 13 April, 2022; v1 submitted 12 April, 2022;
originally announced April 2022.
-
The MIT Supercloud Dataset
Authors:
Siddharth Samsi,
Matthew L Weiss,
David Bestor,
Baolin Li,
Michael Jones,
Albert Reuther,
Daniel Edelman,
William Arcand,
Chansup Byun,
John Holodnack,
Matthew Hubbell,
Jeremy Kepner,
Anna Klein,
Joseph McDonald,
Adam Michaleas,
Peter Michaleas,
Lauren Milechin,
Julia Mullen,
Charles Yee,
Benjamin Price,
Andrew Prout,
Antonio Rosa,
Allan Vanterpool,
Lindsey McEvoy,
Anson Cheng
, et al. (2 additional authors not shown)
Abstract:
Artificial intelligence (AI) and Machine learning (ML) workloads are an increasingly larger share of the compute workloads in traditional High-Performance Computing (HPC) centers and commercial cloud systems. This has led to changes in deployment approaches of HPC clusters and the commercial cloud, as well as a new focus on approaches to optimized resource usage, allocations and deployment of new…
▽ More
Artificial intelligence (AI) and Machine learning (ML) workloads are an increasingly larger share of the compute workloads in traditional High-Performance Computing (HPC) centers and commercial cloud systems. This has led to changes in deployment approaches of HPC clusters and the commercial cloud, as well as a new focus on approaches to optimized resource usage, allocations and deployment of new AI frame- works, and capabilities such as Jupyter notebooks to enable rapid prototy** and deployment. With these changes, there is a need to better understand cluster/datacenter operations with the goal of develo** improved scheduling policies, identifying inefficiencies in resource utilization, energy/power consumption, failure prediction, and identifying policy violations. In this paper we introduce the MIT Supercloud Dataset which aims to foster innovative AI/ML approaches to the analysis of large scale HPC and datacenter/cloud operations. We provide detailed monitoring logs from the MIT Supercloud system, which include CPU and GPU usage by jobs, memory usage, file system logs, and physical monitoring data. This paper discusses the details of the dataset, collection methodology, data availability, and discusses potential challenge problems being developed using this data. Datasets and future challenge announcements will be available via https://dcc.mit.edu.
△ Less
Submitted 4 August, 2021;
originally announced August 2021.
-
Active Learning Pipeline for Brain Map** in a High Performance Computing Environment
Authors:
Adam Michaleas,
Lars A. Gjesteby,
Michael Snyder,
David Chavez,
Meagan Ash,
Matthew A. Melton,
Damon G. Lamb,
Sara N. Burke,
Kevin J. Otto,
Lee Kamentsky,
Webster Guan,
Kwanghun Chung,
Laura J. Brattain
Abstract:
This paper describes a scalable active learning pipeline prototype for large-scale brain map** that leverages high performance computing power. It enables high-throughput evaluation of algorithm results, which, after human review, are used for iterative machine learning model training. Image processing and machine learning are performed in a batch layer. Benchmark testing of image processing usi…
▽ More
This paper describes a scalable active learning pipeline prototype for large-scale brain map** that leverages high performance computing power. It enables high-throughput evaluation of algorithm results, which, after human review, are used for iterative machine learning model training. Image processing and machine learning are performed in a batch layer. Benchmark testing of image processing using pMATLAB shows that a 100$\times$ increase in throughput (10,000%) can be achieved while total processing time only increases by 9% on Xeon-G6 CPUs and by 22% on Xeon-E5 CPUs, indicating robust scalability. The images and algorithm results are provided through a serving layer to a browser-based user interface for interactive review. This pipeline has the potential to greatly reduce the manual annotation burden and improve the overall performance of machine learning-based brain map**.
△ Less
Submitted 25 June, 2020;
originally announced June 2020.