arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2202.01940stat.MLcs.LG
收藏

基于分布嵌入网络的分类任务泛化

Distribution Embedding Networks for Generalization from a Diverse Set of Classification Tasks

Google Research(谷歌研究院);University of Washington(华盛顿大学)

Lang Liu, Mahdi Milani Fard, Sen Zhao

摘要:

我们提出了分布嵌入网络(DEN)的分类与小数据。本着元学习的精神,DEN从一组不同的训练任务中学习,目标是推广到看不见的目标任务。与现有的方法,需要输入的训练和目标任务具有相同的尺寸可能类似的分布,DEN允许训练和目标任务生活在异构输入空间。这对于表格数据任务特别有用,因为相关任务的标记数据很少。DEN使用三块架构:协变量变换块,然后是分布嵌入块,然后是分类块。我们提供了理论见解,表明这种架构允许在对不同的任务进行预训练后固定嵌入和分类块;只有具有相对较少参数的协变量变换块需要为每个新任务进行微调。为了便于训练,我们还提出了一种方法来合成二进制分类任务,并证明DEN优于现有的方法,在一些合成和真实的任务,在数值研究。

英文摘要:

We propose Distribution Embedding Networks (DEN) for classification with small data. In the same spirit of meta-learning, DEN learns from a diverse set of training tasks with the goal to generalize to unseen target tasks. Unlike existing approaches which require the inputs of training and target tasks to have the same dimension with possibly similar distributions, DEN allows training and target tasks to live in heterogeneous input spaces. This is especially useful for tabular-data tasks where labeled data from related tasks are scarce. DEN uses a three-block architecture: a covariate transformation block followed by a distribution embedding block and then a classification block. We provide theoretical insights to show that this architecture allows the embedding and classification blocks to be fixed after pre-training on a diverse set of tasks; only the covariate transformation block with relatively few parameters needs to be fine-tuned for each new task. To facilitate training, we also propose an approach to synthesize binary classification tasks, and demonstrate that DEN outperforms existing methods in a number of synthetic and real tasks in numerical studies.

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

↑