Artificial Intelligence1 min read

Speeding up language models for travel applications

By · Published by Everything Blog

In short

The researchers created their pipeline by coupling knowledge distillation with Medusa, an inference framework capable of generating multiple tokens in parallel. In trials, the team used the 70-billion-parameter Llama-3.1 model as a teacher to transfer capabilities directly into TinyLlama, a 1.1-billion-parameter student model outfitted with prediction heads. This architecture was assessed across critical travel tools, including real-time filter conversions, conversational trip planning, and bulk property description generation. The breakthrough earned the Industry Track best-paper award at the Association for Computational Linguistics meeting.

Key points

  • Knowledge distillation paired with parallel token generation preserves response quality w…: Knowledge distillation paired with parallel token generation preserves response quality while drastically reducing hardware deployment costs.
  • Sub-second processing safeguards customer satisfaction on platforms where delays as short…: Sub-second processing safeguards customer satisfaction on platforms where delays as short as 100 milliseconds hurt business.
  • Compact models equipped with Medusa heads can efficiently handle large-scale tasks like g…: Compact models equipped with Medusa heads can efficiently handle large-scale tasks like generating millions of listing descriptions.

Sources checked by our analysis pipeline: 1

Travel platforms serve millions of daily users by answering questions about destinations, offering hotel recommendations, and helping users plan their trips. In recent years, many platforms have turned to large language models (LLMs) to improve these services.

The LLMs give accurate, contextually relevant travel advice thanks to their size: they typically have tens of billions of parameters. Unfortunately, such large models are also computationally expensive, and that expense scales with the number of customer queries. What’s more, such large models can be too slow for web and mobile applications, where a delay of even 100 milliseconds can cost a company customers and revenue.

To address these scaling and latency challenges, we explored two complementary optimization techniques. One is a compression technique known as knowledge distillation (KD), in which a slow but powerful model produces training targets for a faster, lightweight model. The other, called Medusa, is a framework that transforms sequential next-token predictions — a common bottleneck in LLMs — into parallel computations.

Working with leading online travel agency Booking.com, we applied KD and Medusa techniques to the LLMs that…

Artificial Intelligence