Salvato in:
| Autori principali: | , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2025
|
| Soggetti: | |
| Accesso online: | https://arxiv.org/abs/2505.17505 |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866909799243841536 |
|---|---|
| author | Liu, Xiaohao Xia, Xiaobo Zhao, Weixiang Zhang, Manyi Yu, Xianzhi Su, Xiu Yang, Shuo Ng, See-Kiong Chua, Tat-Seng |
| author_facet | Liu, Xiaohao Xia, Xiaobo Zhao, Weixiang Zhang, Manyi Yu, Xianzhi Su, Xiu Yang, Shuo Ng, See-Kiong Chua, Tat-Seng |
| contents | Large language models (LLMs) have achieved notable progress. Despite their success, next-token prediction (NTP), the dominant method for LLM training and inference, is constrained in both contextual coverage and inference efficiency due to its inherently sequential process. To overcome these challenges, we propose leap multi-token prediction~(L-MTP), an innovative token prediction method that extends the capabilities of multi-token prediction (MTP) by introducing a leap-based mechanism. Unlike conventional MTP, which generates multiple tokens at adjacent positions, L-MTP strategically skips over intermediate tokens, predicting non-sequential ones in a single forward pass. This structured leap not only enhances the model's ability to capture long-range dependencies but also enables a decoding strategy specially optimized for non-sequential leap token generation, effectively accelerating inference. We theoretically demonstrate the benefit of L-MTP in improving inference efficiency. Experiments across diverse benchmarks validate its merit in boosting both LLM performance and inference speed. The source code is available at https://github.com/Xiaohao-Liu/L-MTP. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2505_17505 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Models Liu, Xiaohao Xia, Xiaobo Zhao, Weixiang Zhang, Manyi Yu, Xianzhi Su, Xiu Yang, Shuo Ng, See-Kiong Chua, Tat-Seng Computation and Language Large language models (LLMs) have achieved notable progress. Despite their success, next-token prediction (NTP), the dominant method for LLM training and inference, is constrained in both contextual coverage and inference efficiency due to its inherently sequential process. To overcome these challenges, we propose leap multi-token prediction~(L-MTP), an innovative token prediction method that extends the capabilities of multi-token prediction (MTP) by introducing a leap-based mechanism. Unlike conventional MTP, which generates multiple tokens at adjacent positions, L-MTP strategically skips over intermediate tokens, predicting non-sequential ones in a single forward pass. This structured leap not only enhances the model's ability to capture long-range dependencies but also enables a decoding strategy specially optimized for non-sequential leap token generation, effectively accelerating inference. We theoretically demonstrate the benefit of L-MTP in improving inference efficiency. Experiments across diverse benchmarks validate its merit in boosting both LLM performance and inference speed. The source code is available at https://github.com/Xiaohao-Liu/L-MTP. |
| title | L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Models |
| topic | Computation and Language |
| url | https://arxiv.org/abs/2505.17505 |