End-to-End Neural Ad-hoc Ranking with Kernel Pooling
Carnegie Mellon University · Tsinghua University · +1 more institution
Abstract
This paper proposes K-NRM, a kernel based neural model for document ranking. Given a query and a set of documents, K-NRM uses a translation matrix that models word-level similarities via word embeddings, a new kernel-pooling technique that uses kernels to extract multi-level soft match features, and a learning-to-rank layer that combines those features into the final ranking score. The whole model is trained end-to-end. The ranking layer learns desired feature patterns from the pairwise ranking loss. The kernels transfer the feature patterns into soft-match targets at each similarity level and enforce them on the translation matrix. The word embeddings are tuned accordingly so that they can produce the desired…
Citation impact
- FWCI
- 49.06
- Percentile
- 100%
- References
- 29
Authors
5Topics & keywords
- Computer science
- Ranking (information retrieval)
- Kernel (algebra)
- Artificial intelligence
- Pooling
- Feature (linguistics)
- Similarity (geometry)
- Word (group theory)