Gyu-Hwung Cho
2024
RRADistill: Distilling LLMs’ Passage Ranking Ability for Long-Tail Queries Document Re-Ranking on a Search Engine
Nayoung Choi
|
Youngjune Lee
|
Gyu-Hwung Cho
|
Haeyu Jeong
|
Jungmin Kong
|
Saehun Kim
|
Keunchan Park
|
Sarah Cho
|
Inchang Jeong
|
Gyohee Nam
|
Sunghoon Han
|
Wonil Yang
|
Jaeho Choi
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing: Industry Track
Large Language Models (LLMs) excel at understanding the semantic relationships between queries and documents, even with lengthy and complex long-tail queries. These queries are challenging for feedback-based rankings due to sparse user engagement and limited feedback, making LLMs’ ranking ability highly valuable. However, the large size and slow inference of LLMs necessitate the development of smaller, more efficient models (sLLMs). Recently, integrating ranking label generation into distillation techniques has become crucial, but existing methods underutilize LLMs’ capabilities and are cumbersome. Our research, RRADistill: Re-Ranking Ability Distillation, propose an efficient label generation pipeline and novel sLLM training methods for both encoder and decoder models. We introduce an encoder-based method using a Term Control Layer to capture term matching signals and a decoder-based model with a ranking layer for enhanced understanding. A/B testing on a Korean-based search platform, validates the effectiveness of our approach in improving re-ranking for long-tail queries.
SLM as Guardian: Pioneering AI Safety with Small Language Model
Ohjoon Kwon
|
Donghyeon Jeon
|
Nayoung Choi
|
Gyu-Hwung Cho
|
Hwiyeol Jo
|
Changbong Kim
|
Hyunwoo Lee
|
Inho Kang
|
Sun Kim
|
Taiwoo Park
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing: Industry Track
Most prior safety research of large language models (LLMs) has focused on enhancing the alignment of LLMs to better suit the safety requirements of their use cases. However, internalizing such safeguard features into larger models brought challenges of higher training cost and unintended degradation of helpfulness. In this paper, we leverage a smaller LLM for both harmful query detection and safeguard response generation. We introduce our safety requirements and the taxonomy of harmfulness categories, and then propose a multi-task learning mechanism fusing the two tasks into a single model. We demonstrate the effectiveness of our approach, providing on par or surpassing harmful query detection and safeguard response performance compared to the publicly available LLMs.
Search
Co-authors
- Nayoung Choi 2
- Youngjune Lee 1
- Haeyu Jeong 1
- Jungmin Kong 1
- Saehun Kim 1
- show all...