用Transformer预测用户下一首歌,准确率超其他模型。
Sequential choice in ordered bundles
- 采用解码器架构的自定义Transformer模型预测用户行为
- 在Spotify数据上准确预测个体与整体消费选择
- 可应用于个性化推荐和提升内容需求
体验型内容如体育赛事、艺术表演、歌曲、视频、新闻、播客和电视剧常以有序包形式呈现,需逐项消费。本文研究用户是否能根据前序消费模式预测下一项选择。基于Spotify数据,评估了多种模型:两种自定义Transformer(仅解码器与编码器-解码器)、微调GPT-3、自定义LSTM、强化学习模型、两个马尔可夫模型及零阶模型。结果表明,仅解码器架构的自定义Transformer表现最优,既精准预测个体选择,也有效预测整体需求。该模型捕捉到普遍的状态依赖性。注意力权重分析显示,下一项消费受此前所有选择近似均等影响。研究证明,该模型可用于推荐下一首可能播放的歌曲,预测单个内容需求,并个性化推广以提升需求。
原文摘要 · Abstract (English)
Experience goods such as sporting and artistic events, songs, videos, news stories, podcasts, and television series, are often packaged and consumed in bundles. Many such bundles are ordered in the sense that the individual items are consumed sequentially, one at a time. We examine if an individual's decision to consume the next item in an ordered bundle can be predicted based on his/her consumption pattern for the preceding items. We evaluate several predictive models, including two custom Transformers using decoder-only and encoder-decoder architectures, fine-tuned GPT-3, a custom LSTM model, a reinforcement learning model, two Markov models, and a zero-order model. Using data from Spotify, we find that the custom Transformer with a decoder-only architecture provides the most accurate predictions, both for individual choices and aggregate demand. This model captures a general form of state dependence. Analysis of Transformer attention weights suggests that the consumption of the next item in a bundle is based on approximately equal weighting of all preceding choices. Our results indicate that the Transformer can assist in queuing the next item that an individual is likely to consume from an ordered bundle, predicting the demand for individual items, and personalizing promotions to increase demand.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。