Aligning to Thousands of Preferences via System Message Generalization

Although humans inherently have diverse values, current large language model (LLM) alignment methods often assume that aligning LLMs with the general public's preferences is optimal. A major challenge in adopting a more individualized approach to LLM alignment is its lack of scalability, as it involves repeatedly acquiring preference data and training new reward models and LLMs for each individual's preferences. To address these challenges, we propose a new paradigm where users specify what they value most within the system message, steering the LLM's generation behavior to better align with the user's intentions. However, a naive application of such an approach is non-trivial since LLMs are typically trained on a uniform system message (e.g.,"You are a helpful assistant") which limits their ability to generalize to diverse, unseen system messages. To improve this generalization, we create the Multifaceted Collection, a preference dataset with 192k combinations of values beyond generic helpfulness and harmlessness, spanning 65k user instructions. Using this dataset, we train a 7B LLM called Janus and test it on 921 prompts from 5 benchmarks (AlpacaEval 2.0, FLASK, Koala, MT-Bench, and Self-Instruct) by adding various unseen system messages that reflect user preferences. Janus achieves tie+win rate of 75.2%, 72.4%, and 66.4% against Mistral 7B Instruct v0.2, GPT-3.5 Turbo, and GPT-4, respectively. Unexpectedly, on three benchmarks focused on response helpfulness (AlpacaEval 2.0, MT-Bench, Arena Hard Auto v0.1), Janus also outperforms LLaMA 3 8B Instruct by a +4.0%, +0.1%, +3.0% margin, underscoring that training with a vast array of system messages could also enhance alignment to the general public's preference as well. Our code, dataset, benchmark, and models are available at https://github.com/kaistAI/Janus.

Paper

Similar papers

Peer review

Reviewer bEUV6/10 · confidence 5/52024-06-30

Summary

In this paper, the authors introduce a novel method to align LLMs with diverse user preferences without requiring continual retraining for each specific preference. The approach utilizes a unique system message protocol that guides LLMs to produce responses tailored to specific, nuanced user preferences.

Strengths

1. The idea is novel: adapting to different users' preferences by training on preference data with different system prompts. 2. It is intuitively a useful research, effectively strengthening the alignment effectiveness by clarifying the user identities besides the alignment data themselves. 3. The paper is well-written and easy to follow.

Weaknesses

1. How many humans did you hired to perform the human evaluation? There lacks a illustration on this. If we want to illustrate its ability to cater to a wide range of peoples with different backgrounds, then we should also apply a wide range of people to evaluate. 2. I feel that there lacks an illustration to show that the model is not becoming verbose. As all of the benchmarks you use prefer verbose responses.

Questions

Is this a way to "hack" the Chatbot Arena? (in the right way lol)

Rating

6

Confidence

5

Soundness

2

Presentation

2

Contribution

3

Limitations

Yes

Authorsrebuttal2024-08-12

Dear reviewer bEUV, thank you for the clarification! Our training dataset and method does improve the model in preference-based benchmarks like AlpacaEval 2.0 (Table 3). However, our method is not centered on how to extract user preferences from the human population. Instead, we created training data simulating user preferences by distilling GPT-4 Turbo’s knowledge, which itself was effective in achieving both personalization and general helpfulness. Also, since Chatbot Arena is updated via *real-time user votes*, we do not see it feasible to fit the arena users’ preferences as well. We emphasize that our method cannot be used to mine or fit existing users’ preferences. The intended use of our method is to align to diverse individuals’ preferences and generalize to unseen system messages. We hope this answers your question!

Reviewer k2dN5/10 · confidence 4/52024-07-05

Summary

This paper addresses the issue that humans inherently have diverse values, while current LLMs are primarily aligned with general public preferences such as helpfulness and harmlessness. Previous work has trained new reward models (RMs) and LLMs for individual preferences, which is time-consuming and costly. The authors propose a new paradigm where users specify their values within system messages. Specifically, they create the Multifaceted Collection, containing 65k user instructions, each with 3 system messages, resulting in 192k combinations of fine-grained values with the assistance of GPT-4. They then train mistra-7b-v0.2 into Janus-7B and test it on 921 prompts collected from 5 benchmarks. Compared to leading open-sourced and proprietary models using pairwise ranking human evaluation or GPT-4-as-a-judge, Janus-7B achieves a higher win rate while maintaining harmlessness.

