内存受限的推荐系统训练:应优先增加批次大小还是负样本数?

_reachsumit · x · 2026-08-12

在大规模神经推荐系统中,采样 softmax 常用于减少最终分类层的内存消耗。但在固定的内存预算下,究竟该优先扩大批次大小还是增加负样本数量,一直缺乏明确指导。

本论文在标准平滑性和方差假设下,从理论和实验两方面给出了答案:最快的收敛速度来自于尽可能多地包含对象(即增大批次大小 $n$)。研究在包括 MovieLens-20M 在内的四个真实基准测试中验证了该结论。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →