-
Notifications
You must be signed in to change notification settings - Fork 46
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
doc: add cloud-edge speculative decoding strategy
Signed-off-by: Yu Fan <[email protected]>
- Loading branch information
1 parent
852973a
commit 7c0e001
Showing
4 changed files
with
100 additions
and
0 deletions.
There are no files selected for viewing
100 changes: 100 additions & 0 deletions
100
...proposals/algorithms/joint-inference/cloud-edge-speculative-decoding-for-llm.md
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,100 @@ | ||
- [Cloud-edge collaborative speculative decoding for LLM based on KubeEdge-Ianvs](#cloud-edge-collaborative-speculative-decoding-for-llm-based-on-kubeedge-ianvs) | ||
- [Motivation](#motivation) | ||
- [Goals](#goals) | ||
- [Proposal](#proposal) | ||
- [Design Details](#design-details) | ||
- [Overall Architecture](#overall-architecture) | ||
- [Speculative Decoding Implementation](#speculative-decoding-implementation) | ||
- [Roadmap](#roadmap) | ||
- [October](#october) | ||
- [November](#november) | ||
- [References](#references) | ||
|
||
|
||
# Cloud-edge collaborative speculative decoding for LLM based on KubeEdge-Ianvs | ||
|
||
## Motivation | ||
|
||
The autoregressive decoding mode of LLM determines that LLM can only be decoded serially, which limits its inference speed. Speculative decoding technique can be used to decode LLM in parallel with the help of draft model, so as to improve the inference speed of LLM without loss of accuracy. However, the speculative decoding technology of LLM does not consider the application in the cloud-edge distributed environment. | ||
|
||
This project aims to implement cloud-edge collaborative speculative decoding based on KubeEdge-Ianvs, an open source cloud-edge collaborative distributed machine learning platform, so as to further improve the LLM inference speed in cloud-edge environment. | ||
|
||
### Goals | ||
|
||
- Implement an example of cloud-edge collaborative speculative decoding based on KubeEdge-Ianvs platform. | ||
- (Optional) Propose a more efficient cloud-edge collaborative speculative decoding algorithm. | ||
|
||
## Proposal | ||
|
||
We propose KubeEdge-Ianvs to adopt the cloud-edge collaborative speculative decoding strategy to enhance LLM system efficiency according to emerging computational scenarios' needs. | ||
|
||
This proposal will utilize Sedna's Joint Inference interface and `query-routing` example | ||
|
||
## Design Details | ||
|
||
### Overall Architecture | ||
|
||
The architecture of this proposal is shown in the figure below. We leverage the existed *TestEnvManager*, *TestCaseController* and *StoryManager* in Ianvs. | ||
|
||
- In *TestEnvManager*, we plan to add Human Eval as LLM benchmark and *Accuracy*, *Latency*, *Throughput*, *Internal Token Latency* as metrics. | ||
- In *TestCaseController*, we plan to add a cloud-edge collaboration algorithm named *Speculative Decoding*. This will inherit the process of *Query Routing*. | ||
- In *StoryManager*, we plan to show Leaderboard and Test Report for users. | ||
|
||
<img src="./images/image-20241014224119569.png" alt="image-20241014224119569" style="zoom: 50%;" /> | ||
|
||
The core progress is herited from the [Query Routing](https://github.com/kubeedge/ianvs/tree/main/examples/cloud-edge-collaborative-inference-for-llm). | ||
|
||
|
||
### Speculative Decoding Implementation | ||
|
||
Basically, there are two types of speculative decoding strategy. | ||
|
||
#### Draft Model Mechanism | ||
|
||
One method uses SLM as a draft model to quickly generate tokens, which are then verified by LLM. | ||
|
||
<img src="./images/image-20241014224729431.png" alt="image-20241014224729431" style="zoom: 33%;" /> | ||
|
||
#### Lookahead Mechanism | ||
The other method builds a Trie tree based on existing documents to predict tokens and also has them verified by LLM. | ||
|
||
<img src="./images/image-20241014224258112.png" alt="image-20241014224258112" style="zoom: 50%;" /> | ||
|
||
### Cloud Edge Collaboration Strategy | ||
|
||
The cloud-edge collaboration for speculative decoding can take many forms, such as: | ||
- The DraftModel is placed on the edge device and Large Model is placed in the cloud. | ||
- Trie tree is placed on the edge device while the Large Model remains in the cloud. | ||
|
||
However, these methods face some challenges in a cloud-edge architecture: | ||
- The edge device might lack sufficient resources, leading to slow model generation and rendering the Draft Model approach ineffective. | ||
- The cloud-edge system often face higher propagation delays, significantly higher than Internal Token Latency (ITL) (e.g., 50ms vs. 5ms), which may lead to a worse performance than the cloud-only system. | ||
|
||
To minimize communication as much as possible, we consider deploying both Draft Model or Trie Tree and Large Model together on the Cloud side, achieving edge-cloud collaboration through [Query Routing](https://github.com/kubeedge/ianvs/tree/main/examples/cloud-edge-collaborative-inference-for-llm). | ||
|
||
The entire process is as follows: | ||
|
||
- Using a Hard Example Mining Module to judge a query is hard or not. | ||
- If easy, send the query to the small model on Edge. | ||
- If hard, send the query to the Large Model equipped with Speculative Decoding Mechanisms on Cloud. | ||
- Mechanism 1: Draft Model Mechanism | ||
- Mechanism 2: Lookahead Mechanism | ||
|
||
## Roadmap | ||
|
||
### October | ||
|
||
- Submit a proposal and build a prototype of speculative decoding. | ||
|
||
### November | ||
|
||
- Implement an example using either the draft model strategy or the documents retrieval strategy. | ||
- PR and merge. | ||
|
||
## References | ||
|
||
[1] H. Xia *et al.*, “Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding.” arXiv, Feb. 20, 2024. Accessed: May 31, 2024. | ||
|
||
[2] He, Zhenyu, et al. "Rest: Retrieval-based speculative decoding." *Proceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies*. 2024 | ||
|
||
[3] Zhao, Yao, et al. "Lookahead: An inference acceleration framework for large language model with lossless generation accuracy." *Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining*. 2024. |
Binary file added
BIN
+76.3 KB
docs/proposals/algorithms/joint-inference/images/image-20241014224119569.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added
BIN
+111 KB
docs/proposals/algorithms/joint-inference/images/image-20241014224258112.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added
BIN
+115 KB
docs/proposals/algorithms/joint-inference/images/image-20241014224729431.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.