论文标题

专家与类别层次结构软约束的对抗性混合

Adversarial Mixture Of Experts with Category Hierarchy Soft Constraint

论文作者

Xiao, Zhuojian, jiang, Yunjiang, Tang, Guoyu, Liu, Lin, Xu, Sulong, Xiao, Yun, Yan, Weipeng

论文摘要

产品搜索是人们在电子商务网站上满足购物需求的最常见方法。产品通常用几个广泛的分类标签之一(例如“服装”或“电子产品”)以及诸如“冰箱”或“电视”(均在“电子产品”下”)进行注释。这些标签用于构建查询类别的层次结构。在查询类别中,价格和品牌知名度等功能的分布差异很大。此外,出于CTR/CVR预测的目的,特征重要性因一个类别而异。在这项工作中,我们利用专家(MOE)框架的混合物来学习专门针对每个查询类别的排名模型。特别是,我们的门网络仅依赖于从用户查询中提取的类别ID。虽然对于每个输入示例,我们会自发地自发地选择专家塔,但我们探索了两种技术,以在专家和查询类别之间建立更明确和透明的连接。为了帮助将专家在其领域专业方面进行区分,我们在专家产出中引入了一种对抗性正规化形式,迫使他们彼此不同意。结果,他们倾向于从不同的角度解决每个预测问题,而不是相互复制。通过不同类别的门输出向量的聚类效果更强,这可以验证这一点。此外,基于分类层次结构的软门控限制被施加以帮助类似产品选择相似的门值。并使他们更有可能分享类似的专家。这允许在较小的兄弟姐妹类别中汇总培训数据,以克服数据稀缺性。

Product search is the most common way for people to satisfy their shopping needs on e-commerce websites. Products are typically annotated with one of several broad categorical tags, such as "Clothing" or "Electronics", as well as finer-grained categories like "Refrigerator" or "TV", both under "Electronics". These tags are used to construct a hierarchy of query categories. Distributions of features such as price and brand popularity vary wildly across query categories. In addition, feature importance for the purpose of CTR/CVR predictions differs from one category to another. In this work, we leverage the Mixture of Expert (MoE) framework to learn a ranking model that specializes for each query category. In particular, our gate network relies solely on the category ids extracted from the user query. While classical MoE's pick expert towers spontaneously for each input example, we explore two techniques to establish more explicit and transparent connections between the experts and query categories. To help differentiate experts on their domain specialties, we introduce a form of adversarial regularization among the expert outputs, forcing them to disagree with one another. As a result, they tend to approach each prediction problem from different angles, rather than copying one another. This is validated by a much stronger clustering effect of the gate output vectors under different categories. In addition, soft gating constraints based on the categorical hierarchy are imposed to help similar products choose similar gate values. and make them more likely to share similar experts. This allows aggregation of training data among smaller sibling categories to overcome data scarcity.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源