arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.10434cs.IRcs.CLcs.LG
收藏

Conformer-Kernel with Query Term Independence for Document Retrieval

Microsoft(微软);UCL(伦敦大学学院);TU Wien(维也纳工业大学)

Bhaskar Mitra, Sebastian Hofstatter, Hamed Zamani, Nick Craswell

英文摘要:

The Transformer-Kernel (TK) model has demonstrated strong reranking performance on the TREC Deep Learning benchmark---and can be considered to be an efficient (but slightly less effective) alternative to BERT-based ranking models. In this work, we extend the TK architecture to the full retrieval setting by incorporating the query term independence assumption. Furthermore, to reduce the memory complexity of the Transformer layers with respect to the input sequence length, we propose a new Conformer layer. We show that the Conformer's GPU memory requirement scales linearly with input sequence length, making it a more viable option when ranking long documents. Finally, we demonstrate that incorporating explicit term matching signal into the model can be particularly useful in the full retrieval setting. We present preliminary results from our work in this paper.

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

↑