FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUs
Authors: Zhenheng Tang, Yuxin Wang, Xin He, Longteng Zhang, Xinglin Pan, Qiang Wang, Rongfei Zeng, Kaiyong Zhao, Shaohuai Shi, Bingsheng He, Xiaowen Chu
AI-generated Key Points
- Rapid growth of memory and computation requirements of large language models (LLMs)
- Challenges for individuals without access to high-end GPUs
- Consumer-level GPUs often overlooked in LLM due to weaker computing performance, smaller storage capacity, and lower communication bandwidth
- Proposal of a decentralized system called FusionAI to leverage consumer-level GPUs for pre-training, inference, and fine-tuning of LLMs while ensuring privacy protection
- Key components of FusionAI include a broker with a backup pool for dynamic join and quit of computing providers, task scheduling based on hardware performance, and abstracting machine learning procedures into directed acyclic graphs (DAGs)
- DAG approach enables model and task universality and provides compatibility across different devices and deep learning frameworks
- Performance analysis shows that 50 RTX 3080 GPUs can achieve comparable throughputs to 4 H100 GPUs at a significantly lower cost
- FusionAI unlocks the potential of consumer-level GPUs for training and deploying LLMs in a decentralized manner while addressing privacy concerns
- Proposed system design effectively tackles challenges related to resource limitations and device heterogeneity
Abstract: The rapid growth of memory and computation requirements of large language models (LLMs) has outpaced the development of hardware, hindering people who lack large-scale high-end GPUs from training or deploying LLMs. However, consumer-level GPUs, which constitute a larger market share, are typically overlooked in LLM due to their weaker computing performance, smaller storage capacity, and lower communication bandwidth. Additionally, users may have privacy concerns when interacting with remote LLMs. In this paper, we envision a decentralized system unlocking the potential vast untapped consumer-level GPUs in pre-training, inference and fine-tuning of LLMs with privacy protection. However, this system faces critical challenges, including limited CPU and GPU memory, low network bandwidth, the variability of peer and device heterogeneity. To address these challenges, our system design incorporates: 1) a broker with backup pool to implement dynamic join and quit of computing providers; 2) task scheduling with hardware performance to improve system efficiency; 3) abstracting ML procedures into directed acyclic graphs (DAGs) to achieve model and task universality; 4) abstracting intermediate represention and execution planes to ensure compatibility of various devices and deep learning (DL) frameworks. Our performance analysis demonstrates that 50 RTX 3080 GPUs can achieve throughputs comparable to those of 4 H100 GPUs, which are significantly more expensive.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.