arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.03466cs.CVcs.LG

隐含几何的味觉:可扩展的低秩矩阵分解用于推荐系统

Latent Geometry of Taste: Scalable Low-Rank Matrix Factorization for Recommender Systems

African Institute for Mathematical Sciences (AIMS) South Africa(非洲数学科学研究所(AIMS)南非)

Joshua Salako

AI总结:

本文提出了一种可扩展的低秩矩阵分解方法,用于推荐系统,通过优化模型参数和可视化嵌入空间,提升了推荐精度和冷启动场景下的表现。

摘要:

可扩展性和数据稀疏性仍然是在大规模交互数据集上进行协同过滤的关键瓶颈。本文利用MovieLens 32M数据集研究用户偏好的隐含几何,实现了一个高性能、并行化的交替最小二乘法(ALS)框架。通过广泛的超参数优化,我们证明了受约束的低秩模型在一般化方面显著优于高维模型,在均方误差(RMSE)和排名精度之间达到最佳平衡。我们可视化了学习的嵌入空间,揭示了无监督出现的语义类型聚类,证实了模型仅从交互数据中捕捉到深层次的结构关系。最后,我们在冷启动场景中验证了系统的实际实用性,引入了一个可调的评分参数,以有效管理流行度偏差与个性化亲和力之间的权衡。该研究的代码库可在此处找到:https://github.com/joshsalako/recommender.git

英文摘要:

Scalability and data sparsity remain critical bottlenecks for collaborative filtering on massive interaction datasets. This work investigates the latent geometry of user preferences using the MovieLens 32M dataset, implementing a high-performance, parallelized Alternating Least Squares (ALS) framework. Through extensive hyperparameter optimization, we demonstrate that constrained low-rank models significantly outperform higher dimensional counterparts in generalization, achieving an optimal balance between Root Mean Square Error (RMSE) and ranking precision. We visualize the learned embedding space to reveal the unsupervised emergence of semantic genre clusters, confirming that the model captures deep structural relationships solely from interaction data. Finally, we validate the system's practical utility in a cold-start scenario, introducing a tunable scoring parameter to manage the trade-off between popularity bias and personalized affinity effectively. The codebase for this research can be found here: https://github.com/joshsalako/recommender.git

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