Shrinking the Upper Confidence Bound: A Dynamic Product Selection Problem for Urban Warehouses: ConsUCB Has Regret at Most K + αK + 10α√(dKT log(KT)) with Probability at Least 1 − δResearch Paper
Selecting many products over a short season
An online retailer choosing which low-demand products to display, and to stock in its urban warehouses, faces an unusual learning problem. Thousands of products must be chosen at once, the selection can change only a few times per season, and sales data arrive only for the products that were offered. Jin, Simchi-Levi, Wang, Wang and Yang (arXiv:1903.07844v2) model this as a stochastic linear semi-bandit in which the number of selections per period is large and the number of periods is small. Their study uses data from Alibaba Tmall.
The problem belongs to a line of work on linear bandits. Abbasi-Yadkori, Pál and Szepesvári (2011) gave the OFUL algorithm and its confidence ellipsoids for one action per period. Chu, Li, Reyzin and Schapire (2011) bounded the sum of confidence widths used in the analysis. Qin, Chen and Zhu (2014) and Wen, Kveton and Ashkan (2015) studied richer semi-bandit models, but their bounds carry terms such as , which are of little use when is large and is small.
The model and the algorithm
There are products, each with a known feature vector , and an unknown parameter with , . Product has positive sales with probability . In each period the retailer offers a set of products and observes binary outcomes for , which are independent across products and periods. The regret against an optimal -set is
ConsUCB keeps the matrices and vectors of regularized least squares, with and , and the estimate . Within a period it selects products one at a time. The -th selection maximizes, over the products not yet selected, the score
where and for the selected product . The second radical shrinks after each selection, using features only: no outcome of the period is seen until all products are chosen.
Formalization targets
Goal: Theorem 3
For , , , , every instance, every optimal set and every tie-break depending on the observed history, ConsUCB with satisfies, with probability at least ,
This is the explicit bound printed at the end of the proof; it implies the theorem's .
Milestones
In the order of the proof: the confidence event (4), the monotonicity of the widths, the inequalities (5)–(6), Proposition 1, the regret decomposition, the first line of the final display, the boundary terms, Lemma 2 with the correction , and the rewriting of the bound in form.
What the result gives
Theorem 3 splits the regret into a fixed cost , independent of , and a variable cost . The same paper shows that the standard upper confidence bound algorithm has fixed cost and at least . ConsUCB therefore improves the fixed cost by a factor of , which is what matters when is small, since the total reward over the season is of order . Since an regret in the first period cannot be avoided, the fixed cost is optimal up to a factor .
The result is proved in the paper; no machine-checked proof is known. The formalization adds a checked account of the within-period selection argument and a statement that makes the constants, the tie-breaking and the failure event explicit. It also records one defect: Lemma 2 as printed fails at , a gap in the proof of Theorem 3 that closes with one line, since there .
Where the difficulty lies
Within a period, selections must be made before any outcome of that period is seen. The obvious argument treats the period as one optimistic choice of products, all scored with . It then has to compare the width of a selected product under with its width after the earlier selections of the same period, and that comparison is what produces the fixed cost , which the paper's lower bound shows is not an artefact of a loose analysis in alone. Any improvement must change the algorithm, and the new score no longer equals the upper confidence bound: for it can fall below it, so the usual optimism step does not apply.
The probabilistic step has its own care: the confidence event must hold for a run in which the design vectors of a period are chosen sequentially, at horizon , including .
Formalization scope
All declarations are in the namespace ShrinkingUCB, over Mathlib; the conventions below are binding.
- Vectors are
Fin d → ℝ, norms are written , andMatrix.invis the inverse (every dominates , so it is genuine). Logarithms are natural. - Periods and selections are -based. ConsUCB is a recursion on an outcome table, so every object is a deterministic function of the table, evaluated at the table realized at .
- Outcomes form a table of independent
Boolvariables, with the algorithm reading only the offered entries. A tie-break is a deterministic function of the observed history, the selections so far in the period and the tied set. It cannot read an unobserved outcome; a rule that could would make the goal false, and restricting to one fixed rule would weaken it. - Failure probabilities are stated as with the outer measure.
- The bound is kept at , and no regime such as is assumed. A statement with a hard-coded tie-break, a matching of and in the score, or an added is not this theorem.
A complete development needs a self-normalized bound along the observation sequence ordered within periods, a bound on the sum of unsquared widths, and the bookkeeping that the selections are distinct. Published tools such as ImprovedLinBandits.OFUL.confidence_ellipsoid and BanditAlgorithm.elliptical_potential_lemma are relevant but are not stated in this form. Contributions of the linear algebra of rank-one updates are reusable beyond this mission and welcome.
Selected references
- R. Jin, D. Simchi-Levi, L. Wang, X. Wang, S. Yang, Shrinking the Upper Confidence Bound: A Dynamic Product Selection Problem for Urban Warehouses, arXiv:1903.07844v2, 2019. https://arxiv.org/abs/1903.07844
- Y. Abbasi-Yadkori, D. Pál, C. Szepesvári, Improved Algorithms for Linear Stochastic Bandits, NIPS, 2011. https://papers.nips.cc/paper/2011/hash/e1d5be1c7f2f456670de3d53c7b54f4a-Abstract.html
- W. Chu, L. Li, L. Reyzin, R. Schapire, Contextual Bandits with Linear Payoff Functions, AISTATS, 2011. https://proceedings.mlr.press/v15/chu11a.html
- L. Qin, S. Chen, X. Zhu, Contextual Combinatorial Bandit and its Application on Diversified Online Recommendation, SIAM SDM, 2014. https://doi.org/10.1137/1.9781611973440.53
- Z. Wen, B. Kveton, A. Ashkan, Efficient Learning in Large-Scale Combinatorial Semi-Bandits, ICML, 2015. https://proceedings.mlr.press/v37/wen15.html