用盒子嵌入解决推荐中的集合关系问题,支持复杂筛选条件。
A Geometric Approach to Personalized Recommendation with Set-Theoretic Constraints Using Box Embeddings
- 用超矩形表示用户和属性,实现集合运算的几何建模
- 在复杂查询上比向量方法最高提升30%准确率
- 适合需要精确筛选条件的个性化推荐场景
个性化推荐常因数据稀疏而受限,传统低秩矩阵分解假设用户与物品由线性相关特征表示,难以捕捉复杂交互。本文将推荐问题建模为具有集合依赖关系的矩阵补全任务,采用超矩形(盒子)嵌入表示用户和属性,即笛卡尔积形式的区间组合。盒子嵌入可直观理解为可训练的维恩图,不仅通过交集指数自然表达相似性,还能精准支持任意集合运算(如“喜剧且动作,非浪漫”)。通过在嵌入空间直接进行几何操作,可高效处理包含集合约束的查询。实验表明,在简单与复杂推荐查询上,该方法相比基于向量的神经方法性能提升最高达30%。
原文摘要 · Abstract (English)
Personalized item recommendation typically suffers from data sparsity, which is most often addressed by learning vector representations of users and items via low-rank matrix factorization. While this effectively densifies the matrix by assuming users and movies can be represented by linearly dependent latent features, it does not capture more complicated interactions. For example, vector representations struggle with set-theoretic relationships, such as negation and intersection, e.g. recommending a movie that is "comedy and action, but not romance". In this work, we formulate the problem of personalized item recommendation as matrix completion where rows are set-theoretically dependent. To capture this set-theoretic dependence we represent each user and attribute by a hyper-rectangle or box (i.e. a Cartesian product of intervals). Box embeddings can intuitively be understood as trainable Venn diagrams, and thus not only inherently represent similarity (via the Jaccard index), but also naturally and faithfully support arbitrary set-theoretic relationships. Queries involving set-theoretic constraints can be efficiently computed directly on the embedding space by performing geometric operations on the representations. We empirically demonstrate the superiority of box embeddings over vector-based neural methods on both simple and complex item recommendation queries by up to 30 \% overall.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。