Scaling Laws For Dense Retrieval
首次系统研究稠密检索模型的 scaling law,发现检索性能随模型参数量和标注数据量遵循精确的幂律关系。
INSIGHT稠密检索的性能可以用幂律公式精确预测:基于对比似然度(contrastive log-likelihood)指标,模型规模和数据规模与检索效果之间存在可预测的 scaling 关系。这为搜索模型的资源分配提供了定量指导。
业务关联社区搜索升级检索模型时,可利用 scaling law 预测不同模型规模和训练数据量下的 nDCG 提升,从而在有限预算内做出最优资源配置(增加模型参数 vs 增加标注数据)。
证据SIGIR 2024 发表。实验验证在 MS MARCO 等标准数据集上,使用对比似然度(CLL)指标,检索性能与模型规模/数据量之间的幂律拟合度 R² 接近 1.0。同时发现数据增强可显著提升标注效率。