Preprint
Machine Learning

Persona Hub

Xin Chan, Xiaoyang Wang, Dian Yu, Haitao Mi, Dong Yu
June 1, 2024arXiv.org421 citations

421

Citations

35

Influential Citations

arXiv.org

Venue

2024

Year

Abstract

A persona-driven method for creating diverse synthetic data using LLMs, introducing Persona Hub, a collection of 1 billion automatically curated personas from web data. These personas act as diverse knowledge carriers, enabling the LLM to generate varied synthetic data (e.g., math problems, text, game NPCs) at scale.

Analysis

Why This Paper Matters

Large language models (LLMs) thrive on diverse, high-quality data, but collecting such data at scale is expensive and time-consuming. Synthetic data generation offers a promising alternative, yet existing methods often produce repetitive or narrow outputs. The Persona Hub paper tackles this by introducing a persona-driven approach that uses a massive collection of 1 billion personas to inject diversity into synthetic data. This is significant because it provides a systematic way to generate varied data across many domains, potentially reducing the need for manual curation and improving LLM performance on downstream tasks.

How It Works

The core idea is simple: by assigning a persona (e.g., a mathematician, a chef, a teenager) to an LLM during data generation, the model produces outputs that reflect that persona's perspective, knowledge, and style. To scale this, the authors built Persona Hub, a collection of 1 billion personas automatically curated from web data.

Figure 1

Building Persona Hub relies on two complementary methods:

  • Text-to-Persona: Given a piece of web text, an LLM infers the persona of the likely reader or writer. Since web data is vast and diverse, this yields a wide range of personas.

Figure 3

  • Persona-to-Persona: To reach personas not directly inferable from text, the method derives new personas based on interpersonal relationships (e.g., "Who is in close relationship with the given persona?"). This is repeated up to six iterations, leveraging the six degrees of separation theory.

Figure 5

After generating billions of personas, deduplication is performed using MinHash (1-gram, signature size 128, similarity threshold 0.9) and embedding-based methods (cosine similarity > 0.9) to ensure diversity. Heuristic filters remove low-quality descriptions, resulting in 1,015,863,523 personas.

Using Personas for Data Synthesis is flexible. The paper proposes three prompting methods:

  • Zero-shot: No examples given, maximizing creativity.
  • Few-shot: Provides demonstrations to guide output.
  • Persona-enhanced few-shot: Each demonstration is paired with its inferred persona, improving the LLM's ability to adopt the target persona.

0-shot, few-shot and persona-enhanced few-shot prompting methods.

Results

The paper demonstrates the approach across several use cases:

  • Math Problem Synthesis: Adding a persona (e.g., a math professor) leads to more challenging and contextually relevant problems.

Figure 7

  • Logical Reasoning: Personas enable creation of diverse reasoning problems, including Ruozhiba-style puzzles.

Figure 9

  • Instructions (User Prompts): Persona Hub simulates diverse user requests, improving instruction-following capabilities.

Figure 11

  • Knowledge-rich Texts: Personas help generate varied articles (e.g., Quora-style) for pre-training or post-training.

Figure 12

  • Game NPCs: Personas are projected into game worlds to create diverse non-player characters.

Figure 13

  • Tool Development: Personas predict user needs, enabling pre-building of tools that LLMs can call.

Figure 14

Significance

Persona Hub represents a practical, scalable solution for generating diverse synthetic data. By leveraging the vastness of web data and the power of LLMs, it reduces the manual effort needed for data curation. The approach is domain-agnostic and can be adapted to many tasks, from education to game design. However, it inherits biases from web data and relies on the LLM's ability to accurately model personas. Future work could explore mitigating biases, improving persona quality, and integrating with reinforcement learning from human feedback. Overall, this paper provides a valuable tool for the AI community to create richer, more varied training data at unprecedented scale.