跳到正文
arXiv · NLP· Haolin Li, Haipeng Zhang, Mang Li, Zhaohui Xia, Xun Zhu, Miao Li, Yu Zhang, Ji Wu·· 4 小时前AI 评分28

CRIM:面向高效个性化检索的可组合、可复用物品记忆框架

CRIM: Composable and Reusable Item Memory for Efficient Personalized Retrieval

AI 导读

CRIM框架将物品级语义编码与请求级历史建模分离,在MovieLens-1M等数据集上取得SOTA检索性能并大幅降低推理成本。 通过可学习摘要token将物品压缩为连续记忆,利用REC token引入预测监督,且记忆可跨请求复用以消除重复编码。 该机制支持联合训练以优化编码器,作者将公开代码和LazSeq数据集。

来源:arXiv · NLP · arxiv.org