Strengths

The paper proposes a new paradigm to address the problem of aligning to fine-grained values without repeatedly training multiple RMs and LLMs. It also creates a large-scale preference dataset with fine-grained system prompts representing multiple alignment targets. Detailed experiments across many leading open-source and proprietary models validate the method's effectiveness.

Weaknesses

While the paper is well-written and presents a solid analysis, several weaknesses need addressing: 1. The construction of the Multifaceted Collection is costly and not scalable. GPT-4 generates preference sets for each of the 65k instructions, converts each of the 198k preference sets into a system message, and crafts gold-standard multifaceted responses for each system message. 2. The role of system message generalization has not been clarified. What are the differences between varying the system messages for individualized alignment targets and simply using different instructions in the user prompt while maintaining the system message unchanged to reach the specified alignment targets?

Questions

1. What are the differences between varying the system messages for individualized alignment targets and simply using different instructions in the user prompt while maintaining the system message unchanged to reach the specified alignment targets? It seems that I can just use specialized instructions in user prompts to reach the same goal of aligning to individualized values. Can you comment on this? 2. There are several typos: - Line 136: 5 datasets, not 4 - Line 152: Appendix B.1? - Line 286: Appendix G.1? And so on

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

2

Limitations

The construction of the Multifaceted Collection is costly and not scalable. And the role of system message generalization is unclear. It is not explained how varying system messages for individualized alignment targets differs from using different instructions in user prompts while keeping the system message unchanged.

Authorsrebuttal2024-08-12

Author Response to Reviewer k2dN (1/2)

Dear reviewer k2dN, thank you for raising the discussion. We took your concerns carefully and will provide detailed comments below. First, as per your main concern on our method’s similarity with instruction tuning, we agree with the following statement: > aligning with varying preferences through system message generalization seems somewhat similar to using diversified instructions in instruction tuning, aiming to enable generalization to other instructions. System message itself is an instruction that the model should follow, so the basis of system message generalization is indeed in instruction tuning. With sufficiently many diversified system messages, our method takes advantage of the instruction tuning recipe to generalize to unseen system messages (i.e., individualized alignment targets). However, our focus on the *content* differs from previous instruction tuning works. While previous works scale inputs, input-free tasks, or tasks per input (three paradigms described in [1]) in instructions, **we aim to scale *meta-instructions*, instructions that guide how to respond to subsequent instructions**. Our motivation was that different people expect different responses on the same instruction, so such meta-instructions that set preferences for task execution should also be taught to the model. Including this in instruction tuning helps the model approach various types of prompts and dynamically adjust response strategies, as shown in Section 5.1 and 6.2. Method-wise, our hierarchical data generation strategy enables careful curation of those meta-instructions to specifically simulate user preferences instead of presenting irrelevant challenges or hinting solutions with respect to the instruction. To the best of our knowledge, **there is no work that shares a similar motivation with us or has publicly released a training dataset containing meta-instructions**. --- [1] Lou et al.. MUFFIN: Curating Multi-Faceted Instructions for Improving Instruction Following. ICLR 2024.

Authorsrebuttal2024-08-12

Author Response to Reviewer k2dN (2/2)

