Skip to content

add HGSketch#293

Open
grzhx wants to merge 2 commits intoBUPT-GAMMA:mainfrom
grzhx:feat/HGSketch
Open

add HGSketch#293
grzhx wants to merge 2 commits intoBUPT-GAMMA:mainfrom
grzhx:feat/HGSketch

Conversation

@grzhx
Copy link
Copy Markdown

@grzhx grzhx commented Mar 13, 2026

新增 HGSketch 模型

概述

新增 HGSketch 非参数图级别表示模型,通过提取单纯复形捕获高阶拓扑结构,结合局部敏感哈希(LSH)将异构图映射到低维 Hamming 空间,配合线性分类器完成图分类任务。

算法流程

  1. 提取 k-单纯形并构建 Hodge Laplacian 矩阵
  2. 基于节点类型的 one-hot 编码初始化异构特征
  3. 局部信息放大 + 全局结构增强
  4. 迭代 LSH 生成二值哈希码
  5. 线性化后接 Logistic Regression 分类

改动文件

  • openhgnn/models/HGSketch.py — 模型核心实现
  • openhgnn/trainerflow/HGSketch_trainer.py — 训练流程(计算嵌入 + LR 分类)
  • openhgnn/config.ini / config.py — 配置参数(K=2, R=3, D=128, seed=0)
  • openhgnn/models/__init__.py / trainerflow/__init__.py / experiment.py — 注册绑定

测试

tests/test_hgsketch.py,20 个单元测试全部通过,覆盖注册、配置、核心算法、线性化、Hodge Laplacian 及边界情况。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant