集群就是一切:利用语言模型的语义集群预训练Tsetlin机器以实现可解释性
预训练语言模型如BERT在文本分类上表现强劲,但缺乏透明性限制了其在关键场景的应用。Tsetlin机器(TM)提供完全可解释的基于条款的推理,但语义信息捕获有限,且之前尝试结合两者的方法依赖于静态词嵌入,忽视上下文含义。本文提出了一种语义预训练框架,通过无嵌入方式将预训练语言模型的知识转移到TM。文本样本通过K-means或Top2Vec聚成语义一致的集群,基于集群-样本对对非否定TM进行预训练,增强Type I反馈,使TM学习可解释的语义关键词,并在下游任务中微调。实验证明该方法在五个数据集上显著优于传统及基于嵌入的TM,并达到与BERT竞争的性能,同时保持可解释性。
这条 研究论文 信号说明,来自 arXiv cs.CL 的信息已经不只是单点新闻,而是值得放进产品、研究和行业判断里的趋势线索。