Skip to content

Code for our ICDMW 2018 paper: "Contextual Bandit with Adaptive Feature Extraction".

Notifications You must be signed in to change notification settings

doerlbh/ABaCoDE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ABaCoDE

Code for our ICDMW 2018 paper:

"Contextual Bandit with Adaptive Feature Extraction"

by Baihan Lin (Columbia)*, Djallel Bouneffouf (IBM Research), Guillermo Cecchi (IBM Research) and Irina Rish (Mila, UdeM).

*Corresponding

For the latest full paper: https://arxiv.org/abs/1802.00981

All the experimental results can be reproduced using the code in this repository. Feel free to contact me by [email protected] if you have any question about our work.

Abstract

We consider an online decision making setting known as contextual bandit problem, and propose an approach for improving contextual bandit performance by using an adaptive feature extraction (representation learning) based on online clustering. Our approach starts with an off-line pre-training on unlabeled history of contexts (which can be exploited by our approach, but not by the standard contextual bandit), followed by an online selection and adaptation of encoders. Specifically, given an input sample (context), the proposed approach selects the most appropriate encoding function to extract a feature vector which becomes an input for a contextual bandit, and updates both the bandit and the encoding function based on the context and on the feedback (reward). Our experiments on a variety of datasets, and both in stationary and non-stationary environments of several kinds demonstrate clear advantages of the proposed adaptive representation learning over the standard contextual bandit based on "raw" input contexts.

Info

Language: MATLAB, bash

Platform: MacOS, Linux, Windows

by Baihan Lin, Dec 2017

Citation

If you find this work helpful, please try the models out and cite our works. Thanks!

@inproceedings{lin2018contextual,
  title={Contextual Bandit with Adaptive Feature Extraction},
  author={Lin, Baihan and Bouneffouf, Djallel and Cecchi, Guillermo A and Rish, Irina},
  booktitle={2018 IEEE International Conference on Data Mining Workshops (ICDMW)},
  pages={937--944},
  year={2018},
  organization={IEEE}
}

Tasks

  • MNIST
  • STL-10
  • CIFAR-10
  • Caltech-101 S
  • Warfarin
  • mix: MNIST/Warfarin

Nonstationarity

  • Nonstationary context: varying cluster distribution

  • Nonstationary context: negative images

  • Nonstationary reward: multi-task environment

  • Nonstationary reward: shuffled class labels

Requirements

  • MATLAB with deep learning toolbox

Releases

No releases published

Packages

No packages published

Languages