Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

PROD: Progressive Distillation for Dense Retrieval

About

Knowledge distillation is an effective way to transfer knowledge from a strong teacher to an efficient student model. Ideally, we expect the better the teacher is, the better the student. However, this expectation does not always come true. It is common that a better teacher model results in a bad student via distillation due to the nonnegligible gap between teacher and student. To bridge the gap, we propose PROD, a PROgressive Distillation method, for dense retrieval. PROD consists of a teacher progressive distillation and a data progressive distillation to gradually improve the student. We conduct extensive experiments on five widely-used benchmarks, MS MARCO Passage, TREC Passage 19, TREC Document 19, MS MARCO Document and Natural Questions, where PROD achieves the state-of-the-art within the distillation methods for dense retrieval. The code and models will be released.

Zhenghao Lin, Yeyun Gong, Xiao Liu, Hang Zhang, Chen Lin, Anlei Dong, Jian Jiao, Jingwen Lu, Daxin Jiang, Rangan Majumder, Nan Duan• 2022

Related benchmarks

TaskDatasetResultRank
Passage retrievalMsMARCO (dev)
MRR@1039.3
116
Passage RankingTREC DL 2019 (test)
NDCG@1073.3
33
Showing 2 of 2 rows

Other info

Follow for update