Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Plug-in and Fine-tuning: Bridging the Gap between Small Language Models and Large Language Models

About

Large language models (LLMs) are renowned for their extensive linguistic knowledge and strong generalization capabilities, but their high computational demands make them unsuitable for resource-constrained environments. In contrast, small language models (SLMs) are computationally efficient but often lack the broad generalization capacity of LLMs. To bridge this gap, we propose PiFi, a novel framework that combines the strengths of both LLMs and SLMs to achieve high performance while maintaining efficiency. PiFi integrates a single frozen layer from an LLM into a SLM and fine-tunes the combined model for specific tasks, boosting performance without a significant increase in computational cost. We show that PiFi delivers consistent performance improvements across a range of natural language processing tasks, including both natural language understanding and generation. Moreover, our findings demonstrate PiFi's ability to effectively leverage LLM knowledge, enhancing generalization to unseen domains and facilitating the transfer of linguistic abilities.

Kyeonghyun Kim, Jinhee Jang, Juhwan Choi, Yoonji Lee, Kyohoon Jin, YoungBin Kim• 2025

Related benchmarks

TaskDatasetResultRank
Natural Language InferenceSNLI
Accuracy91.05
174
Question AnsweringSQuAD
F182.83
127
Text ClassificationIMDB
Accuracy89.83
107
Text ClassificationTweet Sentiment
F1 Score93.25
31
Natural Language InferenceMNLI
Accuracy87.98
22
General NLU PerformanceNLU Average
Average Accuracy0.874
10
Natural Language InferenceCOLA
Accuracy86.26
10
Text ClassificationSST-2
Accuracy95.01
10
Text ClassificationTweet Offensive
Accuracy86.03
10
Machine TranslationMulti30K (test)
BLEU54.13
4
Showing 10 of 10 rows

Other info

Code

Follow for update