Customizing Large Language Model Generation Style using PEFT


Results
One-size-fits-all large language models (LLMs) are increasingly being used to help people with their writing. However, the style these models are trained to write in may not suit all users or use cases. LLMs would be more useful as writing assistants if their idiolect could be customized to match each user. In this paper, we explore whether parameter-efficient finetuning (PEFT) with Low-Rank Adaptation can effectively guide the style of LLM generations. We use this method to customize LLaMA-2 to ten different authors. Our findings highlight the potential of PEFT to support efficient, user-level customization of LLMs.

Here are the results from our method compared with two baselines. Our method, StyleTunedLM, is shown in blue.