Lastly, we would like to elaborate on why we have set meta-instructions in the system message and how to further utilize them. While some meta-instructions can only be defined by a specific user message, it can also be more general. For example, the following system message in our dataset is associated with a user prompt giving a specific hyperbola equation, but is also applicable to other kinds of problems. ``` You are a mathematics coach, specializing in presenting complex mathematical concepts to users with an intermediate understanding. Your teaching method is known for its succinctness, offering clear, direct explanations and step-by-step walkthroughs without overcomplicating the information. To facilitate learning, incorporate visual aids like charts and diagrams, making abstract concepts tangible and easier to understand. Your approach always considers the users' math anxiety, striving to create an engaging, supportive environment that reduces stress and fosters confidence. Your aim is to make math accessible and less intimidating, ensuring users can grasp the essentials of mathematical theories quickly and effectively, with patience and creativity at the core of every interaction. ``` In this sense, **meta-instructions can be effective when set programmatically on relevant user messages in the form of system messages**. This (i) allows reflecting the users’ common needs on various instructions and (ii) lowers user burden. In practice, OpenAI has already shipped similar features like custom instructions [2] and memory controls [3] in ChatGPT; a user can specify their preferences in the custom instruction by oneself (e.g., *When I ask you for code, please just give me the code without any explanation on how it works.*) or ChatGPT will memorize the details itself from the conversations. When user preferences are gathered or summarized from interactions, it can be decided to be included as part of the system message without user specification for seamless chat experience. Exactly how to infer user preferences or how to manage user controls on system messages is application-dependent and beyond the scope of our work (noted in broader impact in Appendix I). In the line of works on dialogue summarization [4, 5], we look forward to future work to explore the challenges of inferring preferences from conversations. The focus of our work lies more in **representing user preferences in the form of meta-instructions** and **training models to follow meta-instructions**. Also, from a technical perspective, training meta-instructions in system messages with accompanying gold responses would make the model (with appropriate system tokens) better distinguish between meta-instructions and instructions and tailor responses to the higher-level guidance. We greatly value your feedback and will improve our writing to shape the storyline better. --- [2] OpenAI. Custom instructions for ChatGPT. OpenAI Blog 2023. [3] OpenAI. Memory and new controls for ChatGPT. OpenAI Blog 2024. [4] Zou et al.. Unsupervised Summarization for Chat Logs with Topic-Oriented Ranking and Context-Aware Auto-Encoders. AAAI 2021. [5] Zhao et al.. TODSum: Task-Oriented Dialogue Summarization with State Tracking. arXiv 2021.

Reviewer uZmQ6/10 · confidence 4/52024-07-09

Summary

This paper aims to align Large Language Models (LLMs) with individual user preferences at scale. The authors propose a paradigm where users specify their values within system messages to guide the LLM's generation behavior. To address the challenge of generalizing to diverse system messages, the authors create the MULTIFACETED COLLECTION, a dataset with 192k unique combinations of values and instructions. They train a 7B LLM called JANUS using this dataset and demonstrate its effectiveness in adhering to various user preferences without the need for retraining for each individual. The JANUS model outperforms other models like Mistral 7B Instruct v0.2, GPT-3.5 Turbo, and GPT-4 in benchmarks that assess response helpfulness.

Strengths

1. The creation of the MULTIFACETED COLLECTION, as well as the trained models, makes a good contribution to the community for studying diverse preference alignment. 2. The idea of utilizing system messages, while simple, is demonstrated as a scalable solution to the problem of individualized LLM alignment. 3. The paper shows that training with a diverse array of system messages not only supports personalized responses but also enhances alignment with general public preferences.

Weaknesses

While widely adopted, the GPT4-based data synthesis process may introduce some artifacts. The authors should include quantitative and qualitative analyses to investigate the potential biases and representativeness in human preferences in the collected MULTIFACETED COLLECTION.

Questions

How do you measure the diversity of user preferences in MULTIFACETED COLLECTION?

Rating

6

Confidence

4

Soundness

4

Presentation

3

Contribution

3

Limitations

The authors have discussed the limitations.

Reviewer eXeU5/10 · confidence 4/52024-07-17

Summary

This paper introduces JANUS, a novel approach to aligning large language models (LLMs) with diverse individual user preferences without retraining. The key contributions are the MULTIFACETED COLLECTION: A dataset of 192k diverse system messages reflecting varied user preferences, paired with 65k instructions. JANUS: A 7B parameter LLM trained on this dataset to generalize to unseen system messages at test time. Evaluation showing JANUS outperforms several larger models, including GPT-3.5 and GPT-4, in adhering to specified preferences.

