Bilinear Bandits with Partially Observable Features

Wooseong Cho, Ji Hyeong Park, Min-Hwan Oh
Proceedings of the 43rd International Conference on Machine Learning, PMLR 306:19787-19818, 2026.

Abstract

We study bilinear bandits with partially observable features on both the user and item sides. In each of $T$ rounds, the learner selects a user–item pair and observes only the reward for the chosen pair. The reward model is bilinear in the user and item features with an unknown parameter matrix. Existing work commonly reduces this problem to a linear bandit via Kronecker-product features, increasing dimensionality and losing bilinear structure. We propose BiRoLF, an algorithm robust to latent features, which directly leverages the bilinear structure without such linearization. It augments observed feature spaces on both sides with orthogonal complement bases and employs doubly robust (DR) estimation to impute rewards for unselected pairs, constructing matrix-valued pseudo-rewards. We estimate the effective parameter using a bilinear DR-Lasso estimator, which promotes sparsity in components orthogonal to observed features. BiRoLF achieves a $\tilde{O}(\sqrt{(d_x + d_{h_x})(d_y + d_{h_y}) T})$ regret bound, where $d_x$ and $d_y$ are observable feature dimensions, and $d_{h_x}$ and $d_{h_y}$ denote effective orthogonal-complement dimensions. By exploiting the induced block-diagonal Gram structure, BiRoLF performs an exact blockwise update that preserves the full DR-Lasso objective. Numerical experiments show strong regret performance and computational gains.

Cite this Paper


BibTeX
@InProceedings{pmlr-v306-cho26f, title = {Bilinear Bandits with Partially Observable Features}, author = {Cho, Wooseong and Park, Ji Hyeong and Oh, Min-Hwan}, booktitle = {Proceedings of the 43rd International Conference on Machine Learning}, pages = {19787--19818}, year = {2026}, editor = {Zhang, Tong and Dudik, Miroslav and Jaggi, Martin and Agarwal, Alekh and Li, Sharon and Schuurmans, Dale and Zhu, Jerry and Berkenkamp, Felix and Dong, Hanze and Bietti, Alberto}, volume = {306}, series = {Proceedings of Machine Learning Research}, month = {06--11 Jul}, publisher = {PMLR}, pdf = {https://raw.githubusercontent.com/mlresearch/v306/main/assets/cho26f/cho26f.pdf}, url = {https://proceedings.mlr.press/v306/cho26f.html}, abstract = {We study bilinear bandits with partially observable features on both the user and item sides. In each of $T$ rounds, the learner selects a user–item pair and observes only the reward for the chosen pair. The reward model is bilinear in the user and item features with an unknown parameter matrix. Existing work commonly reduces this problem to a linear bandit via Kronecker-product features, increasing dimensionality and losing bilinear structure. We propose BiRoLF, an algorithm robust to latent features, which directly leverages the bilinear structure without such linearization. It augments observed feature spaces on both sides with orthogonal complement bases and employs doubly robust (DR) estimation to impute rewards for unselected pairs, constructing matrix-valued pseudo-rewards. We estimate the effective parameter using a bilinear DR-Lasso estimator, which promotes sparsity in components orthogonal to observed features. BiRoLF achieves a $\tilde{O}(\sqrt{(d_x + d_{h_x})(d_y + d_{h_y}) T})$ regret bound, where $d_x$ and $d_y$ are observable feature dimensions, and $d_{h_x}$ and $d_{h_y}$ denote effective orthogonal-complement dimensions. By exploiting the induced block-diagonal Gram structure, BiRoLF performs an exact blockwise update that preserves the full DR-Lasso objective. Numerical experiments show strong regret performance and computational gains.} }
Endnote
%0 Conference Paper %T Bilinear Bandits with Partially Observable Features %A Wooseong Cho %A Ji Hyeong Park %A Min-Hwan Oh %B Proceedings of the 43rd International Conference on Machine Learning %C Proceedings of Machine Learning Research %D 2026 %E Tong Zhang %E Miroslav Dudik %E Martin Jaggi %E Alekh Agarwal %E Sharon Li %E Dale Schuurmans %E Jerry Zhu %E Felix Berkenkamp %E Hanze Dong %E Alberto Bietti %F pmlr-v306-cho26f %I PMLR %P 19787--19818 %U https://proceedings.mlr.press/v306/cho26f.html %V 306 %X We study bilinear bandits with partially observable features on both the user and item sides. In each of $T$ rounds, the learner selects a user–item pair and observes only the reward for the chosen pair. The reward model is bilinear in the user and item features with an unknown parameter matrix. Existing work commonly reduces this problem to a linear bandit via Kronecker-product features, increasing dimensionality and losing bilinear structure. We propose BiRoLF, an algorithm robust to latent features, which directly leverages the bilinear structure without such linearization. It augments observed feature spaces on both sides with orthogonal complement bases and employs doubly robust (DR) estimation to impute rewards for unselected pairs, constructing matrix-valued pseudo-rewards. We estimate the effective parameter using a bilinear DR-Lasso estimator, which promotes sparsity in components orthogonal to observed features. BiRoLF achieves a $\tilde{O}(\sqrt{(d_x + d_{h_x})(d_y + d_{h_y}) T})$ regret bound, where $d_x$ and $d_y$ are observable feature dimensions, and $d_{h_x}$ and $d_{h_y}$ denote effective orthogonal-complement dimensions. By exploiting the induced block-diagonal Gram structure, BiRoLF performs an exact blockwise update that preserves the full DR-Lasso objective. Numerical experiments show strong regret performance and computational gains.
APA
Cho, W., Park, J.H. & Oh, M.. (2026). Bilinear Bandits with Partially Observable Features. Proceedings of the 43rd International Conference on Machine Learning, in Proceedings of Machine Learning Research 306:19787-19818 Available from https://proceedings.mlr.press/v306/cho26f.html.

Related Material