Sign up to save tools and stay up to date with the latest in AI
bg
bg
1

A simulation of me: fine-tuning an LLM on 240k text messages - Edward Donner

Jan 04, 2024 - edwarddonner.com
The author of the article experimented with fine-tuning a language model (LLM) using his personal SMS and WhatsApp history to create a simulation of himself. He used a utility called iMazing to download all his conversations, filtered out group chats and rarely messaged contacts, and prepared the dataset by grouping messages into chunks. He then fine-tuned the model using QLoRA on V100 VMs and made improvements to the training data and prompt. The author used Hugging Face’s Text Generation to write conversations, playing the role of himself or his contacts.

Despite initial disappointments, the author found that the LLM became highly effective at imitating him and his friends, with the conversations seeming very real and reflecting the nuances of his different relationships. The author plans to write a series of posts on his approach and is excited to try other techniques to give the model more context on its conversations. He also plans to try other base models, further refinement of the prompt and input data, and more work on the generation.

Key takeaways:

  • The author experimented with fine-tuning a Language Model (LLM) on their SMS and WhatsApp history to create a simulation of themselves, with mixed results.
  • They used a utility called iMazing to download all their SMS/iMessage and WhatsApp conversations, filtered out group chats and people not in their contacts, and people they rarely message, yielding 240,805 messages with 288 people.
  • After several iterations and adjustments to the input data format and the generation approach, the LLM became highly effective at imitating the author and convincingly acting as many of their friends.
  • The author plans to continue refining the model and is excited to try other techniques to give the model more context on its conversations, with the ultimate goal of the model fully replacing them in replying to all their text messages.
View Full Article

Comments (0)

Be the first to comment!