Direct Alignment with Heterogeneous Preferences

Ariel Procaccia (Harvard University) · Ali Shirali (University of California, Berkeley) · Arash Nasr-Esfahany (MIT) · Abdullah Alomar (Massachusetts Institute of Technology) · Parsa Mirtaheri (UC San Diego) · Rediet Abebe (Tübingen AI Center)
average rewardconsistent learningdirect alignment methodsdirect lossfirst-order improvementsfull feedbackheterogeneous preferenceshomogeneity assumptionhuman preferencesminimal informationoptimal policypolicy alignmentsample-efficientuniversal reward functionuser types

Alignment with human preferences is commonly framed using a universal reward function, even though human preferences are inherently heterogeneous. We formalize this heterogeneity by introducing user types and examine the limits of the homogeneity assumption. We show that aligning to heterogeneous preferences with a single policy is best achieved using the average reward across user types. However, this requires additional information about annotators. We examine improvements under different information settings, focusing on direct alignment methods. We find that minimal information can yield first-order improvements, while full feedback from each user type leads to consistent learning of the optimal policy. Surprisingly, however, no sample-efficient consistent direct loss exists in this latter setting. These results reveal a fundamental tension between consistency and sample efficiency in direct policy alignment.