让神经网络直接预测整数标签,提升精度与合理性
Predicting integers from continuous parameters
- 用比特位建模整数,每个位独立设伯努利分布
- 提出离散拉普拉斯分布,以连续均值为中心指数衰减
- 在图像生成等任务中表现优于传统回归方法
我们研究了需预测整数标签(如社交媒体点赞数、公共自行车租赁站剩余数量)的问题。传统回归将离散标签视为连续值,改变了其分布特性。为此,我们探索直接用离散分布建模整数标签的方法,并要求分布参数为连续值以支持反向传播。实验对比了多种现有及新提出的分布,包括基于比特位的伯努利模型和具有指数尾部衰减的离散拉普拉斯分布。结果表明,在表格学习、序列预测和图像生成等任务中,这两种方法表现最佳。
原文摘要 · Abstract (English)
We study the problem of predicting numeric labels that are constrained to the integers or to a subrange of the integers. For example, the number of up-votes on social media posts, or the number of bicycles available at a public rental station. While it is possible to model these as continuous values, and to apply traditional regression, this approach changes the underlying distribution on the labels from discrete to continuous. Discrete distributions have certain benefits, which leads us to the question whether such integer labels can be modeled directly by a discrete distribution, whose parameters are predicted from the features of a given instance. Moreover, we focus on the use case of output distributions of neural networks, which adds the requirement that the parameters of the distribution be continuous so that backpropagation and gradient descent may be used to learn the weights of the network. We investigate several options for such distributions, some existing and some novel, and test them on a range of tasks, including tabular learning, sequential prediction and image generation. We find that overall the best performance comes from two distributions: Bitwise, which represents the target integer in bits and places a Bernoulli distribution on each, and a discrete analogue of the Laplace distribution, which uses a distribution with exponentially decaying tails around a continuous mean.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。