PETALS: run and fine-tune 50B+ LLMs by pooling unreliable consumer GPUs over the Internet

December 13, 20237 min

Overview

Decision SnapshotNeeds Validation

The design is implemented and tested on real geo-distributed machines with multiple LLMs; code is available, but privacy, security, and incentive mechanisms need engineering before wide production use.

Citations13

Evidence Strength0.80

Confidence0.85

Risk Signals11

Trust Signals

Findings with numeric evidence: 4/4

Findings with evidence refs: 4/4

Results with explicit delta: 3/3

Reproducibility

Status: Partial assets available

Open source: Yes

At A Glance

Cost impact: 80%

Production readiness: 75%

Novelty: 60%

Authors

Alexander Borzunov, Max Ryabinin, Artem Chumachenko, Dmitry Baranchuk, Tim Dettmers, Younes Belkada, Pavel Samygin, Colin Raffel

Links

Abstract / PDF / Code

Why It Matters For Business

PETALS lets teams share idle consumer GPUs to run 50B+ models interactively, cutting the need for expensive multi‑GPU servers and lowering inference latency versus RAM offloading; consider privacy and trust tradeoffs.

Who Should Care

Summary TLDR

This paper introduces PETALS, a decentralized system and algorithms that let you run and fine-tune very large language models (50B+ params) by pooling consumer GPUs over the Internet. Key ideas: a fault‑tolerant pipeline-parallel inference algorithm (dual client/server caches), a decentralized load balancer that assigns contiguous transformer blocks to servers, and support for parameter‑efficient fine-tuning. In experiments PETALS runs Llama 2 (70B) and BLOOM (176B) across geo-distributed machines and reports ≥10× speedups versus single-GPU RAM offloading for interactive generation, while using quantization to cut memory and bandwidth needs.

Problem Statement

Large LLMs (50B+ params) need expensive multi‑GPU servers. Offloading parameters to RAM or SSD is slow for interactive use. The paper tackles running inference and parameter‑efficient fine-tuning on many unreliable, heterogeneous, geo-distributed consumer GPUs while handling node disconnections and uneven hardware.

Main Contribution

A fault‑tolerant pipeline‑parallel inference algorithm using dual caches (server-side and client-side) that recovers from server disconnects without restarting generation.

A fully decentralized load‑balancing protocol that assigns contiguous transformer blocks to servers to maximize tokens/sec under churn.

Key Findings

Distributed approach (PETALS) gives big interactive speedups vs single‑GPU offloading.

Numbers≥10× faster for autoregressive generation (paper claim)

Practical UseIf you can pool idle GPUs, PETALS makes interactive use of 50B+ models practical without buying multi‑GPU servers.

Evidence RefAbstract; Section 4.2; Tables 2–3

Algorithm 1 retains throughput under server failures while naive caching fails.

NumbersBLOOM-7.1B, 128 tokens, failure rate 0.01: Algorithm1 3.38 steps/s vs caching+restarts 0.18 steps/s

Practical UseUse PETALS’ dual‑cache method when servers are unreliable (spot/volunteer nodes) to avoid long restarts.

Evidence RefTable 1; Section 4.1

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Sequential inference (steps/s)Llama 2 (70B) on 3×T4: 2.29 steps/s (128 tokens, 1 Gbit/s, <5 ms RTT)Offloading: 0.139 steps/s~16×Table 2, Section 4.2Measured sequential steps/s per client in controlled networkTable 2
Sequential inference (steps/s)BLOOM (176B) on 3×A100: 1.71 steps/s (128 tokens, 1 Gbit/s, <5 ms RTT)Offloading theoretical: 0.0495 steps/s~34×Table 3, Section 4.2Measured sequential steps/s per client in controlled networkTable 3

What To Try In 7 Days

Install PETALS and run a small public model on a home lab to see pipeline behavior.

Benchmark a 7B/70B model over your network vs RAM offloading to quantify speedup.

Enable 8-bit or 4-bit quantization and compare quality on a few downstream tasks (zero-shot checks).

Optimization Features

Token Efficiency
client-side caching reduces per-step data to kilobytes
Infra Optimization
use of volunteer/spot GPUs to lower hardware costs
Model Optimization
8-bit matrix decomposition4-bit NormalFloat
System Optimization
decentralized greedy load balancingshortest-path routing (D* Lite) for chain selection
Training Optimization
parameter-efficient fine-tuning (adapters, soft prompts)gradient checkpointing
Inference Optimization
pipeline parallelism across serversdynamic blockwise quantization of activations

Reproducibility

Code AvailableYes
Data AvailableNo
Open Source StatusYes
LicenseUnknown

Risks & Boundaries

Limitations

Servers holding early model blocks can see client inputs, so privacy is a concern.

Malicious or faulty servers may return incorrect outputs; validators are proposed but not fully deployed.

When Not To Use

Handling highly sensitive data without trusted or private peers.

When you need the absolute lowest single-node latency (NVLink/local multi‑GPU).

Failure Modes

Servers returning incorrect results (malicious or broken)

High network latency or low bandwidth reducing interactive performance

Core Entities

Models

Llama 2 (70B)BLOOM (176B)BLOOM (7.1B)

Metrics

steps/stokens/sfailure rate

Context Entities

Models

OPT-175B