DeAR: Dual-Stage Document Reranking with Reasoning Agents via LLM Distillation

#### Контекст Существующие системы рейтинга документов часто сталкиваются с проблемой достижения баланса между точным оцениванием точности каждого документа и обеспечением глобального анализа взаимодействия между документами. Это приводит к потере точности или неэффективности в обработке крупных наборов документов. Большие языковые модели (LLMs) стали новым стандартом в этом процессе, но их использование часто ограничивается предопределенными архитектурами и проблемами с масштабированием. Необходимо разработать гибкую и прозрачную модель, которая могла быть применена в различных сценариях, включая открытую доменную квиз-информацию и новые базы данных. #### Метод Мы предлагаем DeAR (Dual-Stage Agent Rerank) — распределенную архитектуру с двух этапами, каждый из которых решает свой собственный класс задач. **Стадия 1** (Dual-Loss Distillation) заключается в том, чтобы использовать модель LLaMA 13B в качестве teacher-модели для обучения модели DeAR-Student 3B или 8B с помощью комбинированных целей точного классификатора (cross-entropy), классификации по рейтингу (RankNet) и адаптации вероятности (KL-divergence). **Стадия 2** (List-Wise Reasoning with Chain-of-Thought) включает в себя LoRA-адаптеры и тренировку модели на 20K GPT-4 ответов с разруливанием последовательности (chain-of-thought), что позволяет системе обеспечить логическую интерпретацию рейтингов документов. #### Результаты Мы проверили работу DeAR на 11 задаче библиографического поиска, включая TREC-DL 2019/20, BEIR-датасеты и NovelEval-2306. На TREC-DL 2020 DeAR превысила базовые модели на +5.1 nDCG@5. На NovelEval-2306 она достигла 90.97 nDCG@10, превосходя даже GPT-4 на +3.09. Также DeAR показала отличные результаты в открытом доменном задании на Natural Questions, с Top-1 точностью 54.29%, превосходя MonoT5 и RankGPT. Наблюдения показали, что стратегия двухуровневого распределения обучения (двух loss-функций) обеспечивает стабильность модели и повышает точность в поиске. #### Значимость DeAR может применяться в сценариях открытого доменного поиска, в том числе в системах принятия решений, открытой квиз-информации и открытого поиска в Интернете. Основные преимущества: (1) высокая точность благодаря двухэтапному распределенному рассуждению; (2) легкость интеграции в существующие системы; (3) прозрачность решений за счет цепочек рассуждений. Это делает ее привлекательной для реальных систем, где прозрачность и эффективность ключевые. #### Выводы Мы продемонстрировали, что DeAR предлагает эффективный и прозрачный подход к рейтинговой системе. Мы планируем продолжить работу по улучшению точности и масштабиро