TreeBridge: Aligning LLM Embeddings in Industrial Recommender Systems

Authors

  • Yabo Ni Nanyang Technological University
  • Cao Yuanpeng Shopee Pte Ltd.
  • Wenhang Zhou Shopee Pte Ltd.
  • Bangyang Hong Shopee Pte Ltd.
  • Zhongyi Zhang Shopee Pte Ltd.
  • Enlei Cai Shopee Pte Ltd.
  • Kangle Wu Shopee Pte Ltd.
  • Anxiang Zeng Nanyang Technological University
  • Han Yu Nanyang Technological University
  • Xiaoxiao Li University of British Columbia Vector Institute

DOI:

https://doi.org/10.1609/aaai.v40i47.41478

Abstract

Large language models (LLMs) have shown great potential in enhancing search and recommender systems by providing rich semantic representations from unstructured texts. However, directly integrating LLM embeddings into industrial recommendation pipelines often results in subpar performance due to the semantic and distributional mismatch between pre-trained LLM features and domain-specific, feedback-driven representations. Existing approaches struggle to effectively align LLM embeddings with recommendation objectives, often facing challenges such as label misalignment or the potential loss of semantic diversity during fine-tuning. In this work, we present TreeBridge, a novel framework that introduces a structure-aware generative encoding tree to bridge the semantic gap between LLM embeddings and recommendation tasks. It preserves the external semantic richness of LLM embeddings, while learning label-informed structures that capture user preferences and interaction patterns. This enables the generation of task-adaptive representations without compromising embedding diversity. We further adopt an online-offline hybrid service paradigm to ensure low-latency real-world deployment. TreeBridge has been deployed on the Shopee e-commerce platform, one of the largest online shopping platforms in Southeast Asia serving hundreds of millions of users. Since its deployment in May 2025, it has helped the company achieve a commercially significant 1.55% relative improvement in gross merchandise volume (GMV). The deployment experience demonstrates the effectiveness, scalability, and significant commercial value of TreeBridge.

Published

2026-03-14

How to Cite

Ni, Y., Yuanpeng, C., Zhou, W., Hong, B., Zhang, Z., Cai, E., … Li, X. (2026). TreeBridge: Aligning LLM Embeddings in Industrial Recommender Systems. Proceedings of the AAAI Conference on Artificial Intelligence, 40(47), 40370–40377. https://doi.org/10.1609/aaai.v40i47.41478

Issue

Section

IAAI Technical Track on Emerging Applications of AI