Zhuofan Chen
2024
Leveraging Estimated Transferability Over Human Intuition for Model Selection in Text Ranking
Jun Bai
|
Zhuofan Chen
|
Zhenzi Li
|
Hanhua Hong
|
Jianfei Zhang
|
Chen Li
|
Chenghua Lin
|
Wenge Rong
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Text ranking has witnessed significant advancements, attributed to the utilization of dual-encoder enhanced by Pre-trained Language Models (PLMs). Given the proliferation of available PLMs, selecting the most effective one for a given dataset has become a non-trivial challenge. As a promising alternative to human intuition and brute-force fine-tuning, Transferability Estimation (TE) has emerged as an effective approach to model selection. However, current TE methods are primarily designed for classification tasks, and their estimated transferability may not align well with the objectives of text ranking. To address this challenge, we propose to compute the expected rank as transferability, explicitly reflecting the model’s ranking capability. Furthermore, to mitigate anisotropy and incorporate training dynamics, we adaptively scale isotropic sentence embeddings to yield an accurate expected rank score. Our resulting method, Adaptive Ranking Transferability (AiRTran), can effectively capture subtle differences between models. On challenging model selection scenarios across various text ranking datasets, it demonstrates significant improvements over previous classification-oriented TE methods, human intuition, and ChatGPT with minor time consumption.
Search
Co-authors
- Jun Bai 1
- Zhenzi Li 1
- Hanhua Hong 1
- Jianfei Zhang 1
- Chen Li 1
- show all...