Awesome-Deep-Graph-Clustering
Awesome-Deep-Graph-Clustering 是一个专注于深度图聚类领域的开源资源合集,旨在汇集该方向最前沿(SOTA)的研究成果。它系统地整理了相关的学术论文、代码实现以及数据集,为探索复杂网络结构提供了坚实的技术支撑。
在人工智能应用中,如何从无标号的图数据中自动发现节点间的内在联系并将其合理分组,一直是个极具挑战的难题。Awesome-Deep-Graph-Clustering 正是为了解决这一痛点而生,它帮助使用者快速定位高质量的研究文献与可复现的代码,避免了在海量信息中筛选的高昂时间成本。此外,该资源库还紧跟技术潮流,收录了基于大语言模型(LLM)引导的图聚类等新兴架构方法,展现了领域内的最新突破。
这款工具特别适合人工智能研究人员、算法工程师以及高校师生使用。无论是希望深入了解图神经网络聚类机理的学者,还是需要寻找基准代码进行二次开发的开发者,都能从中获得宝贵的参考资源。通过提供清晰的分类索引和详尽的引用指引,Awesome-Deep-Graph-Clustering 致力于成为连接理论研究与工程实践的桥梁,推动社区共同进步。
使用场景
某生物信息学团队正试图从大规模蛋白质相互作用网络中识别功能未知的蛋白质群落,以加速新药靶点的发现。
没有 Awesome-Deep-Graph-Clustering 时
- 文献调研如大海捞针:研究人员需在 arXiv、IEEE 等各大数据库手动筛选“深度图聚类”相关论文,耗时数周仍难以确认哪些是真正的 SOTA(最先进)方法。
- 代码复现门槛极高:找到的论文往往缺乏官方开源代码,或代码结构混乱、依赖缺失,导致算法复现失败率高达 80%,严重拖慢实验进度。
- 数据预处理重复造轮子:缺乏统一的基准数据集和标准化预处理脚本,团队成员需各自编写数据清洗代码,导致实验结果无法横向对比,可信度存疑。
- 技术选型盲目试错:由于缺乏对 LLM 结合图聚类等新兴架构的系统整理,团队容易错过前沿方案,只能沿用几年前的旧模型,挖掘精度遭遇瓶颈。
使用 Awesome-Deep-Graph-Clustering 后
- 一站式获取前沿成果:直接查阅该仓库整理的分类列表,几分钟内即可锁定 2024 年最新的"LLM 引导图聚类”等顶会论文,调研效率提升十倍。
- 开箱即用的算法实现:直接调用仓库中经过验证的 SOTA 方法代码,无需修复环境兼容性问题,将原本数周的复现周期缩短至几天。
- 统一基准确保可比性:利用仓库提供的标准数据集和预处理流程,团队能快速在不同算法间进行公平性能评估,显著提升了实验结论的说服力。
- 精准把握技术风向:通过清晰的架构分类(如新架构、综述文章),团队迅速引入适合生物网络特性的最新模型,成功将未知蛋白群的识别准确率提升了 15%。
Awesome-Deep-Graph-Clustering 通过聚合高质量的论文、代码与数据,将科研人员从繁琐的基础设施搭建中解放出来,使其能专注于核心算法的创新与业务价值的挖掘。
运行环境要求
未说明
未说明

