结合深度学习的监督主题模型

作者:苑东东; 赵杰煜*; 叶绪伦
来源:模式识别与人工智能, 2018, 31(08): 715-724.
DOI:10.16451/j.cnki.issn1003-6059.201808004

摘要

无监督主题模型在降维过程中缺少标签信息的指导,丢失一些具有判别性的文本特征,导致最终的分类结果不理想.因此,文中提出结合深度学习的监督主题模型,利用深度网络强大的非线性拟合能力建立文档主题分布与标签之间的映射,利用变分期望最大化(EM)和深度网络训练方法共同完成贝叶斯框架下模型参数的更新,通过改变网络结构和激活函数的类型,用于分类和回归任务.实验表明文中模型既能保持无监督主题模型隐含主题的提取能力,还能更好地完成分类和回归任务.

全文