Soft Tokens, Hard Truths

2509.19170v2 cs.CL, cs.AI, cs.LG 2025-09-25
Авторы:

Natasha Butt, Ariel Kwiatkowski, Ismail Labiad, Julia Kempe, Yann Ollivier

Резюме на русском

## Контекст Оптимизация алгоритмов глубокого обучения, особенно в контексте трансформеров и ло LLaMA, ЛлаМа, является ключевым направлением в современной интеллектуальной системе. Акцент в работах по глубокому обучению часто делается на том, как модели могут более эффективно обрабатывать информацию, а также на методах обучения, которые позволяют им решать более сложные задачи. Одной из проблем в этой области является то, что современные модели часто оперируют с дискретными токенами, что ограничивает их гибкость и выразительность. Это делает необходимым исследование способов внедрения непрерывных токенов, которые могут более эффективно применяться в процессе решения задач, достигая более высокого уровня точности и гибкости. ## Метод В этой работе предлагается метод обучения моделей с использованием непрерывных токенов, используя технологию реинфорсмент (RL). Метод учитывает необходимость минимизировать компьютерные затраты и оптимизировать процесс обучения. Из статьи становится ясно, что модели обучаются на основе "мягких" токенов, которые представляют собой смешение дискретных токенов со зерном шума. Это позволяет модели более эффективно исследовать различные варианты решения задачи. Важно отметить, что процесс обучения не требует дистилляции от дискретных токенов, что значительно сокращает вычислительные затраты и быстродействие. В результате модели могут обрабатывать большое количество токенов, что дает новые возможности для решения задач, в том числе и математических решений. ## Результаты Эксперименты проводились на различных моделях глубокого обучения, включая ЛлаМа и Qwen, с размерами 8 миллиардов параметров. Модели обучались с использованием непрерывных токенов, а затем проверялись на бенчмарках математических задач. Результаты показали, что модели с непрерывными токенами не только совпадают с дискретными-токеновыми моделями по показателю pass@1, но и превосходят их в pass@32. Это демонстрирует, что непрерывные токены дают большую выразительность и улучшают разнообразие решения задач. Кроме того, модели, обученные с непрерывными токенами, лучше сохраняют выдачу базовой модели на неотношенных данных, что делает их более универсальными. ## Значимость Метод, предложенный в работе, может быть применен в различных областях, где требуется высокая точность и гибкость решения задач. Он может применяться в областях, таких как робототехника, системы рекомендаций, искусственный интеллект в сфере финансов и медицины. Важно отметить, что технология не только улучшает качество решения задач, но и позволяет экономить вычислительные ресурсы, что отлично подходи

Abstract

The use of continuous instead of discrete tokens during the Chain-of-Thought (CoT) phase of reasoning LLMs has garnered attention recently, based on the intuition that a continuous mixture of discrete tokens could simulate a superposition of several reasoning paths simultaneously. Theoretical results have formally proven that continuous tokens have much greater expressivity and can solve specific problems more efficiently. However, practical use of continuous tokens has been limited by strong training difficulties: previous works either just use continuous tokens at inference time on a pre-trained discrete-token model, or must distill the continuous CoT from ground-truth discrete CoTs and face computational costs that limit the CoT to very few tokens. This is the first work introducing a scalable method to learn continuous CoTs via reinforcement learning (RL), without distilling from reference discrete CoTs. We use "soft" tokens: mixtures of tokens together with noise on the input embedding to provide RL exploration. Computational overhead is minimal, enabling us to learn continuous CoTs with hundreds of tokens. On math reasoning benchmarks with Llama and Qwen models up to 8B, training with continuous CoTs match discrete-token CoTs for pass@1 and surpass them for pass@32, showing greater CoT diversity. In systematic comparisons, the best-performing scenario is to train with continuous CoT tokens then use discrete tokens for inference, meaning the "soft" models can be deployed in a standard way. Finally, we show continuous CoT RL training better preserves the predictions of the base model on out-of-domain tasks, thus providing a softer touch to the base model.

Ссылки и действия