arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.07077cs.LGstat.ML

Interpretable MTL from Heterogeneous Domains using Boosted Tree

Ant Financial Services Group(蚂蚁金融服务集团)

Ya-Lin Zhang, Longfei Li

英文摘要:

Multi-task learning (MTL) aims at improving the generalization performance of several related tasks by leveraging useful information contained in them. However, in industrial scenarios, interpretability is always demanded, and the data of different tasks may be in heterogeneous domains, making the existing methods unsuitable or unsatisfactory. In this paper, following the philosophy of boosted tree, we proposed a two-stage method. In stage one, a common model is built to learn the commonalities using the common features of all instances. Different from the training of conventional boosted tree model, we proposed a regularization strategy and an early-stopping mechanism to optimize the multi-task learning process. In stage two, started by fitting the residual error of the common model, a specific model is constructed with the task-specific instances to further boost the performance. Experiments on both benchmark and real-world datasets validate the effectiveness of the proposed method. What's more, interpretability can be naturally obtained from the tree based method, satisfying the industrial needs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