arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1805.03403cs.IR
收藏

基于对抗学习的神经排序模型跨域正则化

Cross Domain Regularization for Neural Ranking Models Using Adversarial Learning

Daniel Cohen, Bhaskar Mitra, Katja Hofmann, W. Bruce Croft

摘要:

与传统的依赖于手工特征的排名模型学习不同,神经表示学习模型通过在大型数据集上进行训练来学习更高级别的特征。然而,他们直接从数据中学习新功能的能力可能是有代价的。在没有任何特殊监督的情况下,这些模型学习可能仅在训练数据采样的域中保持的关系,并且很难推广到训练期间未观察到的域。我们研究了对抗性学习作为排名任务背景下的跨域正则化器的有效性。我们使用一个对抗性的神经网络,并在一小部分领域上训练我们的神经排名模型。该模型提供了一个负反馈信号,以阻止模型学习领域特定的表示。我们的实验表明,在存在对抗性攻击的情况下,在保留域上的性能始终更好--有时精度高达30%@1。

英文摘要:

Unlike traditional learning to rank models that depend on hand-crafted features, neural representation learning models learn higher level features for the ranking task by training on large datasets. Their ability to learn new features directly from the data, however, may come at a price. Without any special supervision, these models learn relationships that may hold only in the domain from which the training data is sampled, and generalize poorly to domains not observed during training. We study the effectiveness of adversarial learning as a cross domain regularizer in the context of the ranking task. We use an adversarial discriminator and train our neural ranking model on a small set of domains. The discriminator provides a negative feedback signal to discourage the model from learning domain specific representations. Our experiments show consistently better performance on held out domains in the presence of the adversarial discriminator---sometimes up to 30% on precision@1.

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

↑