How to Make Cross Encoder a Good Teacher for Efficient Image-Text Retrieval?
文献类型:会议论文
作者 | chen yuxin1,4,5![]() ![]() ![]() ![]() ![]() |
出版日期 | 2024-06 |
会议日期 | 2024-6 |
会议地点 | 美国西雅图 |
英文摘要 | Dominant dual-encoder models enable efficient imagetext retrieval but suffer from limited accuracy, while the cross-encoder models offer higher accuracy at the expense of efficiency. Distilling cross-modality matching knowledge from cross-encoder to dual-encoder provides a natural approach to harness their strengths. Thus, we investigate the following valuable question: how to make crossencoder a good teacher for dual-encoder? Our findings are threefold: (1) Cross-modal similarity score distribution of cross-encoder is more concentrated, while the result of dual-encoder is nearly normal, making vanilla logit distillation less effective. However, ranking distillation remains practical, as it is not affected by the score distribution. (2) Only the relative order between hard negatives conveys valid knowledge, while the order information between easy negatives has little significance. (3) Maintaining the coordination between distillation loss and dual-encoder training loss is beneficial for knowledge transfer. Based on these findings, we propose a novel Contrastive Partial Ranking Distillation (CPRD) method, which implements the objective of mimicking relative order between hard negative samples with contrastive learning. This approach coordinates with the training of the dual-encoder, effectively transferring valid knowledge from the cross-encoder to the dualencoder. Extensive experiments on image-text retrieval and ranking tasks show that our method surpasses other distillation methods and significantly improves the accuracy of dual-encoder. |
源URL | [http://ir.ia.ac.cn/handle/173211/57582] ![]() |
专题 | 自动化研究所_模式识别国家重点实验室_视频内容安全团队 |
通讯作者 | yuan chunfeng |
作者单位 | 1.School of Artificial Intelligence, University of Chinese Academy of Sciences 2.The University of Hong Kong 3.School of Information Science and Technology, ShanghaiTech University 4.ARC Lab, Tencent PCG 5.State Key Laboratory of Multimodal Artificial Intelligence Systems, Institute of Automation, Chinese Academy of Sciences |
推荐引用方式 GB/T 7714 | chen yuxin,ma zongyang,zhang ziqi,et al. How to Make Cross Encoder a Good Teacher for Efficient Image-Text Retrieval?[C]. 见:. 美国西雅图. 2024-6. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。