快速开始
ADGC:超赞深度图聚类
ADGC 是一个汇集了最先进(SOTA)、新颖的深度图聚类方法(论文、代码和数据集)的资源库。欢迎提交其他有趣的论文和代码。如有任何问题,请联系 yueliu19990731@163.com。如果您觉得本仓库对您的研究或工作有所帮助,真诚地感谢您为本仓库点亮星标。:sparkles: 如果您在研究中使用了本仓库中的代码或处理后的数据集,请在引用部分 此处 引用 2–3 篇相关论文。:heart:
什么是深度图聚类?
深度图聚类旨在揭示底层图结构,并将节点划分为不同的群组,近年来受到了广泛关注。更多细节请参阅综述论文。链接
重要综述论文
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2023 | 高级深度图节点聚类概述 | TCSS | 链接 | - |
| 2022 | 深度图聚类综述:分类、挑战与应用 | arXiv | 链接 | 链接 |
| 2022 | 基于深度学习的社区发现综合综述 | TNNLS | 链接 | - |
| 2020 | 图神经网络综合综述 | TNNLS | 链接 | - |
| 2020 | 深度学习用于社区发现:进展、挑战与机遇 | IJCAI | 链接 | - |
| 2018 | 基于深度学习的聚类综述:从网络架构视角 | IEEE Access | 链接 | - |
论文
基于 LLM 的深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | 大型语言模型引导的图聚类 | LOG | 链接 | - |
新架构深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | 用于图上归一化割的扩张层次结构 | KDD | 链接 | https://zenodo.org/records/12108189 |
| 2024 | 用于图的科尔莫戈洛夫-阿诺德网络(KAN) | - | - | 链接 |
时序深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | 基于图的时间序列聚类,用于端到端分层预测 | arxiv | 链接 | - |
| 2024 | 深度时序图聚类(TGC) | ICLR | 链接 | 链接 |
聚类数目未知的深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | NeuroCUT:一种鲁棒的图划分神经方法 | KDD | 链接 | 链接 |
| 2024 | LSEnet:用于深度图聚类的洛伦兹结构熵神经网络(LSEnet) | ICML | 链接 | 链接 |
| 2024 | 无预先指定聚类数 k 的图聚类掩码自编码器(GCMA) | arXiv | 链接 | - |
| 2023 | 聚类数目未知的强化图聚类(RGC) | ACM MM | 链接 | 链接 |
重构型深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | 协同深度图聚类网络(SynC) | Arxiv | 链接 | 链接 |
| 2024 | 深度掩码图节点聚类(DMGC) | TCSS | 链接 | - |
| 2024 | 多尺度图聚类网络(MGCN) | IS | 链接 | 链接 |
| 2024 | 基于在线互学习的端到端深度图聚类 | TNNLS | 链接 | - |
| 2024 | 用于社区发现的对比深度非负矩阵分解(CDNMF) | ICASSP | 链接 | 链接 |
| 2023 | EGRC-Net:嵌入诱导的图精炼聚类网络(EGRC-Net) | TIP | 链接 | 链接 |
| 2023 | 超越证据下界:用于节点聚类的对偶变分图自编码器(BELBO-VGAE) | SDM | 链接 | 链接 |
| 2023 | 基于图神经网络的图聚类(DMoN) | JMLR | 链接 | 链接 |
| 2023 | 结构嵌入增强的图聚类网络(GC-SEE) | PR | 链接 | 链接 |
| 2023 | 超越同质性:为图无关聚类重建结构(DGCN) | ICML | 链接 | 链接 |
| 2023 | 走向凸流形:单细胞RNA-seq数据深度图聚类的几何视角(scTCM) | IJCAI | 链接 | 链接 |
| 2023 | 探索局部与全局潜在配置的交互作用以进行单细胞RNA-seq聚类:统一视角(scTPF) | AAAI | 链接 | 链接 |
| 2022 | 逃离特征扭曲:用于节点聚类的变分图自编码器(FT-VGAE) | IJCAI | 链接 | 链接 |
| 2022 | 深度注意力引导的双自监督图聚类(DAGC) | TCSVT | 链接 | 链接 |
| 2022 | 重新思考属性图聚类的图自编码器模型(R-GAE) | TKDE | 链接 | 链接 |
| 2022 | 图嵌入聚类:具有簇特异性分布的图注意力自编码器(GEC-CSD) | NN | 链接 | - |
| 2022 | 通过网络嵌入探索时间社区结构(VGRGMM) | TCYB | 链接 | - |
| 2022 | 聚类感知的异构信息网络嵌入(VaCA-HINE) | WSDM | 链接 | - |
| 2022 | 用于联合节点表示学习和聚类的有效图卷积(GCC) | WSDM | 链接 | 链接 |
| 2022 | 基于ZINB的单细胞RNA-seq解释用图嵌入自编码器(scTAG) | AAAI | 链接 | 链接 |
| 2022 | 图社区信息最大化(GCI) | TKDD | 链接 | - |
| 2022 | 多级子空间融合的深度图聚类(DGCSF) | PR | 链接 | - |
| 2022 | 基于变分图嵌入的图聚类(GC-VAE) | PR | 链接 | - |
| 2022 | 深度邻居感知嵌入用于属性图中的节点聚类(DNENC) | PR | 链接 | - |
| 2022 | 协作决策强化的属性图聚类自监督(CDRS) | TNNLS | 链接 | 链接 |
| 2022 | 用于图聚类的嵌入图自编码器(EGAE) | TNNLS | 链接 | 链接 |
| 2021 | 用于多视图聚类的自监督图卷积网络(SGCMC) | TMM | 链接 | 链接 |
| 2021 | 用于关系数据聚类的适应性超图自编码器(AHGAE) | TKDE | 链接 | - |
| 2021 | 注意力驱动的图聚类网络(AGCN) | ACM MM | 链接 | 链接 |
| 2021 | 深度融合聚类网络(DFCN) | AAAI | 链接 | 链接 |
| 2020 | 协作图卷积网络:无监督学习与半监督学习的结合(CGCN) | AAAI | 链接 | 链接 |
| 2020 | 基于注意力跨图关联的深度多图聚类(DMGC) | WSDM | 链接 | 链接 |
| 2020 | 深入探索:图卷积梯形网络(GCLN) | AAAI | 链接 | - |
| 2020 | 用于聚类的多视图属性图卷积网络(MAGCN) | IJCAI | 链接 | 链接 |
| 2020 | 用于多视图图聚类的One2Multi图自编码器(O2MAC) | WWW | 链接 | 链接 |
| 2020 | 结构化深度聚类网络(SDCN/SDCN_Q) | WWW | 链接 | 链接 |
| 2020 | 狄利克雷图变分自编码器(DGVAE) | NeurIPS | 链接 | 链接 |
| 2019 | RWR-GAE:用于图自编码器的随机游走正则化(RWR-GAE) | arXiv | 链接 | 链接 |
| 2019 | 用于无监督图表示学习的对称图卷积自编码器(GALA) | ICCV | 链接 | 链接 |
| 2019 | 属性图聚类:一种深度注意力嵌入方法(DAEGC) | IJCAI | 链接 | 链接 |
| 2019 | 用于属性网络嵌入的网络特定变分自编码器(NetVAE) | IJCAI | 链接 | - |
| 2017 | 动态嵌入的图聚类(GRACE) | arXiv | 链接 | 链接 |
| 2017 | MGAE:用于图聚类的边缘化图自编码器(MGAE) | CIKM | 链接 | 链接 |
| 2017 | 在图上同时进行社区检测与节点嵌入以学习社区嵌入(ComE) | CIKM | 链接 | 链接 |
| 2016 | 用于学习图表示的深度神经网络(DNGR) | AAAI | 链接 | 链接 |
| 2015 | 基于深度架构的异构网络嵌入(HNE) | SIGKDD | 链接 | - |
| 2014 | 用于图聚类的学习深度表示(GraphEncoder) | AAAI | 链接 | 链接 |
对抗性深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2023 | Wasserstein对抗正则化图自编码器(WARGA) | Neurocomputing | 链接 | 链接 |
| 2022 | 超越同质性的无监督网络嵌入(SELENE) | TMLR | 链接 | 链接 |
| 2020 | JANE:联合对抗网络嵌入(JANE) | IJCAI | 链接 | - |
| 2019 | 用于集成聚类的对抗图嵌入(AGAE) | IJCAI | 链接 | - |
| 2019 | CommunityGAN:基于生成对抗网络的社区发现(CommunityGAN) | WWW | 链接 | 链接 |
| 2019 | ProGAN:通过邻近关系生成对抗网络进行网络嵌入(ProGAN) | SIGKDD | 链接 | - |
| 2019 | 利用对抗训练方法学习图嵌入(ARGA/ARVGA) | TCYB | 链接 | 链接 |
| 2019 | 用于图嵌入的对抗正则化图自编码器(ARGA/ARVGA) | IJCAI | 链接 | 链接 |
对比学习深度图聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | 重新审视图聚类中的模块度最大化:对比学习视角 | SIGKDD | 链接 | 链接 |
| 2024 | GLAC-GCN:全局与局部拓扑感知的对比图聚类网络 (GLAC-GCN) | TAI | 链接 | 链接 |
| 2024 | 具有自适应编码器的多视图属性图对比聚类 | TNNLS | 链接 | - |
| 2024 | 用于社区发现的对比深度非负矩阵分解 (CDNMF) | ICASSP | 链接 | 链接 |
| 2023 | 用于节点聚类的对比变分图自编码器 (CVGAE) | PR | 链接 | 链接 |
| 2023 | 用于图聚类的双对比学习网络 | TNNLS | 链接 | 链接 |
| 2023 | 带有保簇增强的对比学习用于属性图聚类 | ECML-PKDD | 链接 | - |
| 2023 | 具有输入感知和簇感知正则化的图对比表示学习 | ECML-PKDD | 链接 | - |
| 2023 | 未知簇数的强化图聚类 (RGC) | ACM MM | 链接 | 链接 |
| 2023 | 自对比图扩散网络 | ACM MM | 链接 | 链接 |
| 2023 | CONVERT:基于可靠增强的对比图聚类 (CONVERT) | ACM MM | 链接 | 链接 |
| 2023 | 通过可学习增强进行属性图聚类 (AGCLA) | arXiv | 链接 | - |
| 2023 | 加速图上表示学习的聚类方法 (CARL-G) | SIGKDD | 链接 | - |
| 2023 | Dink-Net:大规模图上的神经网络聚类 (Dink-Net) | ICML | 链接 | 链接 |
| 2023 | CONGREGATE:曲率空间中的对比图聚类 (CONGREGATE) | IJCAI | 链接 | 链接 |
| 2023 | 多层级图对比原型聚类 | IJCAI | 链接 | - |
| 2023 | 简单对比图聚类 (SCGC) | TNNLS | 链接 | 链接 |
| 2023 | 硬样本感知的对比深度图聚类网络 (HSAN) | AAAI | 链接 | 链接 |
| 2023 | 簇引导的对比图聚类网络 (CCGC) | AAAI | 链接 | 链接 |
| 2022 | NCAGC:用于属性图聚类的邻域对比框架 (NCAGC) | arXiv | 链接 | 链接 |
| 2022 | SCGC:自监督对比图聚类 (SCGC) | arXiv | 链接 | 链接 |
| 2022 | 改进的双重相关性减少网络 (IDCRN) | arXiv | 链接 | - |
| 2022 | 面向异质性的图自监督学习 (HGRL) | CIKM | 链接 | 链接 |
| 2022 | S3GC:可扩展的自监督图聚类 (S3GC) | NeurIPS | 链接 | 链接 |
| 2022 | 带有伪标签提示的自一致属性图对比聚类 (SCAGC) | TMM | 链接 | 链接 |
| 2022 | CGC:用于社区检测与追踪的对比图聚类 (CGC) | WWW | 链接 | - |
| 2022 | 无监督深度图结构学习 (SUBLIME) | WWW | 链接 | 链接 |
| 2022 | 具有双重冗余减少的属性图聚类 (AGC-DRR) | IJCAI | 链接 | 链接 |
| 2022 | 通过双重相关性减少进行深度图聚类 (DCRN) | AAAI | 链接 | 链接 |
| 2022 | RepBin:基于约束的宏基因组分箱用图表示学习 (RepBin) | AAAI | 链接 | 链接 |
| 2022 | 无增强的图自监督学习 (AFGRL) | AAAI | 链接 | 链接 |
| 2022 | SAIL:自我增强的图对比学习 (SAIL) | AAAI | 链接 | - |
| 2021 | 具有联合表示聚类的去偏图对比学习 (GDCL) | IJCAI | 链接 | 链接 |
| 2021 | 多视图对比图聚类 (MCGC) | NeurIPS | 链接 | 链接 |
| 2021 | 具有协同对比学习的自监督异质图神经网络 (HeCo) | SIGKDD | 链接 | 链接 |
| 2020 | 用于属性图嵌入的自适应图编码器 (AGE) | SIGKDD | 链接 | 链接 |
| 2020 | CommDGI:面向社区检测的深度图信息最大化的模型 (CommDGI) | CIKM | 链接 | 链接 |
| 2020 | 图上的多视图对比表示学习 (MVGRL) | ICML | 链接 | 链接 |
应用
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | EyeGraph:面向模块性的时空图聚类,用于连续事件驱动的眼动追踪 | NeurIPS | 链接 | - |
| 2024 | 先识别再推荐:迈向无监督的群体推荐(ITR) | NeurIPS | 链接 | 链接 |
| 2024 | 面向推荐中意图学习的端到端可学习聚类(ELCRec) | NeurIPS | 链接 | 链接 |
| 2023 | GuardFL:通过带属性的客户端图聚类防御联邦学习中的后门攻击 | TIFS | 链接 | 链接 |
其他
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2023 | 基于元学习的鲁棒图聚类,用于处理噪声图(MetaGC) | CIKM | 链接 | 链接 |
其他相关论文
深度聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | ProCom:一种少样本目标社区检测算法 | AAAI | 链接 | 链接 |
| 2024 | 通过融合社区结构与邻居信息的深度图聚类(DIGC) | IS | 链接 | - |
| 2024 | 信息增强型深度图聚类网络(IEDGCN) | Neurocomputing | 链接 | - |
| 2024 | 每个节点都不同:为属性图聚类动态融合自监督任务 | AAAI | 链接 | 链接 |
| 2024 | DGCLUSTER:一种基于模块度最大化的属性图聚类神经框架(DGCluster) | AAAI | 链接 | - |
| 2023 | 用于属性图聚类的互增强网络(MBN) | KBS | 链接 | - |
| 2023 | 面向图聚类的无冗余自监督关系学习 | TNNLS | 链接 | 链接 |
| 2023 | 属性多关系图的谱聚类 | SIGKDD | 链接 | - |
| 2023 | 带有噪声标签的局部图聚类 | Arxiv | 链接 | - |
| 2023 | 对属性图聚类深度学习方法的重新评估 | CIKM | 链接 | 链接 |
| 2023 | 基于元加权的鲁棒图聚类,适用于噪声图 | CIKM | 链接 | 链接 |
| 2023 | 同质性增强的图聚类结构学习 | CIKM | 链接 | 链接 |
| 2023 | 对属性图聚类深度学习方法的重新评估 | CIKM | 链接 | 链接 |
| 2023 | 超越证据下界:用于节点聚类的双变分图自编码器 | SDM | 链接 | 链接 |
| 2023 | GC-Flow:一种基于图的流网络,用于高效聚类 | ICLM | 链接 | 链接 |
| 2023 | 可扩展的属性图子空间聚类(SAGSC) | AAAI | 链接 | 链接 |
| 2022 | 自适应属性与结构子空间聚类网络(AASSC-Net) | TIP | 链接 | 链接 |
| 2022 | 孪生对比学习用于在线聚类 | IJCV | 链接 | 链接 |
| 2022 | 通过O(n)二分图卷积进行非图数据聚类 | TPAMI | 链接 | 链接 |
| 2022 | Ada-nets:通过在结构空间中自适应发现邻居进行人脸聚类 | ICLR | 链接 | 链接 |
| 2021 | 用于通用数据聚类的自适应图自编码器 | TPAMI | 链接 | 链接 |
| 2021 | 对比聚类 | AAAI | 链接 | 链接 |
| 2017 | 迈向适合k-means的空间:同步深度学习与聚类(DCN) | ICML | 链接 | 链接 |
| 2017 | 改进的局部结构保持深度嵌入式聚类(IDEC) | IJCAI | 链接 | 链接 |
| 2016 | 用于聚类分析的无监督深度嵌入(DEC) | ICML | 链接 | 链接 |
深度层次聚类
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2023 | 对比层次聚类(CHC) | ECML PKDD | 链接 | 链接 |
其他相关方法
| 年份 | 标题 | 会议/期刊 | 论文 | 代码 |
|---|---|---|---|---|
| 2024 | PSMC:基于模体导纳的图聚类可证明且可扩展算法 | KDD | 链接 | - |
| 2024 | 大规模属性二分图上的有效聚类(TPO) | arXiv | 链接 | - |
| 2023 | GPUSCAN++:GPU上的高效结构化图聚类 | arXiv | 链接 | - |
| 2022 | 用于属性图聚类的深度线性图注意力模型 | Knowl Based Syst | 链接 | - |
| 2022 | 基于随机游走的自监督学习的可扩展深度图聚类 | WWW | 链接 | - |
| 2022 | X-GOAL:多层异构图原型对比学习(X-GOAL) | arXiv | 链接 | - |
| 2022 | 基于多级子空间融合的深度图聚类 | PR | 链接 | - |
| 2022 | GRACE:用于属性图聚类的通用图卷积框架 | TKDD | 链接 | 链接 |
| 2022 | 细粒度属性图聚类 | SDM | 链接 | 链接 |
| 2022 | 多视图图嵌入聚类网络:联合自监督与块对角表示 | NN | 链接 | 链接 |
| 2022 | SAGES:用于无监督学习的采样式可扩展属性图嵌入 | TKDE | 链接 | - |
| 2022 | 图的自动化自监督学习 | ICLR | 链接 | 链接 |
| 2022 | 用于多视图聚类的平稳扩散状态神经估计 | AAAI | 链接 | 链接 |
| 2021 | 简单谱图卷积 | ICLR | 链接 | 链接 |
| 2021 | 用于属性图聚类的谱嵌入网络(SENet) | NN | 链接 | - |
| 2021 | 平滑度传感器:用于属性图聚类的自适应平滑过渡图卷积 | TCYB | 链接 | 链接 |
| 2021 | 多视图属性图聚类 | TKDE | 链接 | 链接 |
| 2021 | 高阶深度多路信息最大化 | WWW | 链接 | 链接 |
| 2021 | 图InfoClust:在图中最大化粗粒度互信息 | PAKDD | 链接 | 链接 |
| 2021 | 基于图滤波器的多视图属性图聚类 | IJCAI | 链接 | 链接 |
| 2021 | Graph-MVP:用于多层图的多视图原型对比学习 | arXiv | 链接 | 链接 |
| 2021 | 对比拉普拉斯特征映射 | NeurIPS | 链接 | 链接 |
| 2020 | 面向无监督图表示学习的聚类感知图神经网络 | arXiv | 链接 | - |
| 2020 | 分布诱导的双向GAN用于图表示学习 | CVPR | 链接 | 链接 |
| 2020 | 带有注意力机制的自适应图卷积网络用于共同显著性检测的图聚类 | CVPR | 链接 | 链接 |
| 2020 | 利用图神经网络进行图池化的谱聚类(MinCutPool) | ICML | 链接 | 链接 |
| 2020 | MAGNN:用于异质图嵌入的元路径聚合图神经网络 | WWW | 链接 | 链接 |
| 2020 | 无监督属性多层网络嵌入 | AAAI | 链接 | 链接 |
| 2020 | 跨图:鲁棒且无监督的属性图嵌入,适用于结构受损的图 | ICDM | 链接 | 链接 |
| 2020 | 多分类不平衡图卷积网络学习 | IJCAI | 链接 | - |
| 2020 | CAGNN:面向无监督图表示学习的聚类感知图神经网络 | arXiv | 链接 | - |
| 2020 | 通过深度自适应图最大化进行属性图聚类 | ICCKE | 链接 | - |
| 2019 | 异质图注意力网络(HAN) | WWW | 链接 | 链接 |
| 2019 | 多视图一致性图聚类 | TIP | 链接 | 链接 |
| 2019 | 通过自适应图卷积(AGC)进行属性图聚类 | IJCAI | 链接 | 链接 |
| 2016 | node2vec:面向网络的可扩展特征学习(node2vec) | SIGKDD | 链接 | 链接 |
| 2016 | 变分图自动编码器(GAE) | NeurIPS Workshop | 链接 | 链接 |
| 2015 | LINE:大规模信息网络嵌入(LINE) | WWW | 链接 | 链接 |
| 2014 | DeepWalk:社交表征的在线学习(DeepWalk) | SIGKDD | 链接 | 链接 |
基准数据集
我们将数据集分为两类:图数据集和非图数据集。图数据集是现实世界中的各类图结构,例如引用网络、社交网络等。非图数据集则不属于图类型。不过,在必要时,我们可以通过K近邻(KNN)算法构建“邻接矩阵”。
快速入门
- 第一步:从以下链接下载所有数据集:[Google Drive | 坚果云]。也可选择从表格中的Google Drive链接单独下载部分数据集。
- 第二步:将压缩包解压至**./dataset/**目录下。
- 第三步:在main.py中修改数据集的类型和名称。
- 第四步:运行main.py。
代码
- utils.py
- load_graph_data:加载图数据集
- load_data:加载非图数据集
- normalize_adj:对邻接矩阵进行归一化
- diffusion_adj:计算图扩散矩阵
- construct_graph:为非图数据集构建KNN图
- numpy_to_torch:将NumPy数组转换为PyTorch张量
- torch_to_numpy:将PyTorch张量转换为NumPy数组
- clustering.py
- setup_seed:固定随机种子
- evaluation:评估聚类性能
- k_means:K均值算法
- visualization.py
- t_sne:t-SNE算法
- similarity_plot:可视化嵌入或特征的余弦相似度矩阵
数据集详情
关于每个数据集的介绍,请参阅此处。
图数据集
数据集 样本数 维度 边数 类别数 链接 CORA 2708 1433 5278 7 cora.zip CITESEER 3327 3703 4552 6 citeseer.zip CITE 3327 3703 4552 6 cite.zip PUBMED 19717 500 44324 3 pubmed.zip DBLP 4057 334 3528 4 dblp.zip ACM 3025 1870 13128 3 acm.zip AMAP 7650 745 119081 8 amap.zip AMAC 13752 767 245861 10 amac.zip CORAFULL 19793 8710 63421 70 corafull.zip WIKI 2405 4973 8261 17 wiki.zip COCS 18333 6805 81894 15 cocs.zip CORNELL 183 1703 149 5 cornell.zip TEXAS 183 1703 162 5 texas.zip WISC 251 1703 257 5 wisc.zip FILM 7600 932 15009 5 film.zip BAT 131 81 1038 4 bat.zip EAT 399 203 5994 4 eat.zip UAT 1190 239 13599 4 uat.zip
边数:此处仅统计无向边的数量。
非图数据集
数据集 样本数 维度 类型 类别数 链接 USPS 9298 256 图像 10 usps.zip HHAR 10299 561 记录 6 hhar.zip REUT 10000 2000 文本 4 reut.zip
引用
@inproceedings{ITR,
title={识别后再推荐:迈向无监督群体推荐},
author={刘悦、朱世豪、杨天元、马健、钟文亮},
booktitle={NeurIPS会议论文集},
year={2024}
}
@article{ELCRec,
title={面向推荐系统中意图学习的端到端可学习聚类},
author={刘悦、朱世豪、夏俊、马英伟、马健、钟文亮、张冠楠、张克军、刘新旺},
booktitle={国际神经信息处理系统会议论文集},
year={2024}
}
@article{deep_graph_clustering_survey,
title={深度图聚类综述:分类、挑战与应用},
author={刘悦、夏俊、周思航、王思伟、郭锡峰、杨希洪、梁科、涂文轩、李Z·斯坦、刘新旺},
journal={arXiv预印本 arXiv:2211.12875},
year={2022}
}
@article{SCGC,
title={简单对比图聚类},
author={刘悦、杨希洪、周思航、刘新旺、王思伟、梁科、涂文轩、李亮},
journal={IEEE神经网络与学习系统汇刊},
year={2023},
publisher={IEEE}
}
@inproceedings{Dink_Net,
title={Dink-net:大规模图上的神经网络聚类},
author={刘悦、梁科、夏俊、周思航、杨希洪、刘新旺、李Stan Z},
booktitle={国际机器学习会议论文集},
year={2023}
}
@inproceedings{TGC_ML_ICLR,
title={深度时序图聚类},
author={刘萌、刘悦、梁科、涂文轩、王思伟、周思航、刘新旺},
booktitle={第12届国际表征学习大会},
year={2024}
}
@inproceedings{HSAN,
title={针对对比式深度图聚类的硬样本感知网络},
author={刘悦、杨希洪、周思航、刘新旺、王振、梁科、涂文轩、李亮、段景灿、陈灿灿},
booktitle={AAAI人工智能会议论文集},
volume={37},
number={7},
pages={8914-8922},
year={2023}
}
@inproceedings{DCRN,
title={基于双重相关性降低的深度图聚类},
author={刘悦、涂文轩、周思航、刘新旺、宋林轩、杨希洪、朱恩},
booktitle={AAAI人工智能会议论文集},
volume={36},
number={7},
pages={7603-7611},
year={2022}
}
@inproceedings{liuyue_RGC,
title={未知簇数的强化图聚类},
author={刘悦、梁科、夏俊、杨希洪、周思航、刘萌、刘新旺、李Stan Z},
booktitle={第31届ACM国际多媒体会议论文集},
pages={3528--3537},
year={2023}
}
@article{RGAE,
title={重新思考用于属性图聚类的图自编码器模型},
author={Mrabah, Nairouz、Bouguessa, Mohamed、Touati, Mohamed Fawzi、Ksantini, Riadh},
journal={IEEE知识与数据工程汇刊},
year={2022}
}
@article{yu2025guard,
title = {G${}^2$uardFL:通过属性客户端图聚类保护联邦学习免受后门攻击},
author = {于浩、马川、刘萌、杜天宇、丁明、向涛、季守玲、刘新旺},
journal = {IEEE信息取证与安全汇刊},
year = {2025},
doi = {10.1109/TIFS.2025.3639985}
}
其他相关优秀仓库
相似工具推荐
openclaw
OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你
stable-diffusion-webui
stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
ComfyUI
ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。
markitdown
MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器
LLMs-from-scratch
LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备