计算机科学
一般化
理论(学习稳定性)
人工智能
编码器
二次方程
国家(计算机科学)
模式识别(心理学)
机器学习
空格(标点符号)
算法
数据挖掘
自编码
状态空间
编码(内存)
门控
匹配(统计)
钥匙(锁)
序列模式挖掘
作者
Fanrong Kong,Shaopeng Guan
标识
DOI:10.1093/comjnl/bxag107
摘要
Abstract Sequential recommendation aims to predict a user’s next interaction by modeling the sequential dependencies in historical behaviors. However, Transformer-based methods incur quadratic computational cost on long sequences and often struggle to capture behavioral patterns at different temporal scales, particularly short-term interest shifts. To address these limitations, we propose SEMamba, a sequential recommendation model based on state space models. SEMamba employs a Mamba encoder to model global dependencies and capture long-term user preferences. It further introduces a local spatial enhancement module to extract short-term interest patterns from neighboring interactions. A position-wise gating mechanism adaptively modulates the locally aggregated features before integrating them with the Mamba output. Experiments on the ML-1M, KuaiRand, and Mind datasets show that SEMamba improves Hit Rate by 1.60%, NDCG by 2.01%, and MRR by 2.23% on average over strong baselines. The results also demonstrate improved training stability and better generalization across datasets.
科研通智能强力驱动
Strongly Powered by AbleSci AI