Strengths

1. Using diverse system messages for alignment is novel and creative. It tackles the challenge of personalized alignment in a scalable way, avoiding the need to retrain models for each user. 2. The methodology is rigorous, with careful dataset construction and comprehensive evaluations against strong baselines. The use of human evaluation alongside automated metrics strengthens the results. 3. This paper is well-structured and clearly written. The approach and results are explained thoroughly, with helpful figures illustrating key concepts. It work has potentially high impact, addressing a crucial challenge in AI alignment. The ability to adapt to individual preferences without retraining could be transformative for deploying personalized AI systems at scale.

Weaknesses

1. The paper doesn't deeply examine potential risks of allowing such flexible preference specification, such as potential for misuse or unintended biases. - There's no discussion on how to handle conflicting or ethically questionable preferences. 2. While the overall approach is effective, it's unclear which components contribute most to the performance gains. For example, how much does the hierarchical nature of the preference augmentation contribute versus simply having a large number of diverse preferences? The impact of the number of unique system messages or instructions is not explored. 3. The study focuses on 7B parameter models, primarily comparing to other 7B models or larger. It's unclear how well the approach scales to smaller models or much larger models (e.g., 70B+). The comparison to LLaMA 3 8B Instruct is promising, but more exploration of scale effects would be valuable.

Questions

1. Have you explored any potential negative consequences of allowing such flexible specification of preferences through system messages? Are there safeguards to prevent misuse or handling of conflicting ethical preferences? 2. How does the performance of JANUS change as you vary the number of unique system messages in the training data? Is there a point of diminishing returns, and how does this relate to the hierarchical structure of your preference augmentation? 3. Have you tested the approach on models significantly larger or smaller than 7B parameters? How does the effectiveness of this method scale with model size, especially compared to the scaling behavior of traditional fine-tuning approaches? 4. How consistent is JANUS in maintaining the specified preferences across a long conversation or multiple diverse tasks? Have you evaluated this aspect, and if so, what metrics did you use? 5. The paper mentions outperforming larger models like GPT-3.5 and GPT-4. Have you analyzed why your approach seems to be more effective than simply scaling up model size for this task? Could this indicate a fundamental advantage of your method over scale alone? How did you ensure the quality and diversity of the generated system messages and reference answers in the MULTIFACETED COLLECTION? Did you implement any specific measures to mitigate potential biases introduced by using GPT-4-Turbo-0125 for generation?

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

None.

Area Chair DS992024-08-08

Dear reviewers, The rebuttal phase has now ended and the authors have submitted their responses to the reviews. In the coming days (August 7-13) there will be an open discussion between the reviewers and authors. Please read the responses, respond to them early on in the discussion, and discuss points of disagreement. Best, AC

Reviewer uZmQ2024-08-08

Thanks for your response. I have read the author response, and I maintain my rating.

Authorsrebuttal2024-08-09

Looking forward for Discussions

Dear Reviewers, We appreciate for your time and effort on reviewing the paper. As we near the end of the discussion period, we kindly remind you of the upcoming deadline. We are eager to discuss any aspects of the paper that may require further clarification. Thank you once again for your valuable feedback.

Reviewer k2dN2024-08-09

Thank you for your clear response. The role of system messages in individualized alignment can indeed be seen as a method to infer users' preferences. However, in practice, what specific system message would you use to better align with users' goals? Would it be based on past conversation history? If so, have you experimented with "developers inferring preferences (possibly based on past conversation history), verbalizing them in the system message"? Additionally, can system messages inferred from past conversation history be generalized using JANUS? My main concern is that aligning with varying preferences through system message generalization seems somewhat similar to using diversified instructions in instruction tuning, aiming to enable generalization to other instructions.

Reviewer bEUV2024-08-10

Hi authors, Thanks for your responses! To clarify the question: can this method be used as a way to improve the models' ranking on chatbot arena simply by fitting the human preferences or arena users' preferences?

Reviewer eXeU2024-08-12

Thanks the authors for their response. After reading the response, I think my current score is appropriate.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC