Adam: Dense Retrieval Distillation with Adaptive Dark Examples

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tao, Chongyang, Liu, Chang, Shen, Tao, Xu, Can, Geng, Xiubo, Jiao, Binxing, Jiang, Daxin
Format: Preprint
Veröffentlicht: 2022
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910473361817600
author Tao, Chongyang
Liu, Chang
Shen, Tao
Xu, Can
Geng, Xiubo
Jiao, Binxing
Jiang, Daxin
author_facet Tao, Chongyang
Liu, Chang
Shen, Tao
Xu, Can
Geng, Xiubo
Jiao, Binxing
Jiang, Daxin
contents To improve the performance of the dual-encoder retriever, one effective approach is knowledge distillation from the cross-encoder ranker. Existing works construct the candidate passages following the supervised learning setting where a query is paired with a positive passage and a batch of negatives. However, through empirical observation, we find that even the hard negatives from advanced methods are still too trivial for the teacher to distinguish, preventing the teacher from transferring abundant dark knowledge to the student through its soft label. To alleviate this issue, we propose ADAM, a knowledge distillation framework that can better transfer the dark knowledge held in the teacher with Adaptive Dark exAMples. Different from previous works that only rely on one positive and hard negatives as candidate passages, we create dark examples that all have moderate relevance to the query through mixing-up and masking in discrete space. Furthermore, as the quality of knowledge held in different training instances varies as measured by the teacher's confidence score, we propose a self-paced distillation strategy that adaptively concentrates on a subset of high-quality instances to conduct our dark-example-based knowledge distillation to help the student learn better. We conduct experiments on two widely-used benchmarks and verify the effectiveness of our method.
format Preprint
id arxiv_https___arxiv_org_abs_2212_10192
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Adam: Dense Retrieval Distillation with Adaptive Dark Examples
Tao, Chongyang
Liu, Chang
Shen, Tao
Xu, Can
Geng, Xiubo
Jiao, Binxing
Jiang, Daxin
Computation and Language
To improve the performance of the dual-encoder retriever, one effective approach is knowledge distillation from the cross-encoder ranker. Existing works construct the candidate passages following the supervised learning setting where a query is paired with a positive passage and a batch of negatives. However, through empirical observation, we find that even the hard negatives from advanced methods are still too trivial for the teacher to distinguish, preventing the teacher from transferring abundant dark knowledge to the student through its soft label. To alleviate this issue, we propose ADAM, a knowledge distillation framework that can better transfer the dark knowledge held in the teacher with Adaptive Dark exAMples. Different from previous works that only rely on one positive and hard negatives as candidate passages, we create dark examples that all have moderate relevance to the query through mixing-up and masking in discrete space. Furthermore, as the quality of knowledge held in different training instances varies as measured by the teacher's confidence score, we propose a self-paced distillation strategy that adaptively concentrates on a subset of high-quality instances to conduct our dark-example-based knowledge distillation to help the student learn better. We conduct experiments on two widely-used benchmarks and verify the effectiveness of our method.
title Adam: Dense Retrieval Distillation with Adaptive Dark Examples
topic Computation and Language
url https://arxiv.org/abs/2212.10192