Training generative neural network systems using multiple reward models
AssigneeGDM Holding LLC
InventorsPragya Srivastava, Rahul Madhavan, Karthikeyan Shanmugam, Aravindan Raghuveer
Life of this patent
Estimating the term…
Loading patent…
AssigneeGDM Holding LLC
InventorsPragya Srivastava, Rahul Madhavan, Karthikeyan Shanmugam, Aravindan Raghuveer
Estimating the term…
Methods, systems, and computer storage media are provided for training generative neural network systems, such as Large Language Models (LLMs) or Vision-Language Models (VLMs), using multiple reward models. The process involves generating one or more output sequences from a training input sequence and processing these examples using a plurality of different reward models to generate respective reward values. These reward values are combined to obtain an aggregated reward, which may be calculated as a product of the values, a weighted geometric mean, or a Nash score. The generative neural network is trained using an objective function determined using the aggregated reward, such as a cross-entropy loss or a contrastive objective function, often utilizing log score differences between the system and a reference model. These techniques allow for multi-objective alignment and are adapted for implementation on parallel processing computer systems.
A computer-implemented method of training a generative neural network system, the generative neural network system configured to generate an output sequence by processing an input sequence using a sequence processing neural network, the method comprising:
One or more non-transitory computer storage media storing instructions that when executed by one or more computers cause the one or more computers to perform operations for training a generative neural network system, the generative neural network system configured to generate an output sequence by processing an input sequence using a sequence processing neural network, the operations comprising:
A system comprising:
Dependencies are inferred from the claim text.