Efficient Multinomial Logistic Bandit via Robust Frequent Directions
Abstract
This paper studies efficient online algorithms for multinomial logistic bandits (MLogB), where the feedback distribution over outcomes follows a multinomial logistic model of -dimensional action vectors. A representative UCB-type algorithm, OFUL-MLogB, achieves a regret bound of , but still requires time and space per round due to parameter estimation and optimistic reward construction, which is prohibitive in high-dimensional settings. To address this limitation, we propose EOFD-MLogB, an efficient algorithm integrating robust frequent directions matrix sketching. By maintaining a low-rank SVD sketch of the Hessian matrix from online Newton step (ONS)-based parameter estimation, we make both nonlinear estimation and optimistic reward construction efficient: the constrained ONS parameter update is reduced to a one-dimensional root-finding problem, while the spectral-norm computation in the optimistic reward is reduced to a eigenvalue problem. This yields dominant per-round time complexity and space complexity , where is the sketch size. We further prove a regret bound of , where the sketching error factor is controlled by the -truncated spectral tail of the ONS Hessian. Thus, when the ONS Hessian is approximately low-rank, the regret remains close to that of OFUL-MLogB. Experiments validate the computational efficiency and competitive performance.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.