Wen-Yun Yang


2025

pdf bib
pEBR: A Probabilistic Approach to Embedding Based Retrieval
Han Zhang | Yunjiang Jiang | Mingming Li | Haowei Yuan | Yiming Qiu | Wen-Yun Yang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track

Embedding-based retrieval aims to learn a shared semantic representation space for both queries and items, enabling efficient and effective item retrieval through approximate nearest neighbor (ANN) algorithms. In current industrial practice, retrieval systems typically retrieve a fixed number of items for each query. However, this fixed-size retrieval often results in insufficient recall for head queries and low precision for tail queries. This limitation largely stems from the dominance of frequentist approaches in loss function design, which fail to address this challenge in industry. In this paper, we propose a novel probabilistic Embedding-Based Retrieval (pEBR) framework. Our method models the item distribution conditioned on each query, enabling the use of a dynamic cosine similarity threshold derived from the cumulative distribution function (CDF) of the probabilistic model. Experimental results demonstrate that pEBR significantly improves both retrieval precision and recall. Furthermore, ablation studies reveal that the probabilistic formulation effectively captures the inherent differences between head-to-tail queries.

2009

pdf bib
A Structural Support Vector Method for Extracting Contexts and Answers of Questions from Online Forums
Wen-Yun Yang | Yunbo Cao | Chin-Yew Lin
Proceedings of the 2009 Conference on Empirical Methods in Natural Language Processing