Search query classification, as an effective way to understand user intents, is of great importance in real-world online ads systems. To ensure a lower latency, a shallow model (e.g. FastText) is widely used for efficient online inference. However, the representation ability of the FastText model is insufficient, resulting in poor classification performance, especially on some low-frequency queries and tailed categories. Using a deeper and more complex model (e.g. BERT) is an effective solution, but it will cause a higher online inference latency and more expensive computing costs. Thus, how to juggle both inference efficiency and classification performance is obviously of great practical importance. To overcome this challenge, in this paper, we propose knowledge condensation (KC), a simple yet effective knowledge distillation framework to boost the classification performance of the online FastText model under strict low latency constraints. Specifically, we propose to train an offline BERT model to retrieve more potentially relevant data. Benefiting from its powerful semantic representation, more relevant labels not exposed in the historical data will be added into the training set for better FastText model training. Moreover, a novel distribution-diverse multi-expert learning strategy is proposed to further improve the mining ability of relevant data. By training multiple BERT models from different data distributions, it can respectively perform better at high, middle, and low-frequency search queries. The model ensemble from multi-distribution makes its retrieval ability more powerful. We have deployed two versions of this framework in JD search, and both offline experiments and online A/B testing from multiple datasets have validated the effectiveness of the proposed approach.
翻译:搜索查询分类作为理解用户意图的有效手段,在真实在线广告系统中具有至关重要的意义。为确保低延迟,浅层模型(如FastText)被广泛用于高效的在线推理。然而,FastText模型的表示能力不足,导致分类性能较差,尤其在处理低频查询和长尾类别时表现不佳。采用更深层、更复杂的模型(如BERT)是一种有效解决方案,但这将导致更高的在线推理延迟和更昂贵的计算成本。因此,如何兼顾推理效率与分类性能显然具有重要的实际价值。为应对这一挑战,本文提出知识凝练(KC)——一种简单而有效的知识蒸馏框架,旨在严格低延迟约束下提升在线FastText模型的分类性能。具体而言,我们提出训练离线BERT模型以检索更多潜在相关数据。得益于其强大的语义表示能力,历史数据中未暴露的更多相关标签将被添加至训练集,从而优化FastText模型训练。此外,我们进一步提出一种新颖的分布多样化多专家学习策略,以增强相关数据挖掘能力。通过从不同数据分布训练多个BERT模型,该方法能分别在高、中、低频搜索查询上表现更优。多分布模型集成使其检索能力更为强大。我们在京东搜索中部署了该框架的两个版本,多数据集的离线实验与在线A/B测试均已验证所提方法的有效性。