Distilling Large Language Models for Text-Attributed Graph Learning

LLM在文本属性图学习中的应用与蒸馏
本文探讨如何将大型语言模型(LLM)的知识转移到局部图模型,用于文本属性图(TAG)学习。研究发现,尽管LLM在小样本和零样本学习中有优势,但存在可扩展性、成本和隐私问题。为此,提出了一种新框架,通过LLM教导的解释器模型,使图模型能够模仿LLM的推理,提高TAG学习效果。实验结果显示,该框架在四个数据集上平均提升了1.25%的性能。

本文是LLM系列文章,针对《Distilling Large Language Models for Text-Attributed Graph Learning》的翻译。

文本属性图学习的大型语言模型提取

摘要

文本属性图(TAG)是连接的文本文档的图。图模型可以有效地学习标签,但它们的训练在很大程度上依赖于人工注释标签,而在许多应用程序中,人工注释标签很少甚至不可用。大型语言模型(LLM)最近在小样本和零样本TAG学习方面表现出了显著的能力,但它们存在可扩展性、成本和隐私问题。因此,在这项工作中,我们专注于通过在TAG学习中提取LLM到局部图模型的能力,将LLM和图模型与其互补的优势协同起来。为了解决LLM(文本的生成模型)和图模型(图的判别模型)之间的固有差距,我们建议首先让LLM教授具有丰富文本基本原理的口译员,然后让学生模型在没有LLM文本基本原理情况下模仿口译员的推理。大量实验验证了我们提出的框架的有效性。

1 引言

2 相关工作

3 前言

4 方法

5 实验

6 结论

在本文中,我们提出了一个将LLM提取到TAG学习的图模型的框架,该框架允许使用预测和推理进行训练。我们训练解释器模型,将LLM的基本原理与图模型连接起来,并根据TAG数据的性质调整解释器模型。在四个数据集上,我们提出的蒸馏方法平均超过基线1.25%。

已下架不支持订阅

Distilling Aggregated Knowledge for Weakly - Supervised Video Anomaly Detection”聚焦于弱监督视频异常检测领域。在弱监督的场景下,训练数据仅带有粗略的标签信息,不像强监督那样具有精确的逐帧标注,这给视频异常检测带来了挑战。 该研究提出的方法核心在于知识蒸馏和聚合。知识蒸馏是一种将复杂模型(教师模型)的知识传递给简单模型(学生模型)的技术,通过这种方式可以让学生模型学习到教师模型所蕴含的更丰富的信息。在视频异常检测中,利用知识蒸馏可以帮助模型更好地捕捉视频中的异常模式。 聚合知识的过程则是将多个不同层面或者不同视角的信息进行整合。在视频数据中,可能包括不同帧之间的时序信息、不同特征维度的空间信息等。通过聚合这些信息,能够让模型从更全面的角度去判断视频是否存在异常。 例如,在实际应用中,对于监控视频里的异常行为检测,可能只有少量的视频被标注为异常,但不清楚具体哪一帧或者哪一段是异常的。利用该研究的方法,模型可以通过知识蒸馏和聚合,从有限的标注信息中学习到异常行为的特征,从而对新的未标注视频进行异常检测。 ```python # 这里可以简单示意一个可能的代码思路,仅为概念示意 # 假设定义一个简单的学生模型和教师模型 import torch import torch.nn as nn # 简单定义教师模型 class TeacherModel(nn.Module): def __init__(self): super(TeacherModel, self).__init__() # 这里简单用一个线性层示意 self.fc = nn.Linear(10, 2) def forward(self, x): return self.fc(x) # 简单定义学生模型 class StudentModel(nn.Module): def __init__(self): super(StudentModel, self).__init__() # 简单用一个线性层示意 self.fc = nn.Linear(10, 2) def forward(self, x): return self.fc(x) teacher = TeacherModel() student = StudentModel() # 模拟输入数据 input_data = torch.randn(1, 10) # 教师模型输出 teacher_output = teacher(input_data) # 这里可以添加知识蒸馏的损失函数等后续处理步骤 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UnknownBody

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值