Two Heads Are Better than One: Distilling Large Language Model Features into Small Models with Feature Decomposition and Mixture

Read full text
Authors: Association for Artificial Intelligence 2026, Xixin Cao, Jian Cao, Jue Chen, Bowen Deng, Tianhao Fu, Ruilong Ren, Xinxin Xu, Weichen Xu, Xinyu Zhao
Cited by: 0

Abstract

Market making (MM) through Reinforcement Learning (RL) has attracted significant attention in financial trading. With the development of Large Language Models (LLMs), more and more attempts are being made to apply LLMs to financial areas. A simple, direct application of LLM as an agent shows significant performance. Such methods are hindered by their slow inference speed, while most of the current research has not studied LLM distillation for this specific task. To address this, we first propose the normalized fluorescent probe to study the mechanism of the LLM’s feature. Based on the observation found by our investigation, we propose Cooperative Market Making(CMM), a novel framework that decouples LLM features across three orthogonal dimensions: layer, task, and data. Various student models collaboratively learn simple LLM features along with different dimensions, with each model responsible for a distinct feature to achieve knowledge distillation. Furthermore, CMM introduces an H´ajek-MoE to integrate the output of the student models by investigating the contribution of different models in a kernel function-generated common feature space. Extensive experimental results on four real-world market datasets demonstrate the superiority of CMM over the current distillation method and RL-based market-making strategies. The code is provided in the supplementary material.

Fields of Study & Subjects

Stock Market Forecasting Methods Financial Markets and Investment Strategies Topic Modeling

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN