Search papers, labs, and topics across Lattice.
TSUBASA enhances personalized LLMs (PLLMs) for long-horizon tasks by dynamically evolving memory writing and using self-learning with context distillation to improve memory reading. The dynamic memory evolution addresses the issue of existing memory mechanisms failing to capture evolving behaviors, while context distillation internalizes user experiences, mitigating the train-inference gap. Evaluations using the Qwen-3 model family (4B to 32B) demonstrate that TSUBASA outperforms memory-augmented systems like Mem0 and Memory-R1, achieving Pareto improvements in quality and efficiency.
Achieve robust, high-fidelity personalization with a reduced token budget by dynamically evolving memory and self-learning with context distillation.
Personalized large language models (PLLMs) have garnered significant attention for their ability to align outputs with individual's needs and preferences. However, they still struggle with long-horizon tasks, such as tracking a user's extensive history of conversations or activities. Existing memory mechanisms often fail to capture evolving behaviors, and RAG paradigms are trapped by a quality-efficiency tradeoff. Meanwhile, parametric adaptation is bottlenecked by train-inference gap due to the scarcity of labeled data. To enhance the long-horizon capabilities of PLLMs, we introduce TSUBASA, a two-pronged approach designed to improve memory writing via dynamic memory evolution, and memory reading via self-learning with a context distillation objective to internalize user experiences. Extensive evaluations on long-horizon benchmarks using the Qwen-3 model family (4B to 32B) validate the effectiveness of TSUBASA, surpassing competitive memory-augmented systems that rely primarily on memory writing, such as Mem0 and Memory-R1. Our analyses further confirms that TSUBASA breaks the quality-efficiency barrier to achieve Pareto improvements, delivering robust, high-fidelity personalization with a reduced token budget.
