about
Leveraging Human Revisions for Improving Text-to-Layout Models (arxiv.org)
2 points by PaulHoule on Jun 3, 2024 | hide | past | pdf | discuss on HN

In plain words: Expert designers fix layouts that a generator makes for mobile screens, and a scoring model learns from those fixes to judge layout quality. Training the generator with that score produces more modern, designer-aligned layouts than learning from simple preference labels.

Abstract

Learning from human feedback has shown success in aligning large, pretrained models with human values. Prior works have mostly focused on learning from high-level labels, such as preferences between pairs of model outputs. On the other hand, many domains could benefit from more involved, detailed feedback, such as revisions, explanations, and reasoning of human users. Our work proposes using nuanced feedback through the form of human revisions for stronger alignment. In this paper, we ask expert designers to fix layouts generated from a generative layout model that is pretrained on a large-scale dataset of mobile screens. Then, we train a reward model based on how human designers revise these generated layouts. With the learned reward model, we optimize our model with reinforcement learning from human feedback (RLHF). Our method, Revision-Aware Reward Models ($\method$), allows a generative text-to-layout model to produce more modern, designer-aligned layouts, showing the potential for utilizing human revisions and stronger forms of feedback in improving generative models.

Amber Xie, Chin-Yi Cheng, Forrest Huang, Yang Li
arXiv:2405.13026 · cs.CL, cs.AI · submitted May 16, 2024
abstract · pdf · html

add comment on HN