arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2104.00896cs.LGcs.AI
收藏

贝叶斯神经网络在主动学习中的有效性

Efficacy of Bayesian Neural Networks in Active Learning

Interdigital AI Lab(InterDigital AI实验室)

Vineeth Rakesh, Swayambhoo Jain

摘要:

获取用于机器学习任务的标记数据可能非常昂贵。主动学习通过探索未标记的数据空间并优先选择可以最好地提高模型性能的数据来缓解这个问题。主动学习的一种常见方法是选择模型最不确定的一小部分数据样本。在本文中,我们探讨了贝叶斯神经网络的主动学习,自然模型的不确定性,通过学习分布在神经网络的权重的有效性。通过进行一系列的实验,我们表明,贝叶斯神经网络是更有效的集成技术在捕捉不确定性。我们的研究结果还揭示了一些关键的集成技术,这是最近被证明是更有效的比蒙特卡洛辍学的缺点。

英文摘要:

Obtaining labeled data for machine learning tasks can be prohibitively expensive. Active learning mitigates this issue by exploring the unlabeled data space and prioritizing the selection of data that can best improve the model performance. A common approach to active learning is to pick a small sample of data for which the model is most uncertain. In this paper, we explore the efficacy of Bayesian neural networks for active learning, which naturally models uncertainty by learning distribution over the weights of neural networks. By performing a comprehensive set of experiments, we show that Bayesian neural networks are more efficient than ensemble based techniques in capturing uncertainty. Our findings also reveal some key drawbacks of the ensemble techniques, which was recently shown to be more effective than Monte Carlo dropouts.

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

↑