arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2110.04572cs.LG
收藏

X-model:使用Minimax模型提高深度学习中的数据效率

X-model: Improving Data Efficiency in Deep Learning with A Minimax Model

Tsinghua University(清华大学)

Ximei Wang, Xinyang Chen, Jianmin Wang, Mingsheng Long

摘要:

为了减轻数据标记的负担,我们的目标是提高深度学习中分类和回归设置的数据效率。然而,目前的重点是分类问题,而很少注意到深度回归,这通常需要更多的人力来标记。此外,由于分类和连续标签空间之间的内在差异,分类的常见直觉,例如,聚类假设或伪标记策略不能自然地适应深度回归。为此,我们首先深入研究了深度学习中现有的数据高效方法,发现它们要么鼓励数据随机性的不变性(例如,不同增广下的一致性正则化)或模型随机性(例如,具有不同dropout的模型的预测的差异惩罚)。为了利用两个世界的力量,我们提出了一个新的X模型,同时鼓励{数据随机性}和{模型随机性}的不变性。此外,X模型在特征提取器和特定任务头部之间进行极大极小博弈,以进一步增强模型随机性的不变性。大量的实验验证了X模型在各种任务中的优越性,从年龄估计的单值预测任务到关键点定位的密集值预测任务,2D合成和3D真实数据集,以及多类别对象识别任务。

英文摘要:

To mitigate the burden of data labeling, we aim at improving data efficiency for both classification and regression setups in deep learning. However, the current focus is on classification problems while rare attention has been paid to deep regression, which usually requires more human effort to labeling. Further, due to the intrinsic difference between categorical and continuous label space, the common intuitions for classification, e.g., cluster assumptions or pseudo labeling strategies, cannot be naturally adapted into deep regression. To this end, we first delved into the existing data-efficient methods in deep learning and found that they either encourage invariance to data stochasticity (e.g., consistency regularization under different augmentations) or model stochasticity (e.g., difference penalty for predictions of models with different dropout). To take the power of both worlds, we propose a novel X-model by simultaneously encouraging the invariance to {data stochasticity} and {model stochasticity}. Further, the X-model plays a minimax game between the feature extractor and task-specific heads to further enhance the invariance to model stochasticity. Extensive experiments verify the superiority of the X-model among various tasks, from a single-value prediction task of age estimation to a dense-value prediction task of keypoint localization, a 2D synthetic, and a 3D realistic dataset, as well as a multi-category object recognition task.

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

↑