Back to results

Massachusetts Institute of Technology

Choice Modeling and Assortment Optimization on the Transformer Model

Abstract

dc:description.abstract

The problem of modeling customer choices and finding assortments with maximal revenue has been widely studied in revenue management. Random utility models (RUMs) are typically used to model choice. These models implicitly enforce a rational decision making process whereby a customer is endowed with utilities for each product in the assortment and picks the product that maximizes her utility. This work seeks to explore a general class of choice models where the customer’s decision making process is not constrained in this fashion. To allow for departures from rational choice (and RUMs), we posit that the customer indirect utility associated with a product is a function of the assortment offered to her. Motivated by the success of transformer models in deep learning, we investigate the case where this utility function is defined through a trained transformer network. This leads to a new class of neural network-based discrete choice models, which we call transformer choice models. The universal approximation property of the transformer network ensures that our model can approximate any discrete choice model, and thus it can capture irrationalities in choice behavior. We perform computational experiments with real data to verify the generalization performance of our transformer choice model to unseen assortments. To ensure that our model does not overfit on the training data, we use dropout as the regularization method during training. We compare our model to both traditional choice models (the multinomial logit model and its synergistic variant that considers cross-product interaction) and machine learning-based choice models (decision forest choice model and feedforward neural network choice model) on two datasets: a large grocery panel dataset and an online hotel search dataset. We show that on both datasets, the transformer choice model has generalized well to unseen assortments with proper regularization. Moreover, on the more complex dataset of online hotel search, the transformer choice model has outperformed all other models in terms of out-of sample error. We finally consider the assortment optimization problem on transformer choice models. While the general assortment optimization problem is complex and in-tractable, we empirically evaluate and compare several heuristic algorithms, including random search, quadratic approximation, and local search. Our experiments on transformer choice models with real prices show that a simple local search heuristic finds the global optimum for the assortment optimization problem in three-fourths of the data categories, while achieving a good approximation on the rest of the categories. This shows that in practice, local search can be a reasonable heuristic for assortment optimization on transformer choice models.

Degree

thesis:*
Name thesis:degree_name
Master
Department dc:contributor.department
Massachusetts Institute of Technology. Operations Research Center
Grantor dc:publisher
Massachusetts Institute of Technology
Year dc:date.issued
2024

Author and committee

dc:creator, dc:contributor.*
Author dc:creator
  • Jiang, Qingxuan
Advisors dc:contributor.advisor
  • Levi, Retsef
  • Farias, Vivek

Rights

dc:rights
Statement dc:rights
  • In Copyright - Educational Use Permitted
  • Copyright retained by author(s)

Identifiers

dc:identifier.*
Handle dc:identifier.uri
https://hdl.handle.net/1721.1/153714
OAI identifier oai:identifier
oai:dspace.mit.edu:1721.1/153714

Chain of custody

source
Harvested from
MIT
Base URL
dspace.mit.edu/oai/request
Last updated
2026-07-22
Source record
OAI-PMH GetRecord
related terms
citation

Jiang, Qingxuan. Choice Modeling and Assortment Optimization on the Transformer Model. Massachusetts Institute of Technology, 2024. https://hdl.handle.net/1721.1/153714