TGHE: Template-based Graph Homomorphic Encryption for Privacy-Preserving GNN Inference in Edge-Cloud Systems

TL;DR

提出TGHE:基于模板的图同态加密,支持大规模动态金融图的隐私保护推理。

cs.CR 🔴 高级 2026-06-25 45 次浏览
Ngoc Bao Anh Le Thai T. Vu John Le Heath Cooper Jun Shen
图神经网络 同态加密 隐私保护 边缘计算 金融图

核心发现

方法论

TGHE采用边缘端提取和标准化局部自我图(ego-graph),利用模板现象将局部结构归一化。通过结构签名匹配,将相似的局部树打包到共享CKKS密文中,实现SIMD并行推理。引入近似模板拟合和拓扑折叠两大优化器,提升SIMD覆盖率,显著降低计算复杂度。该方法解耦全局图规模与查询成本,适应动态大规模金融图。核心算法包括节点特征提取、树形化、模板匹配、密文打包及优化策略。

关键结果

  • 在3.7百万节点、4.3百万边的DGraphFin数据集上,TGHE-Collapse实现66.9倍端到端加速,AUC损失小于0.002,超越传统顺序加密基线。
  • 采用模板匹配后,绝大部分查询(98%以上)通过SIMD批处理完成,极大提升效率。
  • 拓扑折叠策略完全消除fallback路径,显著减少旋转次数,提升整体性能。

研究意义

该研究突破了HE-GNN在大规模动态图上的应用瓶颈,推动隐私保护机器学习在金融、医疗等敏感行业的落地。通过边缘端局部处理与云端推理结合,有效解决数据隐私与计算效率的矛盾,为大规模隐私保护图分析提供新思路。该框架实现了在百万级节点规模下的高效加密推理,具有重要的学术和工业价值。

技术贡献

提出基于模板的图结构归一化与密文打包技术,创新性地利用模板现象实现结构相似图的共享密文,突破了传统HE-GNN受全局图规模限制的瓶颈。引入近似模板拟合和拓扑折叠两大优化器,显著提升SIMD利用率。系统设计兼容GEARSage架构,支持线性邻居聚合,优化了多层GNN的加密推理流程。整体方案实现了百万节点级别的高效隐私推理,推动HE-GNN的实用化。

新颖性

首次系统性利用局部结构模板现象,将局部图归一化并实现密文共享,极大扩展HE-GNN的应用规模。不同于以往图中心设计,TGHE采用边缘端提取、结构归一化策略,解决动态大规模金融图的隐私保护难题。引入两大长尾优化器,确保几乎全部查询通过SIMD批处理,大幅提升效率,填补了HE-GNN在大规模场景下的空白。

局限性

  • 当前方法依赖特定的GNN架构(如GEARSage)和线性邻居聚合,非线性或注意力机制模型难以直接应用。
  • 拓扑折叠策略在极端异质或高复杂度图中可能效果有限,需进一步优化。
  • 密文打包和优化器虽提升效率,但在极端大规模或高动态场景中仍存在潜在性能瓶颈。

未来方向

未来将扩展支持更多异构GNN架构,研究非线性激活的近似策略,优化拓扑折叠在复杂场景中的适应性。同时,探索多模态、多任务场景下的隐私保护方案,推动HE-GNN在实际行业中的广泛应用。

AI 总览摘要

在金融、医疗等行业,数据隐私保护与大规模图分析的需求日益增长。传统的同态加密(HE)技术在图神经网络(GNN)中的应用受限于全局图规模,难以应对百万级节点的动态场景。本文提出TGHE(基于模板的图同态加密)框架,创新性地利用局部结构模板现象,将局部自我图归一化并打包到共享密文中,实现边缘端提取、标准化与云端推理的高效结合。

核心思想在于:尽管金融交易图复杂多变,但局部结构表现出高度的模板化。通过模板签名匹配,将相似的局部树结构归入同一密文,实现SIMD并行推理。引入近似模板拟合和拓扑折叠两大优化器,极大提升密文利用率,减少不匹配查询的计算负担。

在3.7百万节点、4.3百万边的DGraphFin数据集上,TGHE-Collapse实现66.9倍的端到端加速,几乎无AUC性能损失(<0.002),验证了其在大规模金融图场景中的实用性。这一突破为隐私保护的图分析提供了新的技术路径,推动HE-GNN在实际行业中的应用落地。未来工作将扩展支持多样化GNN架构,优化复杂场景下的拓扑折叠策略,促进其在更多行业的推广。

深度分析

研究背景

随着金融、医疗等行业对数据隐私的重视,基于同态加密的图神经网络(HE-GNN)逐渐成为研究热点。早期工作如CryptoGCN、LinGCN等,主要关注在有限规模图上的加密推理,但受限于全局邻接矩阵的存储与计算复杂度,难以扩展到百万级节点。近年来,边缘端提取局部子图、结构归一化的思想逐渐兴起,旨在突破全局图的限制,实现大规模动态场景下的隐私保护推理。

核心问题

现有HE-GNN方法多采用图中心设计,依赖全局邻接矩阵,导致计算复杂度随图规模线性增长,限制在数万节点以内。同时,静态图结构难以适应金融系统中节点频繁新增的动态特性。此外,结构预处理可能泄露敏感信息,限制了实际应用。如何在保证隐私的同时,实现大规模、动态、结构多变的金融图的高效推理,成为亟待解决的难题。

核心创新

