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

面向数据有限属性推断的文本图层次知识蒸馏

Hierarchical Knowledge Distillation on Text Graph for Data-limited Attribute Inference

Pennsylvania State University(宾夕法尼亚州立大学);Wright State University(莱特州立大学)

Quan Li, Shixiong Jing, Lingwei Chen

AI总结:

针对社交媒体文本属性推断中标注数据有限且文本图内存开销大的问题,本文提出基于文本图的少样本学习模型,通过流形学习构建文本图并引入层次知识蒸馏,显著提升了少样本属性推断性能。

摘要:

社交媒体的普及提高了用户参与度,并产生了大量面向用户的数据。其中,文本数据(如推文、博客)显著吸引研究者和投机者推断用户属性(如年龄、性别、地点)以实现其意图。通常,这类工作将属性推断视为文本分类问题,并开始利用图神经网络(GNNs)来利用源文本的更高层次表示。然而,这些文本图是在词上构建的,存在内存消耗高和在少量标注文本上效果不佳的问题。为应对这一挑战,我们设计了一种基于文本图的少样本学习模型,用于社交媒体文本数据上的属性推断。我们的模型首先利用流形学习和消息传递构建并细化文本图,从而在表达能力和复杂度之间取得更好的权衡。随后,为了进一步利用跨领域文本和未标注文本来提升少样本性能,我们在文本图上设计了一种层次知识蒸馏方法来优化该问题,从而获得更好的文本表示,并提升模型泛化能力。在社交媒体数据集上的实验表明,我们的模型在使用显著更少的标注文本进行属性推断时达到了最先进的性能。

英文摘要:

The popularization of social media increases user engagements and generates a large amount of user-oriented data. Among them, text data (e.g., tweets, blogs) significantly attracts researchers and speculators to infer user attributes (e.g., age, gender, location) for fulfilling their intents. Generally, this line of work casts attribute inference as a text classification problem, and starts to leverage graph neural networks (GNNs) to utilize higher-level representations of source texts. However, these text graphs are constructed over words, suffering from high memory consumption and ineffectiveness on few labeled texts. To address this challenge, we design a text-graph-based few-shot learning model for attribute inferences on social media text data. Our model first constructs and refines a text graph using manifold learning and message passing, which offers a better trade-off between expressiveness and complexity. Afterwards, to further use cross-domain texts and unlabeled texts to improve few-shot performance, a hierarchical knowledge distillation is devised over text graph to optimize the problem, which derives better text representations, and advances model generalization ability. Experiments on social media datasets demonstrate the state-of-the-art performance of our model on attribute inferences with considerably fewer labeled texts.

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

↑