Rethinking Large Language Model Architectures for Sequential Recommendations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Hanbing, Liu, Xiaorui, Fan, Wenqi, Zhao, Xiangyu, Kini, Venkataramana, Yadav, Devendra, Wang, Fei, Wen, Zhen, Tang, Jiliang, Liu, Hui
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910332301082624
author Wang, Hanbing
Liu, Xiaorui
Fan, Wenqi
Zhao, Xiangyu
Kini, Venkataramana
Yadav, Devendra
Wang, Fei
Wen, Zhen
Tang, Jiliang
Liu, Hui
author_facet Wang, Hanbing
Liu, Xiaorui
Fan, Wenqi
Zhao, Xiangyu
Kini, Venkataramana
Yadav, Devendra
Wang, Fei
Wen, Zhen
Tang, Jiliang
Liu, Hui
contents Recently, sequential recommendation has been adapted to the LLM paradigm to enjoy the power of LLMs. LLM-based methods usually formulate recommendation information into natural language and the model is trained to predict the next item in an auto-regressive manner. Despite their notable success, the substantial computational overhead of inference poses a significant obstacle to their real-world applicability. In this work, we endeavor to streamline existing LLM-based recommendation models and propose a simple yet highly effective model Lite-LLM4Rec. The primary goal of Lite-LLM4Rec is to achieve efficient inference for the sequential recommendation task. Lite-LLM4Rec circumvents the beam search decoding by using a straight item projection head for ranking scores generation. This design stems from our empirical observation that beam search decoding is ultimately unnecessary for sequential recommendations. Additionally, Lite-LLM4Rec introduces a hierarchical LLM structure tailored to efficiently handle the extensive contextual information associated with items, thereby reducing computational overhead while enjoying the capabilities of LLMs. Experiments on three publicly available datasets corroborate the effectiveness of Lite-LLM4Rec in both performance and inference efficiency (notably 46.8% performance improvement and 97.28% efficiency improvement on ML-1m) over existing LLM-based methods. Our implementations will be open sourced.
format Preprint
id arxiv_https___arxiv_org_abs_2402_09543
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Rethinking Large Language Model Architectures for Sequential Recommendations
Wang, Hanbing
Liu, Xiaorui
Fan, Wenqi
Zhao, Xiangyu
Kini, Venkataramana
Yadav, Devendra
Wang, Fei
Wen, Zhen
Tang, Jiliang
Liu, Hui
Information Retrieval
Recently, sequential recommendation has been adapted to the LLM paradigm to enjoy the power of LLMs. LLM-based methods usually formulate recommendation information into natural language and the model is trained to predict the next item in an auto-regressive manner. Despite their notable success, the substantial computational overhead of inference poses a significant obstacle to their real-world applicability. In this work, we endeavor to streamline existing LLM-based recommendation models and propose a simple yet highly effective model Lite-LLM4Rec. The primary goal of Lite-LLM4Rec is to achieve efficient inference for the sequential recommendation task. Lite-LLM4Rec circumvents the beam search decoding by using a straight item projection head for ranking scores generation. This design stems from our empirical observation that beam search decoding is ultimately unnecessary for sequential recommendations. Additionally, Lite-LLM4Rec introduces a hierarchical LLM structure tailored to efficiently handle the extensive contextual information associated with items, thereby reducing computational overhead while enjoying the capabilities of LLMs. Experiments on three publicly available datasets corroborate the effectiveness of Lite-LLM4Rec in both performance and inference efficiency (notably 46.8% performance improvement and 97.28% efficiency improvement on ML-1m) over existing LLM-based methods. Our implementations will be open sourced.
title Rethinking Large Language Model Architectures for Sequential Recommendations
topic Information Retrieval
url https://arxiv.org/abs/2402.09543