Affiliations: [a] Department of Information and Computer Science, Helsinki Institute for Information Technology HIIT, Aalto University, Aalto, Finland | [b] Istanbul Technical University, Department of Computer Engineering, Maslak, Istanbul, Turkey
Correspondence:
[*]
Corresponding author: Samuel Kaski, Aalto University, Department of Information and Computer Science, Helsinki Institute for Information Technology HIIT, P.O. Box 15400, FI-00076 Aalto, Finland. Tel.: +358 9 47098903; Fax: +358 9 47023277; E-mail: samuel.kaski@tkk.fi.
Abstract: We introduce relevant subtask learning, a new learning problem which is a variant of multi-task learning. The goal is to build a classifier for a task-of-interest for which we have too few training samples. We additionally have “supplementary data” collected from other tasks, but it is uncertain which of these other samples are relevant, that is, which samples are classified in the same way as in the task-of-interest. The research problem is how to use the “supplementary data” from the other tasks to improve the classifier in the task-of-interest. We show how to solve the problem, and demonstrate the solution with logistic regression classifiers. The key idea is to model all tasks as mixtures of relevant and irrelevant samples, and model the irrelevant part with a sufficiently flexible model such that it does not distort the model of relevant data. We give two learning algorithms for the method – a simple maximum likelihood optimization algorithm and a more advanced variational Bayes inference algorithm; in both cases we show that the method works better than a comparable multi-task learning model and naive methods.
