Decoding in Latent Spaces for Efficient Inference in LLM-based Recommendation
By: Chengbing Wang , Yang Zhang , Zhicheng Wang and more
Potential Business Impact:
Makes computer recommendations much faster.
Fine-tuning large language models (LLMs) for recommendation in a generative manner has delivered promising results, but encounters significant inference overhead due to autoregressive decoding in the language space. This work explores bypassing language-space decoding by directly matching candidate items with the LLM's internal thought representations in the latent space, eliminating the time-consuming autoregressive process to reduce computational costs. Towards this, we introduce Light Latent-space Decoding (L2D), an effective and efficient latent-space decoding method. L2D represents user-preferred items by using the hidden states of test sequences reflecting the LLM's internal thought, and obtains candidate item representations from the hidden states of training sequences labeled with the corresponding candidate items. It then matches the two types of representations to decode items, achieving latent-space decoding. In this way, it enables efficient decoding without altering the LLM's generative tuning paradigm, thereby preserving performance. Extensive empirical results demonstrate that L2D is more than 10x faster than language-space decoding while maintaining or enhancing performance.
Similar Papers
Latent Refinement Decoding: Enhancing Diffusion-Based Language Models by Refining Belief States
Computation and Language
Makes AI write faster and smarter.
Latent Diffusion Inversion Requires Understanding the Latent Space
Machine Learning (CS)
Finds hidden personal data in AI art.
LaDiR: Latent Diffusion Enhances LLMs for Text Reasoning
Machine Learning (CS)
Helps computers think and fix their own mistakes.