SDR: Efficient Neural Re-ranking using Succinct Document Representation

Nachshon Cohen, Amit Portnoy, Besnik Fetahu, Amir Ingber


Abstract
BERT based ranking models have achieved superior performance on various information retrieval tasks. However, the large number of parameters and complex self-attention operations come at a significant latency overhead. To remedy this, recent works propose late-interaction architectures, which allow pre-computation of intermediate document representations, thus reducing latency. Nonetheless, having solved the immediate latency issue, these methods now introduce storage costs and network fetching latency, which limit their adoption in real-life production systems. In this work, we propose the Succinct Document Representation (SDR) scheme that computes highly compressed intermediate document representations, mitigating the storage/network issue. Our approach first reduces the dimension of token representations by encoding them using a novel autoencoder architecture that uses the document’s textual content in both the encoding and decoding phases. After this token encoding step, we further reduce the size of the document representations using modern quantization techniques. Evaluation on MSMARCO’s passage re-reranking task show that compared to existing approaches using compressed document representations, our method is highly efficient, achieving 4x–11.6x higher compression rates for the same ranking quality. Similarly, on the TREC CAR dataset, we achieve 7.7x higher compression rate for the same ranking quality.
Anthology ID:
2022.acl-long.457
Volume:
Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
May
Year:
2022
Address:
Dublin, Ireland
Editors:
Smaranda Muresan, Preslav Nakov, Aline Villavicencio
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
6624–6637
Language:
URL:
https://aclanthology.org/2022.acl-long.457
DOI:
10.18653/v1/2022.acl-long.457
Bibkey:
Cite (ACL):
Nachshon Cohen, Amit Portnoy, Besnik Fetahu, and Amir Ingber. 2022. SDR: Efficient Neural Re-ranking using Succinct Document Representation. In Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 6624–6637, Dublin, Ireland. Association for Computational Linguistics.
Cite (Informal):
SDR: Efficient Neural Re-ranking using Succinct Document Representation (Cohen et al., ACL 2022)
Copy Citation:
PDF:
https://aclanthology.org/2022.acl-long.457.pdf
Video:
 https://aclanthology.org/2022.acl-long.457.mp4
Data
MS MARCO