In plain words: Instead of scoring every question pair with a language model, this groups question embeddings into a search tree and only checks the most promising branches for matches. It speeds prediction by 500% to 1000% versus a strong baseline, with only a small drop in ranking accuracy.
Abstract · Revisiting Semantic Representation and Tree Search for Similar Question Retrieval
This paper studies the performances of BERT combined with tree structure in short sentence ranking task. In retrieval-based question answering system, we retrieve the most similar question of the query question by ranking all the questions in datasets. If we want to rank all the sentences by neural rankers, we need to score all the sentence pairs. However it consumes large amount of time. So we design a specific tree for searching and combine deep model to solve this problem. We fine-tune BERT on the training data to get semantic vector or sentence embeddings on the test data. We use all the sentence embeddings of test data to build our tree based on k-means and do beam search at predicting time when given a sentence as query. We do the experiments on the semantic textual similarity dataset, Quora Question Pairs, and process the dataset for sentence ranking. Experimental results show that our methods outperform the strong baseline. Our tree accelerate the predicting speed by 500%-1000% without losing too much ranking accuracy.
Tong Guo, Huilin Gao
arXiv:1908.08326 · cs.CL · submitted Aug 22, 2019 · updated Sep 6, 2019
abstract · pdf · html