Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TriAlign: Towards Universal Truth Consistency in Personalized LLM Alignment

About

Personalized large language models adapt responses to users' preferences and social attributes, but can introduce substantial universal truth inconsistencies across social groups, where some groups systematically receive less accurate responses on objective tasks. Existing alignment methods either ignore personalization or mainly focus on subjective preference alignment, largely overlooking fairness and consistency in universal truths. To address this gap, we study Truth-Invariant Alignment (TIA), an alignment problem for personalized LLMs that aims to ensure universal truths remain consistent across social groups while preserving personalization. We propose TriAlign, the first offline multi-agent reinforcement learning (MARL) framework for TIA, where each social group is modeled as an agent interacting. TriAlign jointly optimizes universal truth accuracy, cross-group truth consistency, and personalization through a fairness-aware objective and an explicit inconsistency penalty. Experiments across diverse benchmarks demonstrate that TriAlign achieves a stronger balance among these three objectives than strong baselines, reducing universal truth disparities across social groups while improving both objective task performance and personalization quality.

Thi-Nhung Nguyen, Linhao Luo, Rollin Omari, Junae Kim, Thuy-Trang Vu, Dinh Phung• 2026

Related benchmarks

TaskDatasetResultRank
Personalized Response GenerationAIME Implicit Preference 2025 (test)
Preference Score0.221
8
Personalized Response GenerationAIME Explicit Preference 2025 (test)
Pref69.9
8
Personalized Response GenerationStereoSet Implicit Preference (test)
Pref Score0.474
8
Personalized Response GenerationStereoSet Explicit Preference (test)
Preference Score79.1
8
Showing 4 of 4 rows

Other info

Follow for update