第一,提出局部结构模板现象,利用局部自我图的结构相似性实现归一化与密文共享,突破全局图规模限制;第二,设计模板签名匹配机制,将相似结构的局部树归入同一密文,支持SIMD批处理;第三,引入近似模板拟合和拓扑折叠优化器,极大提升密文利用率,减少fallback路径,显著提升效率。这些创新结合边缘提取、结构归一化和密文打包技术,为HE-GNN的实用化提供新思路。

方法详解

  • �� 边缘端提取局部自我图(ego-graph),限制最大深度和邻居数以控制密文复杂度。• 通过确定性截断(capping)和树形化(treeification)将局部子图转化为结构唯一的树形角色图。• 计算每个树的结构签名(模板签名),识别不同查询的结构相似性。• 利用模板签名将结构相似的局部树打包到共享CKKS密文中,实现SIMD并行推理。• 引入近似模板拟合,将不匹配的树通过填充映射到最接近的模板,保证大部分查询在批处理内完成。• 采用拓扑折叠策略,将邻居聚合中的Hop-2结构预先在明文中合并,消除复杂拓扑差异,完全避免fallback。• 设计多层GNN架构的线性邻居聚合,支持模板折叠,简化密文操作流程。

实验设计

采用DGraphFin(3.7M节点、4.3M边)作为测试平台,训练基于GEARSage架构的两层模型,特征维度为17和32,激活函数为二次多项式。对比非加密顺序基线与TGHE不同优化策略(Base和Collapse),评估指标为ROC-AUC。实验中随机采样10,000查询节点,测量端到端运行时间、旋转次数和密文操作数。通过不同阈值τ调节模板匹配的精度-速度折衷,验证优化器效果。

结果分析

TGHE-Collapse在DGraphFin上实现66.9倍加速,端到端时间从39107秒降至115.9秒,旋转次数减少99%,AUC损失小于0.002。模板匹配后,98%以上查询通过SIMD批处理,大幅提升效率。拓扑折叠策略完全消除fallback路径,显著降低计算成本。实验验证了方法在大规模金融图上的可行性与高效性,优于传统HE-GNN方案。

应用场景

该技术适用于金融欺诈检测、反洗钱、医疗数据分析等场景,能在保证数据隐私的前提下实现大规模图的高效推理。边缘端提取局部子图,云端进行加密推理,满足行业对数据安全和实时性的双重需求。未来可扩展到多模态、多任务场景,推动隐私保护机器学习的产业化。

局限与展望

目前方法依赖特定的GNN架构(如GEARSage)和线性邻居聚合,难以直接支持非线性激活或注意力机制模型。拓扑折叠在极端异质或高复杂度图中效果有限,需进一步优化。密文打包和优化器虽提升效率,但在超大规模或高动态场景中仍存在性能瓶颈。未来需研究更通用的结构归一化策略和多样化模型支持。

通俗解读 非专业人士也能看懂

想象你在一个大型工厂里工作,每天都要处理很多不同的产品。工厂里的每个工人只负责自己的一部分工作,比如装配、包装或检验。为了确保产品的隐私和安全,工厂采用一种特殊的秘密箱子(类似加密箱),工人把自己的工作内容放进去,然后交给中央控制室处理。控制室可以在箱子里操作,比如拼装、检测,但看不到具体内容,只能得到一些加密的结果。工厂发现,虽然每个产品都不同,但很多产品的结构都很相似,比如都是装配某几件零件组成的。这种相似性让工厂可以把很多产品的操作合并到一起,用少量的秘密箱子同时处理大量产品,大大提高效率。这个方法就像TGHE一样,利用局部结构的模板,把相似的局部图结构打包在一起,既保护了隐私,又能快速处理海量数据。

简单解释 像给14岁少年讲一样

想象你在学校里,有很多不同的朋友,每个人都喜欢不同的游戏。有些朋友喜欢玩像素风格,有些喜欢3D动作游戏,但他们的游戏玩法其实有很多相似的地方。现在,你想帮大家玩游戏,但又不想让别人知道你们的秘密(比如你们的游戏密码)。于是,你用一种神奇的盒子,把每个朋友的游戏操作都装进去,这样别人只能看到盒子里的内容,不能知道具体细节。更厉害的是,很多朋友的游戏玩法其实很像,比如都需要跳跃、跑步、攻击。你发现,把这些相似的玩法放在一起,用一个大盒子装起来,就可以同时帮很多朋友玩游戏,既快又安全。这就像TGHE用模板把相似的图结构打包,让计算变得更快、更安全,大家都能享受到隐私保护的智能游戏体验!

原文摘要

Existing homomorphic encryption (HE)-based GNN systems adopt a graph-centric paradigm that couples per-query cost to global graph size, limiting evaluations to at most ~20k nodes and making them incompatible with dynamic, large-scale financial graphs. We propose TGHE (Template-based Graph Homomorphic Encryption), an ego-centric framework that resolves this by exploiting a template phenomenon: local computation trees in transaction graphs converge into a small set of structural shapes. TGHE canonicalizes ego-graphs at the edge and packs structurally identical trees into shared CKKS ciphertexts for SIMD-parallel encrypted inference, with two long-tail optimizers (Approximate Template Fitting and Topology Collapse) ensuring full SIMD coverage. On DGraphFin (3.7M nodes, 4.3M edges), TGHE-Collapse achieves a 66.9x speedup over the sequential encrypted baseline with less than 0.002 AUC loss.

cs.CR cs.AI